summaryrefslogtreecommitdiff
path: root/gerber/tests/test_excellon.py
diff options
context:
space:
mode:
authorHamilton Kibbe <hamilton.kibbe@gmail.com>2016-11-05 21:11:09 -0400
committerGitHub <noreply@github.com>2016-11-05 21:11:09 -0400
commitd2fe4441662435e55f2dc481bf94a2729b9d6a48 (patch)
treedd60a0b21e1d1ca7258b9f978ce973354d96062c /gerber/tests/test_excellon.py
parent318a81382e074a5897489299a58e029815d23492 (diff)
parent5af19af190c1fb0f0c5be029d46d63e657dde4d9 (diff)
downloadgerbonara-d2fe4441662435e55f2dc481bf94a2729b9d6a48.tar.gz
gerbonara-d2fe4441662435e55f2dc481bf94a2729b9d6a48.tar.bz2
gerbonara-d2fe4441662435e55f2dc481bf94a2729b9d6a48.zip
Merge pull request #3 from garretfick/merge-curtacircuitos
Merge curtacircuitos
Diffstat (limited to 'gerber/tests/test_excellon.py')
-rw-r--r--gerber/tests/test_excellon.py195
1 files changed, 181 insertions, 14 deletions
diff --git a/gerber/tests/test_excellon.py b/gerber/tests/test_excellon.py
index 72e3d7d..1402938 100644
--- a/gerber/tests/test_excellon.py
+++ b/gerber/tests/test_excellon.py
@@ -2,31 +2,198 @@
# -*- coding: utf-8 -*-
# Author: Hamilton Kibbe <ham@hamiltonkib.be>
-from ..excellon import read, detect_excellon_format, ExcellonFile
-from tests import *
-
import os
+from ..cam import FileSettings
+from ..excellon import read, detect_excellon_format, ExcellonFile, ExcellonParser
+from ..excellon_statements import ExcellonTool
+from .tests import *
+
+
NCDRILL_FILE = os.path.join(os.path.dirname(__file__),
- 'resources/ncdrill.DRD')
+ 'resources/ncdrill.DRD')
+
def test_format_detection():
""" Test file type detection
"""
- settings = detect_excellon_format(NCDRILL_FILE)
+ with open(NCDRILL_FILE, "rU") as f:
+ data = f.read()
+ settings = detect_excellon_format(data)
assert_equal(settings['format'], (2, 4))
- assert_equal(settings['zero_suppression'], 'leading')
+ assert_equal(settings['zeros'], 'trailing')
+
+ settings = detect_excellon_format(filename=NCDRILL_FILE)
+ assert_equal(settings['format'], (2, 4))
+ assert_equal(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_equal(line.strip(), srclines[idx].strip())
+ os.remove('test.ncd')
+
+
def test_read_settings():
ncdrill = read(NCDRILL_FILE)
- assert_equal(ncdrill.settings.format, (2, 4))
- assert_equal(ncdrill.settings.zero_suppression, 'leading')
-
-
-
-
-
+ assert_equal(ncdrill.settings['format'], (2, 4))
+ assert_equal(ncdrill.settings['zeros'], 'trailing')
+
+
+def test_bounds():
+ ncdrill = read(NCDRILL_FILE)
+ xbound, ybound = ncdrill.bounds
+ assert_array_almost_equal(xbound, (0.1300, 2.1430))
+ assert_array_almost_equal(ybound, (0.3946, 1.7164))
+
+
+def test_report():
+ ncdrill = read(NCDRILL_FILE)
+
+
+def test_conversion():
+ import copy
+ ncdrill = read(NCDRILL_FILE)
+ assert_equal(ncdrill.settings.units, 'inch')
+ ncdrill_inch = copy.deepcopy(ncdrill)
+ ncdrill.to_metric()
+ assert_equal(ncdrill.settings.units, 'metric')
+ inch_primitives = ncdrill_inch.primitives
+ for tool in iter(ncdrill_inch.tools.values()):
+ tool.to_metric()
+ for primitive in inch_primitives:
+ primitive.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_equal(i_tool, m_tool)
+
+ for m, i in zip(ncdrill.primitives, inch_primitives):
+ assert_equal(m.position, i.position, '%s not equal to %s' % (m, i))
+ assert_equal(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_equal(p.hole_count, 36)
+
+
+def test_parser_hole_sizes():
+ settings = FileSettings(**detect_excellon_format(NCDRILL_FILE))
+ p = ExcellonParser(settings)
+ p.parse(NCDRILL_FILE)
+ assert_equal(p.hole_sizes, [0.0236, 0.0354, 0.04, 0.126, 0.128])
+
+
+def test_parse_whitespace():
+ p = ExcellonParser(FileSettings())
+ assert_equal(p._parse_line(' '), None)
+
+
+def test_parse_comment():
+ p = ExcellonParser(FileSettings())
+ p._parse_line(';A comment')
+ assert_equal(p.statements[0].comment, 'A comment')
+
+
+def test_parse_format_comment():
+ p = ExcellonParser(FileSettings())
+ p._parse_line('; FILE_FORMAT=9:9 ')
+ assert_equal(p.format, (9, 9))
+
+
+def test_parse_header():
+ p = ExcellonParser(FileSettings())
+ p._parse_line('M48 ')
+ assert_equal(p.state, 'HEADER')
+ p._parse_line('M95 ')
+ assert_equal(p.state, 'DRILL')
+
+
+def test_parse_rout():
+ p = ExcellonParser(FileSettings())
+ p._parse_line('G00 ')
+ assert_equal(p.state, 'ROUT')
+ p._parse_line('G05 ')
+ assert_equal(p.state, 'DRILL')
+
+
+def test_parse_version():
+ p = ExcellonParser(FileSettings())
+ p._parse_line('VER,1 ')
+ assert_equal(p.statements[0].version, 1)
+ p._parse_line('VER,2 ')
+ assert_equal(p.statements[1].version, 2)
+
+
+def test_parse_format():
+ p = ExcellonParser(FileSettings())
+ p._parse_line('FMAT,1 ')
+ assert_equal(p.statements[0].format, 1)
+ p._parse_line('FMAT,2 ')
+ assert_equal(p.statements[1].format, 2)
+
+
+def test_parse_units():
+ settings = FileSettings(units='inch', zeros='trailing')
+ p = ExcellonParser(settings)
+ p._parse_line(';METRIC,LZ')
+ assert_equal(p.units, 'inch')
+ assert_equal(p.zeros, 'trailing')
+ p._parse_line('METRIC,LZ')
+ assert_equal(p.units, 'metric')
+ assert_equal(p.zeros, 'leading')
+
+
+def test_parse_incremental_mode():
+ settings = FileSettings(units='inch', zeros='trailing')
+ p = ExcellonParser(settings)
+ assert_equal(p.notation, 'absolute')
+ p._parse_line('ICI,ON ')
+ assert_equal(p.notation, 'incremental')
+ p._parse_line('ICI,OFF ')
+ assert_equal(p.notation, 'absolute')
+
+
+def test_parse_absolute_mode():
+ settings = FileSettings(units='inch', zeros='trailing')
+ p = ExcellonParser(settings)
+ assert_equal(p.notation, 'absolute')
+ p._parse_line('ICI,ON ')
+ assert_equal(p.notation, 'incremental')
+ p._parse_line('G90 ')
+ assert_equal(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_equal(p.statements[0].count, 3)
+
+
+def test_parse_incremental_position():
+ p = ExcellonParser(FileSettings(notation='incremental'))
+ p._parse_line('X01Y01')
+ p._parse_line('X01Y01')
+ assert_equal(p.pos, [2., 2.])
+
+
+def test_parse_unknown():
+ p = ExcellonParser(FileSettings())
+ p._parse_line('Not A Valid Statement')
+ assert_equal(p.statements[0].stmt, 'Not A Valid Statement')