diff options
Diffstat (limited to 'gerber/tests/test_am_statements.py')
| -rw-r--r-- | gerber/tests/test_am_statements.py | 395 |
1 files changed, 395 insertions, 0 deletions
diff --git a/gerber/tests/test_am_statements.py b/gerber/tests/test_am_statements.py new file mode 100644 index 0000000..0d100b5 --- /dev/null +++ b/gerber/tests/test_am_statements.py @@ -0,0 +1,395 @@ +#! /usr/bin/env python +# -*- coding: utf-8 -*- + +# Author: Hamilton Kibbe <ham@hamiltonkib.be> + +import pytest + +from ..am_statements import * +from ..am_statements import inch, metric + + +def test_AMPrimitive_ctor(): + for exposure in ("on", "off", "ON", "OFF"): + for code in (0, 1, 2, 4, 5, 6, 7, 20, 21, 22): + p = AMPrimitive(code, exposure) + assert p.code == code + assert p.exposure == exposure.lower() + + +def test_AMPrimitive_validation(): + pytest.raises(TypeError, AMPrimitive, "1", "off") + pytest.raises(ValueError, AMPrimitive, 0, "exposed") + pytest.raises(ValueError, AMPrimitive, 3, "off") + + +def test_AMPrimitive_conversion(): + p = AMPrimitive(4, "on") + pytest.raises(NotImplementedError, p.to_inch) + pytest.raises(NotImplementedError, p.to_metric) + + +def test_AMCommentPrimitive_ctor(): + c = AMCommentPrimitive(0, " This is a comment *") + assert c.code == 0 + assert c.comment == "This is a comment" + + +def test_AMCommentPrimitive_validation(): + pytest.raises(ValueError, AMCommentPrimitive, 1, "This is a comment") + + +def test_AMCommentPrimitive_factory(): + c = AMCommentPrimitive.from_gerber("0 Rectangle with rounded corners. *") + assert c.code == 0 + assert c.comment == "Rectangle with rounded corners." + + +def test_AMCommentPrimitive_dump(): + c = AMCommentPrimitive(0, "Rectangle with rounded corners.") + assert c.to_gerber() == "0 Rectangle with rounded corners. *" + + +def test_AMCommentPrimitive_conversion(): + c = AMCommentPrimitive(0, "Rectangle with rounded corners.") + ci = c + cm = c + ci.to_inch() + cm.to_metric() + assert c == ci + assert c == cm + + +def test_AMCommentPrimitive_string(): + c = AMCommentPrimitive(0, "Test Comment") + assert str(c) == "<Aperture Macro Comment: Test Comment>" + + +def test_AMCirclePrimitive_ctor(): + test_cases = ( + (1, "on", 0, (0, 0)), + (1, "off", 1, (0, 1)), + (1, "on", 2.5, (0, 2)), + (1, "off", 5.0, (3, 3)), + ) + for code, exposure, diameter, position in test_cases: + c = AMCirclePrimitive(code, exposure, diameter, position) + assert c.code == code + assert c.exposure == exposure + assert c.diameter == diameter + assert c.position == position + + +def test_AMCirclePrimitive_validation(): + pytest.raises(ValueError, AMCirclePrimitive, 2, "on", 0, (0, 0)) + + +def test_AMCirclePrimitive_factory(): + c = AMCirclePrimitive.from_gerber("1,0,5,0,0*") + assert c.code == 1 + assert c.exposure == "off" + assert c.diameter == 5 + assert c.position == (0, 0) + + +def test_AMCirclePrimitive_dump(): + c = AMCirclePrimitive(1, "off", 5, (0, 0)) + assert c.to_gerber() == "1,0,5,0,0*" + c = AMCirclePrimitive(1, "on", 5, (0, 0)) + assert c.to_gerber() == "1,1,5,0,0*" + + +def test_AMCirclePrimitive_conversion(): + c = AMCirclePrimitive(1, "off", 25.4, (25.4, 0)) + c.to_inch() + assert c.diameter == 1 + assert c.position == (1, 0) + + c = AMCirclePrimitive(1, "off", 1, (1, 0)) + c.to_metric() + assert c.diameter == 25.4 + assert c.position == (25.4, 0) + + +def test_AMVectorLinePrimitive_validation(): + pytest.raises( + ValueError, AMVectorLinePrimitive, 3, "on", 0.1, (0, 0), (3.3, 5.4), 0 + ) + + +def test_AMVectorLinePrimitive_factory(): + l = AMVectorLinePrimitive.from_gerber("20,1,0.9,0,0.45,12,0.45,0*") + assert l.code == 20 + assert l.exposure == "on" + assert l.width == 0.9 + assert l.start == (0, 0.45) + assert l.end == (12, 0.45) + assert l.rotation == 0 + + +def test_AMVectorLinePrimitive_dump(): + l = AMVectorLinePrimitive.from_gerber("20,1,0.9,0,0.45,12,0.45,0*") + assert l.to_gerber() == "20,1,0.9,0.0,0.45,12.0,0.45,0.0*" + + +def test_AMVectorLinePrimtive_conversion(): + l = AMVectorLinePrimitive(20, "on", 25.4, (0, 0), (25.4, 25.4), 0) + l.to_inch() + assert l.width == 1 + assert l.start == (0, 0) + assert l.end == (1, 1) + + l = AMVectorLinePrimitive(20, "on", 1, (0, 0), (1, 1), 0) + l.to_metric() + assert l.width == 25.4 + assert l.start == (0, 0) + assert l.end == (25.4, 25.4) + + +def test_AMOutlinePrimitive_validation(): + pytest.raises( + ValueError, + AMOutlinePrimitive, + 7, + "on", + (0, 0), + [(3.3, 5.4), (4.0, 5.4), (0, 0)], + 0, + ) + pytest.raises( + ValueError, + AMOutlinePrimitive, + 4, + "on", + (0, 0), + [(3.3, 5.4), (4.0, 5.4), (0, 1)], + 0, + ) + + +def test_AMOutlinePrimitive_factory(): + o = AMOutlinePrimitive.from_gerber("4,1,3,0,0,3,3,3,0,0,0,0*") + assert o.code == 4 + assert o.exposure == "on" + assert o.start_point == (0, 0) + assert o.points == [(3, 3), (3, 0), (0, 0)] + assert o.rotation == 0 + + +def test_AMOUtlinePrimitive_dump(): + o = AMOutlinePrimitive(4, "on", (0, 0), [(3, 3), (3, 0), (0, 0)], 0) + # New lines don't matter for Gerber, but we insert them to make it easier to remove + # For test purposes we can ignore them + assert o.to_gerber().replace("\n", "") == "4,1,3,0,0,3,3,3,0,0,0,0*" + + +def test_AMOutlinePrimitive_conversion(): + o = AMOutlinePrimitive(4, "on", (0, 0), [(25.4, 25.4), (25.4, 0), (0, 0)], 0) + o.to_inch() + assert o.start_point == (0, 0) + assert o.points == ((1.0, 1.0), (1.0, 0.0), (0.0, 0.0)) + + o = AMOutlinePrimitive(4, "on", (0, 0), [(1, 1), (1, 0), (0, 0)], 0) + o.to_metric() + assert o.start_point == (0, 0) + assert o.points == ((25.4, 25.4), (25.4, 0), (0, 0)) + + +def test_AMPolygonPrimitive_validation(): + pytest.raises(ValueError, AMPolygonPrimitive, 6, "on", 3, (3.3, 5.4), 3, 0) + pytest.raises(ValueError, AMPolygonPrimitive, 5, "on", 2, (3.3, 5.4), 3, 0) + pytest.raises(ValueError, AMPolygonPrimitive, 5, "on", 13, (3.3, 5.4), 3, 0) + + +def test_AMPolygonPrimitive_factory(): + p = AMPolygonPrimitive.from_gerber("5,1,3,3.3,5.4,3,0") + assert p.code == 5 + assert p.exposure == "on" + assert p.vertices == 3 + assert p.position == (3.3, 5.4) + assert p.diameter == 3 + assert p.rotation == 0 + + +def test_AMPolygonPrimitive_dump(): + p = AMPolygonPrimitive(5, "on", 3, (3.3, 5.4), 3, 0) + assert p.to_gerber() == "5,1,3,3.3,5.4,3,0*" + + +def test_AMPolygonPrimitive_conversion(): + p = AMPolygonPrimitive(5, "off", 3, (25.4, 0), 25.4, 0) + p.to_inch() + assert p.diameter == 1 + assert p.position == (1, 0) + + p = AMPolygonPrimitive(5, "off", 3, (1, 0), 1, 0) + p.to_metric() + assert p.diameter == 25.4 + assert p.position == (25.4, 0) + + +def test_AMMoirePrimitive_validation(): + pytest.raises( + ValueError, AMMoirePrimitive, 7, (0, 0), 5.1, 0.2, 0.4, 6, 0.1, 6.1, 0 + ) + + +def test_AMMoirePrimitive_factory(): + m = AMMoirePrimitive.from_gerber("6,0,0,5,0.5,0.5,2,0.1,6,0*") + assert m.code == 6 + assert m.position == (0, 0) + assert m.diameter == 5 + assert m.ring_thickness == 0.5 + assert m.gap == 0.5 + assert m.max_rings == 2 + assert m.crosshair_thickness == 0.1 + assert m.crosshair_length == 6 + assert m.rotation == 0 + + +def test_AMMoirePrimitive_dump(): + m = AMMoirePrimitive.from_gerber("6,0,0,5,0.5,0.5,2,0.1,6,0*") + assert m.to_gerber() == "6,0,0,5.0,0.5,0.5,2,0.1,6.0,0.0*" + + +def test_AMMoirePrimitive_conversion(): + m = AMMoirePrimitive(6, (25.4, 25.4), 25.4, 25.4, 25.4, 6, 25.4, 25.4, 0) + m.to_inch() + assert m.position == (1.0, 1.0) + assert m.diameter == 1.0 + assert m.ring_thickness == 1.0 + assert m.gap == 1.0 + assert m.crosshair_thickness == 1.0 + assert m.crosshair_length == 1.0 + + m = AMMoirePrimitive(6, (1, 1), 1, 1, 1, 6, 1, 1, 0) + m.to_metric() + assert m.position == (25.4, 25.4) + assert m.diameter == 25.4 + assert m.ring_thickness == 25.4 + assert m.gap == 25.4 + assert m.crosshair_thickness == 25.4 + assert m.crosshair_length == 25.4 + + +def test_AMThermalPrimitive_validation(): + pytest.raises(ValueError, AMThermalPrimitive, 8, (0.0, 0.0), 7, 5, 0.2, 0.0) + pytest.raises(TypeError, AMThermalPrimitive, 7, (0.0, "0"), 7, 5, 0.2, 0.0) + + +def test_AMThermalPrimitive_factory(): + t = AMThermalPrimitive.from_gerber("7,0,0,7,6,0.2,45*") + assert t.code == 7 + assert t.position == (0, 0) + assert t.outer_diameter == 7 + assert t.inner_diameter == 6 + assert t.gap == 0.2 + assert t.rotation == 45 + + +def test_AMThermalPrimitive_dump(): + t = AMThermalPrimitive.from_gerber("7,0,0,7,6,0.2,30*") + assert t.to_gerber() == "7,0,0,7.0,6.0,0.2,30.0*" + + +def test_AMThermalPrimitive_conversion(): + t = AMThermalPrimitive(7, (25.4, 25.4), 25.4, 25.4, 25.4, 0.0) + t.to_inch() + assert t.position == (1.0, 1.0) + assert t.outer_diameter == 1.0 + assert t.inner_diameter == 1.0 + assert t.gap == 1.0 + + t = AMThermalPrimitive(7, (1, 1), 1, 1, 1, 0) + t.to_metric() + assert t.position == (25.4, 25.4) + assert t.outer_diameter == 25.4 + assert t.inner_diameter == 25.4 + assert t.gap == 25.4 + + +def test_AMCenterLinePrimitive_validation(): + pytest.raises(ValueError, AMCenterLinePrimitive, 22, 1, 0.2, 0.5, (0, 0), 0) + + +def test_AMCenterLinePrimtive_factory(): + l = AMCenterLinePrimitive.from_gerber("21,1,6.8,1.2,3.4,0.6,0*") + assert l.code == 21 + assert l.exposure == "on" + assert l.width == 6.8 + assert l.height == 1.2 + assert l.center == (3.4, 0.6) + assert l.rotation == 0 + + +def test_AMCenterLinePrimitive_dump(): + l = AMCenterLinePrimitive.from_gerber("21,1,6.8,1.2,3.4,0.6,0*") + assert l.to_gerber() == "21,1,6.8,1.2,3.4,0.6,0.0*" + + +def test_AMCenterLinePrimitive_conversion(): + l = AMCenterLinePrimitive(21, "on", 25.4, 25.4, (25.4, 25.4), 0) + l.to_inch() + assert l.width == 1.0 + assert l.height == 1.0 + assert l.center == (1.0, 1.0) + + l = AMCenterLinePrimitive(21, "on", 1, 1, (1, 1), 0) + l.to_metric() + assert l.width == 25.4 + assert l.height == 25.4 + assert l.center == (25.4, 25.4) + + +def test_AMLowerLeftLinePrimitive_validation(): + pytest.raises(ValueError, AMLowerLeftLinePrimitive, 23, 1, 0.2, 0.5, (0, 0), 0) + + +def test_AMLowerLeftLinePrimtive_factory(): + l = AMLowerLeftLinePrimitive.from_gerber("22,1,6.8,1.2,3.4,0.6,0*") + assert l.code == 22 + assert l.exposure == "on" + assert l.width == 6.8 + assert l.height == 1.2 + assert l.lower_left == (3.4, 0.6) + assert l.rotation == 0 + + +def test_AMLowerLeftLinePrimitive_dump(): + l = AMLowerLeftLinePrimitive.from_gerber("22,1,6.8,1.2,3.4,0.6,0*") + assert l.to_gerber() == "22,1,6.8,1.2,3.4,0.6,0.0*" + + +def test_AMLowerLeftLinePrimitive_conversion(): + l = AMLowerLeftLinePrimitive(22, "on", 25.4, 25.4, (25.4, 25.4), 0) + l.to_inch() + assert l.width == 1.0 + assert l.height == 1.0 + assert l.lower_left == (1.0, 1.0) + + l = AMLowerLeftLinePrimitive(22, "on", 1, 1, (1, 1), 0) + l.to_metric() + assert l.width == 25.4 + assert l.height == 25.4 + assert l.lower_left == (25.4, 25.4) + + +def test_AMUnsupportPrimitive(): + u = AMUnsupportPrimitive.from_gerber("Test") + assert u.primitive == "Test" + u = AMUnsupportPrimitive("Test") + assert u.to_gerber() == "Test" + + +def test_AMUnsupportPrimitive_smoketest(): + u = AMUnsupportPrimitive.from_gerber("Test") + u.to_inch() + u.to_metric() + + +def test_inch(): + assert inch(25.4) == 1 + + +def test_metric(): + assert metric(1) == 25.4 |