ef589a0640
* Migrate to pytest All tests were update to use pytest. Tests were alse black formatted. Eventually all code will be black formatted but need to merge some PRs first.
366 lines
9.6 KiB
Python
366 lines
9.6 KiB
Python
#! /usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# Author: Hamilton Kibbe <ham@hamiltonkib.be>
|
|
import os
|
|
import pytest
|
|
|
|
from ..cam import FileSettings
|
|
from ..excellon import read, detect_excellon_format, ExcellonFile, ExcellonParser
|
|
from ..excellon import DrillHit, DrillSlot
|
|
from ..excellon_statements import ExcellonTool, RouteModeStmt
|
|
|
|
|
|
NCDRILL_FILE = os.path.join(os.path.dirname(__file__), "resources/ncdrill.DRD")
|
|
|
|
|
|
def test_format_detection():
|
|
""" Test file type detection
|
|
"""
|
|
with open(NCDRILL_FILE, "rU") as f:
|
|
data = f.read()
|
|
settings = detect_excellon_format(data)
|
|
assert settings["format"] == (2, 4)
|
|
assert settings["zeros"] == "trailing"
|
|
|
|
settings = detect_excellon_format(filename=NCDRILL_FILE)
|
|
assert settings["format"] == (2, 4)
|
|
assert settings["zeros"] == "trailing"
|
|
|
|
|
|
def test_read():
|
|
ncdrill = read(NCDRILL_FILE)
|
|
assert isinstance(ncdrill, ExcellonFile)
|
|
|
|
|
|
def test_write():
|
|
ncdrill = read(NCDRILL_FILE)
|
|
ncdrill.write("test.ncd")
|
|
with open(NCDRILL_FILE, "rU") as src:
|
|
srclines = src.readlines()
|
|
with open("test.ncd", "rU") as res:
|
|
for idx, line in enumerate(res):
|
|
assert line.strip() == srclines[idx].strip()
|
|
os.remove("test.ncd")
|
|
|
|
|
|
def test_read_settings():
|
|
ncdrill = read(NCDRILL_FILE)
|
|
assert ncdrill.settings["format"] == (2, 4)
|
|
assert ncdrill.settings["zeros"] == "trailing"
|
|
|
|
|
|
def test_bounding_box():
|
|
ncdrill = read(NCDRILL_FILE)
|
|
xbound, ybound = ncdrill.bounding_box
|
|
pytest.approx(xbound, (0.1300, 2.1430))
|
|
pytest.approx(ybound, (0.3946, 1.7164))
|
|
|
|
|
|
def test_report():
|
|
ncdrill = read(NCDRILL_FILE)
|
|
rprt = ncdrill.report()
|
|
|
|
|
|
def test_conversion():
|
|
import copy
|
|
|
|
ncdrill = read(NCDRILL_FILE)
|
|
assert ncdrill.settings.units == "inch"
|
|
ncdrill_inch = copy.deepcopy(ncdrill)
|
|
|
|
ncdrill.to_metric()
|
|
assert ncdrill.settings.units == "metric"
|
|
for tool in iter(ncdrill_inch.tools.values()):
|
|
tool.to_metric()
|
|
|
|
for statement in ncdrill_inch.statements:
|
|
statement.to_metric()
|
|
|
|
for m_tool, i_tool in zip(
|
|
iter(ncdrill.tools.values()), iter(ncdrill_inch.tools.values())
|
|
):
|
|
assert i_tool == m_tool
|
|
|
|
for m, i in zip(ncdrill.primitives, ncdrill_inch.primitives):
|
|
|
|
assert m.position == i.position, "%s not equal to %s" % (m, i)
|
|
assert m.diameter == i.diameter, "%s not equal to %s" % (m, i)
|
|
|
|
|
|
def test_parser_hole_count():
|
|
settings = FileSettings(**detect_excellon_format(NCDRILL_FILE))
|
|
p = ExcellonParser(settings)
|
|
p.parse(NCDRILL_FILE)
|
|
assert p.hole_count == 36
|
|
|
|
|
|
def test_parser_hole_sizes():
|
|
settings = FileSettings(**detect_excellon_format(NCDRILL_FILE))
|
|
p = ExcellonParser(settings)
|
|
p.parse(NCDRILL_FILE)
|
|
assert p.hole_sizes == [0.0236, 0.0354, 0.04, 0.126, 0.128]
|
|
|
|
|
|
def test_parse_whitespace():
|
|
p = ExcellonParser(FileSettings())
|
|
assert p._parse_line(" ") == None
|
|
|
|
|
|
def test_parse_comment():
|
|
p = ExcellonParser(FileSettings())
|
|
p._parse_line(";A comment")
|
|
assert p.statements[0].comment == "A comment"
|
|
|
|
|
|
def test_parse_format_comment():
|
|
p = ExcellonParser(FileSettings())
|
|
p._parse_line("; FILE_FORMAT=9:9 ")
|
|
assert p.format == (9, 9)
|
|
|
|
|
|
def test_parse_header():
|
|
p = ExcellonParser(FileSettings())
|
|
p._parse_line("M48 ")
|
|
assert p.state == "HEADER"
|
|
p._parse_line("M95 ")
|
|
assert p.state == "DRILL"
|
|
|
|
|
|
def test_parse_rout():
|
|
p = ExcellonParser(FileSettings())
|
|
p._parse_line("G00X040944Y019842")
|
|
assert p.state == "ROUT"
|
|
p._parse_line("G05 ")
|
|
assert p.state == "DRILL"
|
|
|
|
|
|
def test_parse_version():
|
|
p = ExcellonParser(FileSettings())
|
|
p._parse_line("VER,1 ")
|
|
assert p.statements[0].version == 1
|
|
p._parse_line("VER,2 ")
|
|
assert p.statements[1].version == 2
|
|
|
|
|
|
def test_parse_format():
|
|
p = ExcellonParser(FileSettings())
|
|
p._parse_line("FMAT,1 ")
|
|
assert p.statements[0].format == 1
|
|
p._parse_line("FMAT,2 ")
|
|
assert p.statements[1].format == 2
|
|
|
|
|
|
def test_parse_units():
|
|
settings = FileSettings(units="inch", zeros="trailing")
|
|
p = ExcellonParser(settings)
|
|
p._parse_line(";METRIC,LZ")
|
|
assert p.units == "inch"
|
|
assert p.zeros == "trailing"
|
|
p._parse_line("METRIC,LZ")
|
|
assert p.units == "metric"
|
|
assert p.zeros == "leading"
|
|
|
|
|
|
def test_parse_incremental_mode():
|
|
settings = FileSettings(units="inch", zeros="trailing")
|
|
p = ExcellonParser(settings)
|
|
assert p.notation == "absolute"
|
|
p._parse_line("ICI,ON ")
|
|
assert p.notation == "incremental"
|
|
p._parse_line("ICI,OFF ")
|
|
assert p.notation == "absolute"
|
|
|
|
|
|
def test_parse_absolute_mode():
|
|
settings = FileSettings(units="inch", zeros="trailing")
|
|
p = ExcellonParser(settings)
|
|
assert p.notation == "absolute"
|
|
p._parse_line("ICI,ON ")
|
|
assert p.notation == "incremental"
|
|
p._parse_line("G90 ")
|
|
assert p.notation == "absolute"
|
|
|
|
|
|
def test_parse_repeat_hole():
|
|
p = ExcellonParser(FileSettings())
|
|
p.active_tool = ExcellonTool(FileSettings(), number=8)
|
|
p._parse_line("R03X1.5Y1.5")
|
|
assert p.statements[0].count == 3
|
|
|
|
|
|
def test_parse_incremental_position():
|
|
p = ExcellonParser(FileSettings(notation="incremental"))
|
|
p._parse_line("X01Y01")
|
|
p._parse_line("X01Y01")
|
|
assert p.pos == [2.0, 2.0]
|
|
|
|
|
|
def test_parse_unknown():
|
|
p = ExcellonParser(FileSettings())
|
|
p._parse_line("Not A Valid Statement")
|
|
assert p.statements[0].stmt == "Not A Valid Statement"
|
|
|
|
|
|
def test_drill_hit_units_conversion():
|
|
""" Test unit conversion for drill hits
|
|
"""
|
|
# Inch hit
|
|
settings = FileSettings(units="inch")
|
|
tool = ExcellonTool(settings, diameter=1.0)
|
|
hit = DrillHit(tool, (1.0, 1.0))
|
|
|
|
assert hit.tool.settings.units == "inch"
|
|
assert hit.tool.diameter == 1.0
|
|
assert hit.position == (1.0, 1.0)
|
|
|
|
# No Effect
|
|
hit.to_inch()
|
|
|
|
assert hit.tool.settings.units == "inch"
|
|
assert hit.tool.diameter == 1.0
|
|
assert hit.position == (1.0, 1.0)
|
|
|
|
# Should convert
|
|
hit.to_metric()
|
|
|
|
assert hit.tool.settings.units == "metric"
|
|
assert hit.tool.diameter == 25.4
|
|
assert hit.position == (25.4, 25.4)
|
|
|
|
# No Effect
|
|
hit.to_metric()
|
|
|
|
assert hit.tool.settings.units == "metric"
|
|
assert hit.tool.diameter == 25.4
|
|
assert hit.position == (25.4, 25.4)
|
|
|
|
# Convert back to inch
|
|
hit.to_inch()
|
|
|
|
assert hit.tool.settings.units == "inch"
|
|
assert hit.tool.diameter == 1.0
|
|
assert hit.position == (1.0, 1.0)
|
|
|
|
|
|
def test_drill_hit_offset():
|
|
TEST_VECTORS = [
|
|
((0.0, 0.0), (0.0, 1.0), (0.0, 1.0)),
|
|
((0.0, 0.0), (1.0, 1.0), (1.0, 1.0)),
|
|
((1.0, 1.0), (0.0, -1.0), (1.0, 0.0)),
|
|
((1.0, 1.0), (-1.0, -1.0), (0.0, 0.0)),
|
|
]
|
|
for position, offset, expected in TEST_VECTORS:
|
|
settings = FileSettings(units="inch")
|
|
tool = ExcellonTool(settings, diameter=1.0)
|
|
hit = DrillHit(tool, position)
|
|
|
|
assert hit.position == position
|
|
|
|
hit.offset(offset[0], offset[1])
|
|
|
|
assert hit.position == expected
|
|
|
|
|
|
def test_drill_slot_units_conversion():
|
|
""" Test unit conversion for drill hits
|
|
"""
|
|
# Inch hit
|
|
settings = FileSettings(units="inch")
|
|
tool = ExcellonTool(settings, diameter=1.0)
|
|
hit = DrillSlot(tool, (1.0, 1.0), (10.0, 10.0), DrillSlot.TYPE_ROUT)
|
|
|
|
assert hit.tool.settings.units == "inch"
|
|
assert hit.tool.diameter == 1.0
|
|
assert hit.start == (1.0, 1.0)
|
|
assert hit.end == (10.0, 10.0)
|
|
|
|
# No Effect
|
|
hit.to_inch()
|
|
|
|
assert hit.tool.settings.units == "inch"
|
|
assert hit.tool.diameter == 1.0
|
|
assert hit.start == (1.0, 1.0)
|
|
assert hit.end == (10.0, 10.0)
|
|
|
|
# Should convert
|
|
hit.to_metric()
|
|
|
|
assert hit.tool.settings.units == "metric"
|
|
assert hit.tool.diameter == 25.4
|
|
assert hit.start == (25.4, 25.4)
|
|
assert hit.end == (254.0, 254.0)
|
|
|
|
# No Effect
|
|
hit.to_metric()
|
|
|
|
assert hit.tool.settings.units == "metric"
|
|
assert hit.tool.diameter == 25.4
|
|
assert hit.start == (25.4, 25.4)
|
|
assert hit.end == (254.0, 254.0)
|
|
|
|
# Convert back to inch
|
|
hit.to_inch()
|
|
|
|
assert hit.tool.settings.units == "inch"
|
|
assert hit.tool.diameter == 1.0
|
|
assert hit.start == (1.0, 1.0)
|
|
assert hit.end == (10.0, 10.0)
|
|
|
|
|
|
def test_drill_slot_offset():
|
|
TEST_VECTORS = [
|
|
((0.0, 0.0), (1.0, 1.0), (0.0, 0.0), (0.0, 0.0), (1.0, 1.0)),
|
|
((0.0, 0.0), (1.0, 1.0), (1.0, 0.0), (1.0, 0.0), (2.0, 1.0)),
|
|
((0.0, 0.0), (1.0, 1.0), (1.0, 1.0), (1.0, 1.0), (2.0, 2.0)),
|
|
((0.0, 0.0), (1.0, 1.0), (-1.0, 1.0), (-1.0, 1.0), (0.0, 2.0)),
|
|
]
|
|
for start, end, offset, expected_start, expected_end in TEST_VECTORS:
|
|
settings = FileSettings(units="inch")
|
|
tool = ExcellonTool(settings, diameter=1.0)
|
|
slot = DrillSlot(tool, start, end, DrillSlot.TYPE_ROUT)
|
|
|
|
assert slot.start == start
|
|
assert slot.end == end
|
|
|
|
slot.offset(offset[0], offset[1])
|
|
|
|
assert slot.start == expected_start
|
|
assert slot.end == expected_end
|
|
|
|
|
|
def test_drill_slot_bounds():
|
|
TEST_VECTORS = [
|
|
((0.0, 0.0), (1.0, 1.0), 1.0, ((-0.5, 1.5), (-0.5, 1.5))),
|
|
((0.0, 0.0), (1.0, 1.0), 0.5, ((-0.25, 1.25), (-0.25, 1.25))),
|
|
]
|
|
for start, end, diameter, expected in TEST_VECTORS:
|
|
settings = FileSettings(units="inch")
|
|
tool = ExcellonTool(settings, diameter=diameter)
|
|
slot = DrillSlot(tool, start, end, DrillSlot.TYPE_ROUT)
|
|
|
|
assert slot.bounding_box == expected
|
|
|
|
|
|
def test_handling_multi_line_g00_and_g1():
|
|
"""Route Mode statements with coordinates on separate line are handled
|
|
"""
|
|
test_data = """
|
|
%
|
|
M48
|
|
M72
|
|
T01C0.0236
|
|
%
|
|
T01
|
|
G00
|
|
X040944Y019842
|
|
M15
|
|
G01
|
|
X040944Y020708
|
|
M16
|
|
"""
|
|
uut = ExcellonParser()
|
|
uut.parse_raw(test_data)
|
|
assert (
|
|
len([stmt for stmt in uut.statements if isinstance(stmt, RouteModeStmt)]) == 2
|
|
)
|