dd63b169f1
Adds a loads() method to the top level module which generates a GerberFile or ExcellonFile from a string
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
#! /usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# Author: Hamilton Kibbe <ham@hamiltonkib.be>
|
|
from ..common import read, loads
|
|
from ..excellon import ExcellonFile
|
|
from ..rs274x import GerberFile
|
|
from .tests import *
|
|
|
|
import os
|
|
|
|
|
|
NCDRILL_FILE = os.path.join(os.path.dirname(__file__),
|
|
'resources/ncdrill.DRD')
|
|
TOP_COPPER_FILE = os.path.join(os.path.dirname(__file__),
|
|
'resources/top_copper.GTL')
|
|
|
|
def test_file_type_detection():
|
|
""" Test file type detection
|
|
"""
|
|
ncdrill = read(NCDRILL_FILE)
|
|
top_copper = read(TOP_COPPER_FILE)
|
|
assert_true(isinstance(ncdrill, ExcellonFile))
|
|
assert_true(isinstance(top_copper, GerberFile))
|
|
|
|
|
|
def test_load_from_string():
|
|
with open(NCDRILL_FILE, 'r') as f:
|
|
ncdrill = loads(f.read())
|
|
with open(TOP_COPPER_FILE, 'r') as f:
|
|
top_copper = loads(f.read())
|
|
assert_true(isinstance(ncdrill, ExcellonFile))
|
|
assert_true(isinstance(top_copper, GerberFile))
|
|
|
|
|
|
def test_file_type_validation():
|
|
""" Test file format validation
|
|
"""
|
|
assert_raises(TypeError, read, 'LICENSE')
|
|
|
|
|
|
|