1 files changed, 83 insertions, 0 deletions
diff --git a/gerber/tests/test_utils.py b/gerber/tests/test_utils.py
new file mode 100644
index 0000000..001a32f
--- /dev/null
+++ b/gerber/tests/test_utils.py
@@ -0,0 +1,83 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# Author: Hamilton Kibbe <ham@hamiltonkib.be>
+
+from .tests import assert_equal
+from ..utils import decimal_string, parse_gerber_value, write_gerber_value
+
+
+def test_zero_suppression():
+    """ Test gerber value parser and writer handle zero suppression correctly.
+    """
+    # Default format
+    fmt = (2, 5)
+
+    # Test leading zero suppression
+    zero_suppression = 'leading'
+    test_cases = [('1', 0.00001), ('10', 0.0001), ('100', 0.001),
+                  ('1000', 0.01), ('10000', 0.1), ('100000', 1.0),
+                  ('1000000', 10.0), ('-1', -0.00001), ('-10', -0.0001),
+                  ('-100', -0.001), ('-1000', -0.01), ('-10000', -0.1),
+                  ('-100000', -1.0), ('-1000000', -10.0), ]
+    for string, value in test_cases:
+        assert(value == parse_gerber_value(string, fmt, zero_suppression))
+        assert(string == write_gerber_value(value, fmt, zero_suppression))
+
+    # Test trailing zero suppression
+    zero_suppression = 'trailing'
+    test_cases = [('1', 10.0), ('01', 1.0), ('001', 0.1), ('0001', 0.01),
+                  ('00001', 0.001), ('000001', 0.0001),
+                  ('0000001', 0.00001), ('-1', -10.0), ('-01', -1.0),
+                  ('-001', -0.1), ('-0001', -0.01), ('-00001', -0.001),
+                  ('-000001', -0.0001), ('-0000001', -0.00001)]
+    for string, value in test_cases:
+        assert(value == parse_gerber_value(string, fmt, zero_suppression))
+        assert(string == write_gerber_value(value, fmt, zero_suppression))
+
+
+def test_format():
+    """ Test gerber value parser and writer handle format correctly
+    """
+    zero_suppression = 'leading'
+    test_cases = [((2, 7), '1', 0.0000001), ((2, 6), '1', 0.000001),
+                  ((2, 5), '1', 0.00001), ((2, 4), '1', 0.0001),
+                  ((2, 3), '1', 0.001), ((2, 2), '1', 0.01),
+                  ((2, 1), '1', 0.1), ((2, 7), '-1', -0.0000001),
+                  ((2, 6), '-1', -0.000001), ((2, 5), '-1', -0.00001),
+                  ((2, 4), '-1', -0.0001), ((2, 3), '-1', -0.001),
+                  ((2, 2), '-1', -0.01), ((2, 1), '-1', -0.1), ]
+    for fmt, string, value in test_cases:
+        assert(value == parse_gerber_value(string, fmt, zero_suppression))
+        assert(string == write_gerber_value(value, fmt, zero_suppression))
+
+    zero_suppression = 'trailing'
+    test_cases = [((6, 5), '1', 100000.0), ((5, 5), '1', 10000.0),
+                  ((4, 5), '1', 1000.0), ((3, 5), '1', 100.0),
+                  ((2, 5), '1', 10.0), ((1, 5), '1', 1.0),
+                  ((6, 5), '-1', -100000.0), ((5, 5), '-1', -10000.0),
+                  ((4, 5), '-1', -1000.0), ((3, 5), '-1', -100.0),
+                  ((2, 5), '-1', -10.0), ((1, 5), '-1', -1.0), ]
+    for fmt, string, value in test_cases:
+        assert(value == parse_gerber_value(string, fmt, zero_suppression))
+        assert(string == write_gerber_value(value, fmt, zero_suppression))
+
+
+def test_decimal_truncation():
+    """ Test decimal_string truncates value to the correct precision
+    """
+    value = 1.123456789
+    for x in range(10):
+        result = decimal_string(value, precision=x)
+        calculated = '1.' + ''.join(str(y) for y in range(1,x+1))
+        assert(result == calculated)
+
+
+def test_decimal_padding():
+    """ Test decimal_string padding
+    """
+    value = 1.123
+    assert_equal(decimal_string(value, precision=3, padding=True), '1.123')
+    assert_equal(decimal_string(value, precision=4, padding=True), '1.1230')
+    assert_equal(decimal_string(value, precision=5, padding=True), '1.12300')
+    assert_equal(decimal_string(value, precision=6, padding=True), '1.123000')