add tests

1 files changed, 21 insertions, 2 deletions

diff --git a/gerber/tests/test_excellon_statements.py b/gerber/tests/test_excellon_statements.py
index 3a10153..49207d3 100644
--- a/gerber/tests/test_excellon_statements.py
+++ b/gerber/tests/test_excellon_statements.py
@@ -20,7 +20,7 @@ def test_ExcellonTool_factory():
 
 
 def test_ExcellonTool_dump():
-    """ Test ExcellonTool to_gerber method
+    """ Test ExcellonTool to_excellon()
     """
     exc_lines = ['T1F00S00C0.01200', 'T2F00S00C0.01500', 'T3F00S00C0.01968',
                  'T4F00S00C0.02800', 'T5F00S00C0.03300', 'T6F00S00C0.03800',
@@ -30,4 +30,23 @@ def test_ExcellonTool_dump():
     for line in exc_lines:
         tool = ExcellonTool.from_excellon(line, settings)
         assert_equal(tool.to_excellon(), line)
-    
\ No newline at end of file
+
+
+def test_ToolSelectionStmt_factory():
+    """ Test ToolSelectionStmt factory method
+    """
+    stmt = ToolSelectionStmt.from_excellon('T01')
+    assert_equal(stmt.tool, 1)
+    assert_equal(stmt.compensation_index, None)
+    stmt = ToolSelectionStmt.from_excellon('T0223')
+    assert_equal(stmt.tool, 2)
+    assert_equal(stmt.compensation_index, 23)
+
+
+def test_ToolSelectionStmt_dump():
+    """ Test ToolSelectionStmt to_excellon()
+    """
+    lines = ['T01', 'T0223', 'T10', 'T09', 'T0000']
+    for line in lines:
+        stmt = ToolSelectionStmt.from_excellon(line)
+        assert_equal(stmt.to_excellon(), line)