summaryrefslogtreecommitdiff
path: root/gerbonara/gerber/tests/test_excellon_statements.py
diff options
context:
space:
mode:
Diffstat (limited to 'gerbonara/gerber/tests/test_excellon_statements.py')
-rw-r--r--gerbonara/gerber/tests/test_excellon_statements.py734
1 files changed, 734 insertions, 0 deletions
diff --git a/gerbonara/gerber/tests/test_excellon_statements.py b/gerbonara/gerber/tests/test_excellon_statements.py
new file mode 100644
index 0000000..41fe294
--- /dev/null
+++ b/gerbonara/gerber/tests/test_excellon_statements.py
@@ -0,0 +1,734 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# Author: Hamilton Kibbe <ham@hamiltonkib.be>
+
+import pytest
+from ..excellon_statements import *
+from ..cam import FileSettings
+
+
+def test_excellon_statement_implementation():
+ stmt = ExcellonStatement()
+ pytest.raises(NotImplementedError, stmt.from_excellon, None)
+ pytest.raises(NotImplementedError, stmt.to_excellon)
+
+
+def test_excellontstmt():
+ """ Smoke test ExcellonStatement
+ """
+ stmt = ExcellonStatement()
+ stmt.to_inch()
+ stmt.to_metric()
+ stmt.offset()
+
+
+def test_excellontool_factory():
+ """ Test ExcellonTool factory methods
+ """
+ exc_line = "T8F01B02S00003H04Z05C0.12500"
+ settings = FileSettings(
+ format=(2, 5), zero_suppression="trailing", units="inch", notation="absolute"
+ )
+ tool = ExcellonTool.from_excellon(exc_line, settings)
+ assert tool.number == 8
+ assert tool.diameter == 0.125
+ assert tool.feed_rate == 1
+ assert tool.retract_rate == 2
+ assert tool.rpm == 3
+ assert tool.max_hit_count == 4
+ assert tool.depth_offset == 5
+
+ stmt = {
+ "number": 8,
+ "feed_rate": 1,
+ "retract_rate": 2,
+ "rpm": 3,
+ "diameter": 0.125,
+ "max_hit_count": 4,
+ "depth_offset": 5,
+ }
+ tool = ExcellonTool.from_dict(settings, stmt)
+ assert tool.number == 8
+ assert tool.diameter == 0.125
+ assert tool.feed_rate == 1
+ assert tool.retract_rate == 2
+ assert tool.rpm == 3
+ assert tool.max_hit_count == 4
+ assert tool.depth_offset == 5
+
+
+def test_excellontool_dump():
+ """ Test ExcellonTool to_excellon()
+ """
+ exc_lines = [
+ "T01F0S0C0.01200",
+ "T02F0S0C0.01500",
+ "T03F0S0C0.01968",
+ "T04F0S0C0.02800",
+ "T05F0S0C0.03300",
+ "T06F0S0C0.03800",
+ "T07F0S0C0.04300",
+ "T08F0S0C0.12500",
+ "T09F0S0C0.13000",
+ "T08B01F02H03S00003C0.12500Z04",
+ "T01F0S300.999C0.01200",
+ ]
+ settings = FileSettings(
+ format=(2, 5), zero_suppression="trailing", units="inch", notation="absolute"
+ )
+ for line in exc_lines:
+ tool = ExcellonTool.from_excellon(line, settings)
+ assert tool.to_excellon() == line
+
+
+def test_excellontool_order():
+ settings = FileSettings(
+ format=(2, 5), zero_suppression="trailing", units="inch", notation="absolute"
+ )
+ line = "T8F00S00C0.12500"
+ tool1 = ExcellonTool.from_excellon(line, settings)
+ line = "T8C0.12500F00S00"
+ tool2 = ExcellonTool.from_excellon(line, settings)
+ assert tool1.diameter == tool2.diameter
+ assert tool1.feed_rate == tool2.feed_rate
+ assert tool1.rpm == tool2.rpm
+
+
+def test_excellontool_conversion():
+ tool = ExcellonTool.from_dict(
+ FileSettings(units="metric"), {"number": 8, "diameter": 25.4}
+ )
+ tool.to_inch()
+ assert tool.diameter == 1.0
+ tool = ExcellonTool.from_dict(
+ FileSettings(units="inch"), {"number": 8, "diameter": 1.0}
+ )
+ tool.to_metric()
+ assert tool.diameter == 25.4
+
+ # Shouldn't change units if we're already using target units
+ tool = ExcellonTool.from_dict(
+ FileSettings(units="inch"), {"number": 8, "diameter": 25.4}
+ )
+ tool.to_inch()
+ assert tool.diameter == 25.4
+ tool = ExcellonTool.from_dict(
+ FileSettings(units="metric"), {"number": 8, "diameter": 1.0}
+ )
+ tool.to_metric()
+ assert tool.diameter == 1.0
+
+
+def test_excellontool_repr():
+ tool = ExcellonTool.from_dict(FileSettings(), {"number": 8, "diameter": 0.125})
+ assert str(tool) == "<ExcellonTool 08: 0.125in. dia.>"
+ tool = ExcellonTool.from_dict(
+ FileSettings(units="metric"), {"number": 8, "diameter": 0.125}
+ )
+ assert str(tool) == "<ExcellonTool 08: 0.125mm dia.>"
+
+
+def test_excellontool_equality():
+ t = ExcellonTool.from_dict(FileSettings(), {"number": 8, "diameter": 0.125})
+ t1 = ExcellonTool.from_dict(FileSettings(), {"number": 8, "diameter": 0.125})
+ assert t == t1
+ t1 = ExcellonTool.from_dict(
+ FileSettings(units="metric"), {"number": 8, "diameter": 0.125}
+ )
+ assert t != t1
+
+
+def test_toolselection_factory():
+ """ Test ToolSelectionStmt factory method
+ """
+ stmt = ToolSelectionStmt.from_excellon("T01")
+ assert stmt.tool == 1
+ assert stmt.compensation_index == None
+ stmt = ToolSelectionStmt.from_excellon("T0223")
+ assert stmt.tool == 2
+ assert stmt.compensation_index == 23
+ stmt = ToolSelectionStmt.from_excellon("T042")
+ assert stmt.tool == 42
+ assert stmt.compensation_index == None
+
+
+def test_toolselection_dump():
+ """ Test ToolSelectionStmt to_excellon()
+ """
+ lines = ["T01", "T0223", "T10", "T09", "T0000"]
+ for line in lines:
+ stmt = ToolSelectionStmt.from_excellon(line)
+ assert stmt.to_excellon() == line
+
+
+def test_z_axis_infeed_rate_factory():
+ """ Test ZAxisInfeedRateStmt factory method
+ """
+ stmt = ZAxisInfeedRateStmt.from_excellon("F01")
+ assert stmt.rate == 1
+ stmt = ZAxisInfeedRateStmt.from_excellon("F2")
+ assert stmt.rate == 2
+ stmt = ZAxisInfeedRateStmt.from_excellon("F03")
+ assert stmt.rate == 3
+
+
+def test_z_axis_infeed_rate_dump():
+ """ Test ZAxisInfeedRateStmt to_excellon()
+ """
+ inputs = [("F01", "F01"), ("F2", "F02"), ("F00003", "F03")]
+ for input_rate, expected_output in inputs:
+ stmt = ZAxisInfeedRateStmt.from_excellon(input_rate)
+ assert stmt.to_excellon() == expected_output
+
+
+def test_coordinatestmt_factory():
+ """ Test CoordinateStmt factory method
+ """
+ settings = FileSettings(
+ format=(2, 5), zero_suppression="trailing", units="inch", notation="absolute"
+ )
+
+ line = "X0278207Y0065293"
+ stmt = CoordinateStmt.from_excellon(line, settings)
+ assert stmt.x == 2.78207
+ assert stmt.y == 0.65293
+
+ # line = 'X02945'
+ # stmt = CoordinateStmt.from_excellon(line)
+ # assert_equal(stmt.x, 2.945)
+
+ # line = 'Y00575'
+ # stmt = CoordinateStmt.from_excellon(line)
+ # assert_equal(stmt.y, 0.575)
+
+ settings = FileSettings(
+ format=(2, 4), zero_suppression="leading", units="inch", notation="absolute"
+ )
+
+ line = "X9660Y4639"
+ stmt = CoordinateStmt.from_excellon(line, settings)
+ assert stmt.x == 0.9660
+ assert stmt.y == 0.4639
+ assert stmt.to_excellon(settings) == "X9660Y4639"
+ assert stmt.units == "inch"
+
+ settings.units = "metric"
+ stmt = CoordinateStmt.from_excellon(line, settings)
+ assert stmt.units == "metric"
+
+
+def test_coordinatestmt_dump():
+ """ Test CoordinateStmt to_excellon()
+ """
+ lines = [
+ "X278207Y65293",
+ "X243795",
+ "Y82528",
+ "Y86028",
+ "X251295Y81528",
+ "X2525Y78",
+ "X255Y575",
+ "Y52",
+ "X2675",
+ "Y575",
+ "X2425",
+ "Y52",
+ "X23",
+ ]
+ settings = FileSettings(
+ format=(2, 4), zero_suppression="leading", units="inch", notation="absolute"
+ )
+ for line in lines:
+ stmt = CoordinateStmt.from_excellon(line, settings)
+ assert stmt.to_excellon(settings) == line
+
+
+def test_coordinatestmt_conversion():
+
+ settings = FileSettings()
+ settings.units = "metric"
+ stmt = CoordinateStmt.from_excellon("X254Y254", settings)
+
+ # No effect
+ stmt.to_metric()
+ assert stmt.x == 25.4
+ assert stmt.y == 25.4
+
+ stmt.to_inch()
+ assert stmt.units == "inch"
+ assert stmt.x == 1.0
+ assert stmt.y == 1.0
+
+ # No effect
+ stmt.to_inch()
+ assert stmt.x == 1.0
+ assert stmt.y == 1.0
+
+ settings.units = "inch"
+ stmt = CoordinateStmt.from_excellon("X01Y01", settings)
+
+ # No effect
+ stmt.to_inch()
+ assert stmt.x == 1.0
+ assert stmt.y == 1.0
+
+ stmt.to_metric()
+ assert stmt.units == "metric"
+ assert stmt.x == 25.4
+ assert stmt.y == 25.4
+
+ # No effect
+ stmt.to_metric()
+ assert stmt.x == 25.4
+ assert stmt.y == 25.4
+
+
+def test_coordinatestmt_offset():
+ stmt = CoordinateStmt.from_excellon("X01Y01", FileSettings())
+ stmt.offset()
+ assert stmt.x == 1
+ assert stmt.y == 1
+ stmt.offset(1, 0)
+ assert stmt.x == 2.0
+ assert stmt.y == 1.0
+ stmt.offset(0, 1)
+ assert stmt.x == 2.0
+ assert stmt.y == 2.0
+
+
+def test_coordinatestmt_string():
+ settings = FileSettings(
+ format=(2, 4), zero_suppression="leading", units="inch", notation="absolute"
+ )
+ stmt = CoordinateStmt.from_excellon("X9660Y4639", settings)
+ assert str(stmt) == "<Coordinate Statement: X: 0.966 Y: 0.4639 >"
+
+
+def test_repeathole_stmt_factory():
+ stmt = RepeatHoleStmt.from_excellon(
+ "R0004X015Y32", FileSettings(zeros="leading", units="inch")
+ )
+ assert stmt.count == 4
+ assert stmt.xdelta == 1.5
+ assert stmt.ydelta == 32
+ assert stmt.units == "inch"
+
+ stmt = RepeatHoleStmt.from_excellon(
+ "R0004X015Y32", FileSettings(zeros="leading", units="metric")
+ )
+ assert stmt.units == "metric"
+
+
+def test_repeatholestmt_dump():
+ line = "R4X015Y32"
+ stmt = RepeatHoleStmt.from_excellon(line, FileSettings())
+ assert stmt.to_excellon(FileSettings()) == line
+
+
+def test_repeatholestmt_conversion():
+ line = "R4X0254Y254"
+ settings = FileSettings()
+ settings.units = "metric"
+ stmt = RepeatHoleStmt.from_excellon(line, settings)
+
+ # No effect
+ stmt.to_metric()
+ assert stmt.xdelta == 2.54
+ assert stmt.ydelta == 25.4
+
+ stmt.to_inch()
+ assert stmt.units == "inch"
+ assert stmt.xdelta == 0.1
+ assert stmt.ydelta == 1.0
+
+ # no effect
+ stmt.to_inch()
+ assert stmt.xdelta == 0.1
+ assert stmt.ydelta == 1.0
+
+ line = "R4X01Y1"
+ settings.units = "inch"
+ stmt = RepeatHoleStmt.from_excellon(line, settings)
+
+ # no effect
+ stmt.to_inch()
+ assert stmt.xdelta == 1.0
+ assert stmt.ydelta == 10.0
+
+ stmt.to_metric()
+ assert stmt.units == "metric"
+ assert stmt.xdelta == 25.4
+ assert stmt.ydelta == 254.0
+
+ # No effect
+ stmt.to_metric()
+ assert stmt.xdelta == 25.4
+ assert stmt.ydelta == 254.0
+
+
+def test_repeathole_str():
+ stmt = RepeatHoleStmt.from_excellon("R4X015Y32", FileSettings())
+ assert str(stmt) == "<Repeat Hole: 4 times, offset X: 1.5 Y: 32>"
+
+
+def test_commentstmt_factory():
+ """ Test CommentStmt factory method
+ """
+ line = ";Layer_Color=9474304"
+ stmt = CommentStmt.from_excellon(line)
+ assert stmt.comment == line[1:]
+
+ line = ";FILE_FORMAT=2:5"
+ stmt = CommentStmt.from_excellon(line)
+ assert stmt.comment == line[1:]
+
+ line = ";TYPE=PLATED"
+ stmt = CommentStmt.from_excellon(line)
+ assert stmt.comment == line[1:]
+
+
+def test_commentstmt_dump():
+ """ Test CommentStmt to_excellon()
+ """
+ lines = [";Layer_Color=9474304", ";FILE_FORMAT=2:5", ";TYPE=PLATED"]
+ for line in lines:
+ stmt = CommentStmt.from_excellon(line)
+ assert stmt.to_excellon() == line
+
+
+def test_header_begin_stmt():
+ stmt = HeaderBeginStmt()
+ assert stmt.to_excellon(None) == "M48"
+
+
+def test_header_end_stmt():
+ stmt = HeaderEndStmt()
+ assert stmt.to_excellon(None) == "M95"
+
+
+def test_rewindstop_stmt():
+ stmt = RewindStopStmt()
+ assert stmt.to_excellon(None) == "%"
+
+
+def test_z_axis_rout_position_stmt():
+ stmt = ZAxisRoutPositionStmt()
+ assert stmt.to_excellon(None) == "M15"
+
+
+def test_retract_with_clamping_stmt():
+ stmt = RetractWithClampingStmt()
+ assert stmt.to_excellon(None) == "M16"
+
+
+def test_retract_without_clamping_stmt():
+ stmt = RetractWithoutClampingStmt()
+ assert stmt.to_excellon(None) == "M17"
+
+
+def test_cutter_compensation_off_stmt():
+ stmt = CutterCompensationOffStmt()
+ assert stmt.to_excellon(None) == "G40"
+
+
+def test_cutter_compensation_left_stmt():
+ stmt = CutterCompensationLeftStmt()
+ assert stmt.to_excellon(None) == "G41"
+
+
+def test_cutter_compensation_right_stmt():
+ stmt = CutterCompensationRightStmt()
+ assert stmt.to_excellon(None) == "G42"
+
+
+def test_endofprogramstmt_factory():
+ settings = FileSettings(units="inch")
+ stmt = EndOfProgramStmt.from_excellon("M30X01Y02", settings)
+ assert stmt.x == 1.0
+ assert stmt.y == 2.0
+ assert stmt.units == "inch"
+ settings.units = "metric"
+ stmt = EndOfProgramStmt.from_excellon("M30X01", settings)
+ assert stmt.x == 1.0
+ assert stmt.y == None
+ assert stmt.units == "metric"
+ stmt = EndOfProgramStmt.from_excellon("M30Y02", FileSettings())
+ assert stmt.x == None
+ assert stmt.y == 2.0
+
+
+def test_endofprogramStmt_dump():
+ lines = ["M30X01Y02"]
+ for line in lines:
+ stmt = EndOfProgramStmt.from_excellon(line, FileSettings())
+ assert stmt.to_excellon(FileSettings()) == line
+
+
+def test_endofprogramstmt_conversion():
+ settings = FileSettings()
+ settings.units = "metric"
+ stmt = EndOfProgramStmt.from_excellon("M30X0254Y254", settings)
+ # No effect
+ stmt.to_metric()
+ assert stmt.x == 2.54
+ assert stmt.y == 25.4
+
+ stmt.to_inch()
+ assert stmt.units == "inch"
+ assert stmt.x == 0.1
+ assert stmt.y == 1.0
+
+ # No effect
+ stmt.to_inch()
+ assert stmt.x == 0.1
+ assert stmt.y == 1.0
+
+ settings.units = "inch"
+ stmt = EndOfProgramStmt.from_excellon("M30X01Y1", settings)
+
+ # No effect
+ stmt.to_inch()
+ assert stmt.x == 1.0
+ assert stmt.y == 10.0
+
+ stmt.to_metric()
+ assert stmt.units == "metric"
+ assert stmt.x == 25.4
+ assert stmt.y == 254.0
+
+ # No effect
+ stmt.to_metric()
+ assert stmt.x == 25.4
+ assert stmt.y == 254.0
+
+
+def test_endofprogramstmt_offset():
+ stmt = EndOfProgramStmt(1, 1)
+ stmt.offset()
+ assert stmt.x == 1
+ assert stmt.y == 1
+ stmt.offset(1, 0)
+ assert stmt.x == 2.0
+ assert stmt.y == 1.0
+ stmt.offset(0, 1)
+ assert stmt.x == 2.0
+ assert stmt.y == 2.0
+
+
+def test_unitstmt_factory():
+ """ Test UnitStmt factory method
+ """
+ line = "INCH,LZ"
+ stmt = UnitStmt.from_excellon(line)
+ assert stmt.units == "inch"
+ assert stmt.zeros == "leading"
+
+ line = "INCH,TZ"
+ stmt = UnitStmt.from_excellon(line)
+ assert stmt.units == "inch"
+ assert stmt.zeros == "trailing"
+
+ line = "METRIC,LZ"
+ stmt = UnitStmt.from_excellon(line)
+ assert stmt.units == "metric"
+ assert stmt.zeros == "leading"
+
+ line = "METRIC,TZ"
+ stmt = UnitStmt.from_excellon(line)
+ assert stmt.units == "metric"
+ assert stmt.zeros == "trailing"
+
+
+def test_unitstmt_dump():
+ """ Test UnitStmt to_excellon()
+ """
+ lines = ["INCH,LZ", "INCH,TZ", "METRIC,LZ", "METRIC,TZ"]
+ for line in lines:
+ stmt = UnitStmt.from_excellon(line)
+ assert stmt.to_excellon() == line
+
+
+def test_unitstmt_conversion():
+ stmt = UnitStmt.from_excellon("METRIC,TZ")
+ stmt.to_inch()
+ assert stmt.units == "inch"
+
+ stmt = UnitStmt.from_excellon("INCH,TZ")
+ stmt.to_metric()
+ assert stmt.units == "metric"
+
+
+def test_incrementalmode_factory():
+ """ Test IncrementalModeStmt factory method
+ """
+ line = "ICI,ON"
+ stmt = IncrementalModeStmt.from_excellon(line)
+ assert stmt.mode == "on"
+
+ line = "ICI,OFF"
+ stmt = IncrementalModeStmt.from_excellon(line)
+ assert stmt.mode == "off"
+
+
+def test_incrementalmode_dump():
+ """ Test IncrementalModeStmt to_excellon()
+ """
+ lines = ["ICI,ON", "ICI,OFF"]
+ for line in lines:
+ stmt = IncrementalModeStmt.from_excellon(line)
+ assert stmt.to_excellon() == line
+
+
+def test_incrementalmode_validation():
+ """ Test IncrementalModeStmt input validation
+ """
+ pytest.raises(ValueError, IncrementalModeStmt, "OFF-ISH")
+
+
+def test_versionstmt_factory():
+ """ Test VersionStmt factory method
+ """
+ line = "VER,1"
+ stmt = VersionStmt.from_excellon(line)
+ assert stmt.version == 1
+
+ line = "VER,2"
+ stmt = VersionStmt.from_excellon(line)
+ assert stmt.version == 2
+
+
+def test_versionstmt_dump():
+ """ Test VersionStmt to_excellon()
+ """
+ lines = ["VER,1", "VER,2"]
+ for line in lines:
+ stmt = VersionStmt.from_excellon(line)
+ assert stmt.to_excellon() == line
+
+
+def test_versionstmt_validation():
+ """ Test VersionStmt input validation
+ """
+ pytest.raises(ValueError, VersionStmt, 3)
+
+
+def test_formatstmt_factory():
+ """ Test FormatStmt factory method
+ """
+ line = "FMAT,1"
+ stmt = FormatStmt.from_excellon(line)
+ assert stmt.format == 1
+
+ line = "FMAT,2"
+ stmt = FormatStmt.from_excellon(line)
+ assert stmt.format == 2
+
+
+def test_formatstmt_dump():
+ """ Test FormatStmt to_excellon()
+ """
+ lines = ["FMAT,1", "FMAT,2"]
+ for line in lines:
+ stmt = FormatStmt.from_excellon(line)
+ assert stmt.to_excellon() == line
+
+
+def test_formatstmt_validation():
+ """ Test FormatStmt input validation
+ """
+ pytest.raises(ValueError, FormatStmt, 3)
+
+
+def test_linktoolstmt_factory():
+ """ Test LinkToolStmt factory method
+ """
+ line = "1/2/3/4"
+ stmt = LinkToolStmt.from_excellon(line)
+ assert stmt.linked_tools == [1, 2, 3, 4]
+
+ line = "01/02/03/04"
+ stmt = LinkToolStmt.from_excellon(line)
+ assert stmt.linked_tools == [1, 2, 3, 4]
+
+
+def test_linktoolstmt_dump():
+ """ Test LinkToolStmt to_excellon()
+ """
+ lines = ["1/2/3/4", "5/6/7"]
+ for line in lines:
+ stmt = LinkToolStmt.from_excellon(line)
+ assert stmt.to_excellon() == line
+
+
+def test_measmodestmt_factory():
+ """ Test MeasuringModeStmt factory method
+ """
+ line = "M72"
+ stmt = MeasuringModeStmt.from_excellon(line)
+ assert stmt.units == "inch"
+
+ line = "M71"
+ stmt = MeasuringModeStmt.from_excellon(line)
+ assert stmt.units == "metric"
+
+
+def test_measmodestmt_dump():
+ """ Test MeasuringModeStmt to_excellon()
+ """
+ lines = ["M71", "M72"]
+ for line in lines:
+ stmt = MeasuringModeStmt.from_excellon(line)
+ assert stmt.to_excellon() == line
+
+
+def test_measmodestmt_validation():
+ """ Test MeasuringModeStmt input validation
+ """
+ pytest.raises(ValueError, MeasuringModeStmt.from_excellon, "M70")
+ pytest.raises(ValueError, MeasuringModeStmt, "millimeters")
+
+
+def test_measmodestmt_conversion():
+ line = "M72"
+ stmt = MeasuringModeStmt.from_excellon(line)
+ assert stmt.units == "inch"
+ stmt.to_metric()
+ assert stmt.units == "metric"
+
+ line = "M71"
+ stmt = MeasuringModeStmt.from_excellon(line)
+ assert stmt.units == "metric"
+ stmt.to_inch()
+ assert stmt.units == "inch"
+
+
+def test_routemode_stmt():
+ stmt = RouteModeStmt()
+ assert stmt.to_excellon(FileSettings()) == "G00"
+
+
+def test_linearmode_stmt():
+ stmt = LinearModeStmt()
+ assert stmt.to_excellon(FileSettings()) == "G01"
+
+
+def test_drillmode_stmt():
+ stmt = DrillModeStmt()
+ assert stmt.to_excellon(FileSettings()) == "G05"
+
+
+def test_absolutemode_stmt():
+ stmt = AbsoluteModeStmt()
+ assert stmt.to_excellon(FileSettings()) == "G90"
+
+
+def test_unknownstmt():
+ stmt = UnknownStmt("TEST")
+ assert stmt.stmt == "TEST"
+ assert str(stmt) == "<Unknown Statement: TEST>"
+
+
+def test_unknownstmt_dump():
+ stmt = UnknownStmt("TEST")
+ assert stmt.to_excellon(FileSettings()) == "TEST"
generated by cgit (git 2.34.1) at 2024-07-16 20:23:44 +0000