summaryrefslogtreecommitdiff
path: root/gerber/tests/test_primitives.py
diff options
context:
space:
mode:
authorHamilton Kibbe <hamilton.kibbe@gmail.com>2016-01-21 03:57:44 -0500
committerGarret Fick <garret@ficksworkshop.com>2016-08-06 09:40:40 +0800
commit8cd842a41a55ab3d8f558a2e3e198beba7da58a1 (patch)
tree895d818072e043ac5275ed6cdc1d724ea5cd489a /gerber/tests/test_primitives.py
parent965d3ce23b92f8aff1063debd6d3364de15791fe (diff)
downloadgerbonara-8cd842a41a55ab3d8f558a2e3e198beba7da58a1.tar.gz
gerbonara-8cd842a41a55ab3d8f558a2e3e198beba7da58a1.tar.bz2
gerbonara-8cd842a41a55ab3d8f558a2e3e198beba7da58a1.zip
Manually mere rendering changes
Diffstat (limited to 'gerber/tests/test_primitives.py')
-rw-r--r--gerber/tests/test_primitives.py416
1 files changed, 249 insertions, 167 deletions
diff --git a/gerber/tests/test_primitives.py b/gerber/tests/test_primitives.py
index 61cf22d..261e6ef 100644
--- a/gerber/tests/test_primitives.py
+++ b/gerber/tests/test_primitives.py
@@ -2,18 +2,23 @@
# -*- coding: utf-8 -*-
# Author: Hamilton Kibbe <ham@hamiltonkib.be>
+from operator import add
+
from ..primitives import *
from .tests import *
-from operator import add
def test_primitive_smoketest():
p = Primitive()
+<<<<<<< HEAD
try:
p.bounding_box
assert_false(True, 'should have thrown the exception')
except NotImplementedError:
pass
+=======
+ #assert_raises(NotImplementedError, p.bounding_box)
+>>>>>>> 5476da8... Fix a bunch of rendering bugs.
p.to_metric()
p.to_inch()
try:
@@ -22,6 +27,7 @@ def test_primitive_smoketest():
except NotImplementedError:
pass
+
def test_line_angle():
""" Test Line primitive angle calculation
"""
@@ -32,19 +38,20 @@ def test_line_angle():
((0, 0), (-1, 0), math.radians(180)),
((0, 0), (-1, -1), math.radians(225)),
((0, 0), (0, -1), math.radians(270)),
- ((0, 0), (1, -1), math.radians(315)),]
+ ((0, 0), (1, -1), math.radians(315)), ]
for start, end, expected in cases:
l = Line(start, end, 0)
line_angle = (l.angle + 2 * math.pi) % (2 * math.pi)
assert_almost_equal(line_angle, expected)
+
def test_line_bounds():
""" Test Line primitive bounding box calculation
"""
cases = [((0, 0), (1, 1), ((-1, 2), (-1, 2))),
((-1, -1), (1, 1), ((-2, 2), (-2, 2))),
((1, 1), (-1, -1), ((-2, 2), (-2, 2))),
- ((-1, 1), (1, -1), ((-2, 2), (-2, 2))),]
+ ((-1, 1), (1, -1), ((-2, 2), (-2, 2))), ]
c = Circle((0, 0), 2)
r = Rectangle((0, 0), 2, 2)
@@ -57,11 +64,12 @@ def test_line_bounds():
cases = [((0, 0), (1, 1), ((-1.5, 2.5), (-1, 2))),
((-1, -1), (1, 1), ((-2.5, 2.5), (-2, 2))),
((1, 1), (-1, -1), ((-2.5, 2.5), (-2, 2))),
- ((-1, 1), (1, -1), ((-2.5, 2.5), (-2, 2))),]
+ ((-1, 1), (1, -1), ((-2.5, 2.5), (-2, 2))), ]
for start, end, expected in cases:
l = Line(start, end, r)
assert_equal(l.bounding_box, expected)
+
def test_line_vertices():
c = Circle((0, 0), 2)
l = Line((0, 0), (1, 1), c)
@@ -69,20 +77,25 @@ def test_line_vertices():
# All 4 compass points, all 4 quadrants and the case where start == end
test_cases = [((0, 0), (1, 0), ((-1, -1), (-1, 1), (2, 1), (2, -1))),
- ((0, 0), (1, 1), ((-1, -1), (-1, 1), (0, 2), (2, 2), (2, 0), (1,-1))),
- ((0, 0), (0, 1), ((-1, -1), (-1, 2), (1, 2), (1, -1))),
- ((0, 0), (-1, 1), ((-1, -1), (-2, 0), (-2, 2), (0, 2), (1, 1), (1, -1))),
- ((0, 0), (-1, 0), ((-2, -1), (-2, 1), (1, 1), (1, -1))),
- ((0, 0), (-1, -1), ((-2, -2), (1, -1), (1, 1), (-1, 1), (-2, 0), (0,-2))),
- ((0, 0), (0, -1), ((-1, -2), (-1, 1), (1, 1), (1, -2))),
- ((0, 0), (1, -1), ((-1, -1), (0, -2), (2, -2), (2, 0), (1, 1), (-1, 1))),
- ((0, 0), (0, 0), ((-1, -1), (-1, 1), (1, 1), (1, -1))),]
+ ((0, 0), (1, 1), ((-1, -1), (-1, 1),
+ (0, 2), (2, 2), (2, 0), (1, -1))),
+ ((0, 0), (0, 1), ((-1, -1), (-1, 2), (1, 2), (1, -1))),
+ ((0, 0), (-1, 1), ((-1, -1), (-2, 0),
+ (-2, 2), (0, 2), (1, 1), (1, -1))),
+ ((0, 0), (-1, 0), ((-2, -1), (-2, 1), (1, 1), (1, -1))),
+ ((0, 0), (-1, -1), ((-2, -2), (1, -1),
+ (1, 1), (-1, 1), (-2, 0), (0, -2))),
+ ((0, 0), (0, -1), ((-1, -2), (-1, 1), (1, 1), (1, -2))),
+ ((0, 0), (1, -1), ((-1, -1), (0, -2),
+ (2, -2), (2, 0), (1, 1), (-1, 1))),
+ ((0, 0), (0, 0), ((-1, -1), (-1, 1), (1, 1), (1, -1))), ]
r = Rectangle((0, 0), 2, 2)
for start, end, vertices in test_cases:
l = Line(start, end, r)
assert_equal(set(vertices), set(l.vertices))
+
def test_line_conversion():
c = Circle((0, 0), 25.4, units='metric')
l = Line((2.54, 25.4), (254.0, 2540.0), c, units='metric')
@@ -113,13 +126,12 @@ def test_line_conversion():
assert_equal(l.end, (10.0, 100.0))
assert_equal(l.aperture.diameter, 1.0)
-
l.to_metric()
assert_equal(l.start, (2.54, 25.4))
assert_equal(l.end, (254.0, 2540.0))
assert_equal(l.aperture.diameter, 25.4)
- #No effect
+ # No effect
l.to_metric()
assert_equal(l.start, (2.54, 25.4))
assert_equal(l.end, (254.0, 2540.0))
@@ -141,56 +153,62 @@ def test_line_conversion():
assert_equal(l.aperture.width, 25.4)
assert_equal(l.aperture.height, 254.0)
+
def test_line_offset():
c = Circle((0, 0), 1)
l = Line((0, 0), (1, 1), c)
l.offset(1, 0)
- assert_equal(l.start,(1., 0.))
+ assert_equal(l.start, (1., 0.))
assert_equal(l.end, (2., 1.))
l.offset(0, 1)
- assert_equal(l.start,(1., 1.))
+ assert_equal(l.start, (1., 1.))
assert_equal(l.end, (2., 2.))
+
def test_arc_radius():
""" Test Arc primitive radius calculation
"""
cases = [((-3, 4), (5, 0), (0, 0), 5),
- ((0, 1), (1, 0), (0, 0), 1),]
+ ((0, 1), (1, 0), (0, 0), 1), ]
for start, end, center, radius in cases:
a = Arc(start, end, center, 'clockwise', 0, 'single-quadrant')
assert_equal(a.radius, radius)
+
def test_arc_sweep_angle():
""" Test Arc primitive sweep angle calculation
"""
cases = [((1, 0), (0, 1), (0, 0), 'counterclockwise', math.radians(90)),
((1, 0), (0, 1), (0, 0), 'clockwise', math.radians(270)),
((1, 0), (-1, 0), (0, 0), 'clockwise', math.radians(180)),
- ((1, 0), (-1, 0), (0, 0), 'counterclockwise', math.radians(180)),]
+ ((1, 0), (-1, 0), (0, 0), 'counterclockwise', math.radians(180)), ]
for start, end, center, direction, sweep in cases:
c = Circle((0,0), 1)
a = Arc(start, end, center, direction, c, 'single-quadrant')
assert_equal(a.sweep_angle, sweep)
+
def test_arc_bounds():
""" Test Arc primitive bounding box calculation
"""
- cases = [((1, 0), (0, 1), (0, 0), 'clockwise', ((-1.5, 1.5), (-1.5, 1.5))),
- ((1, 0), (0, 1), (0, 0), 'counterclockwise', ((-0.5, 1.5), (-0.5, 1.5))),
- #TODO: ADD MORE TEST CASES HERE
+ cases = [((1, 0), (0, 1), (0, 0), 'clockwise', ((-1.5, 1.5), (-1.5, 1.5))),
+ ((1, 0), (0, 1), (0, 0), 'counterclockwise',
+ ((-0.5, 1.5), (-0.5, 1.5))),
+ # TODO: ADD MORE TEST CASES HERE
]
for start, end, center, direction, bounds in cases:
c = Circle((0,0), 1)
a = Arc(start, end, center, direction, c, 'single-quadrant')
assert_equal(a.bounding_box, bounds)
+
def test_arc_conversion():
c = Circle((0, 0), 25.4, units='metric')
a = Arc((2.54, 25.4), (254.0, 2540.0), (25400.0, 254000.0),'clockwise', c, 'single-quadrant', units='metric')
- #No effect
+ # No effect
a.to_metric()
assert_equal(a.start, (2.54, 25.4))
assert_equal(a.end, (254.0, 2540.0))
@@ -203,7 +221,7 @@ def test_arc_conversion():
assert_equal(a.center, (1000.0, 10000.0))
assert_equal(a.aperture.diameter, 1.0)
- #no effect
+ # no effect
a.to_inch()
assert_equal(a.start, (0.1, 1.0))
assert_equal(a.end, (10.0, 100.0))
@@ -218,18 +236,20 @@ def test_arc_conversion():
assert_equal(a.center, (25400.0, 254000.0))
assert_equal(a.aperture.diameter, 25.4)
+
def test_arc_offset():
c = Circle((0, 0), 1)
a = Arc((0, 0), (1, 1), (2, 2), 'clockwise', c, 'single-quadrant')
a.offset(1, 0)
- assert_equal(a.start,(1., 0.))
+ assert_equal(a.start, (1., 0.))
assert_equal(a.end, (2., 1.))
assert_equal(a.center, (3., 2.))
a.offset(0, 1)
- assert_equal(a.start,(1., 1.))
+ assert_equal(a.start, (1., 1.))
assert_equal(a.end, (2., 2.))
assert_equal(a.center, (3., 3.))
+
def test_circle_radius():
""" Test Circle primitive radius calculation
"""
@@ -248,12 +268,13 @@ def test_circle_bounds():
c = Circle((1, 1), 2)
assert_equal(c.bounding_box, ((0, 2), (0, 2)))
+
def test_circle_conversion():
"""Circle conversion of units"""
# Circle initially metric, no hole
c = Circle((2.54, 25.4), 254.0, units='metric')
- c.to_metric() #shouldn't do antyhing
+ c.to_metric() # shouldn't do antyhing
assert_equal(c.position, (2.54, 25.4))
assert_equal(c.diameter, 254.)
assert_equal(c.hole_diameter, None)
@@ -263,7 +284,7 @@ def test_circle_conversion():
assert_equal(c.diameter, 10.)
assert_equal(c.hole_diameter, None)
- #no effect
+ # no effect
c.to_inch()
assert_equal(c.position, (0.1, 1.))
assert_equal(c.diameter, 10.)
@@ -290,7 +311,7 @@ def test_circle_conversion():
# Circle initially inch, no hole
c = Circle((0.1, 1.0), 10.0, units='inch')
- #No effect
+ # No effect
c.to_inch()
assert_equal(c.position, (0.1, 1.))
assert_equal(c.diameter, 10.)
@@ -301,7 +322,7 @@ def test_circle_conversion():
assert_equal(c.diameter, 254.)
assert_equal(c.hole_diameter, None)
- #no effect
+ # no effect
c.to_metric()
assert_equal(c.position, (2.54, 25.4))
assert_equal(c.diameter, 254.)
@@ -325,12 +346,14 @@ def test_circle_conversion():
assert_equal(c.diameter, 254.)
assert_equal(c.hole_diameter, 127.)
+
def test_circle_offset():
c = Circle((0, 0), 1)
c.offset(1, 0)
- assert_equal(c.position,(1., 0.))
+ assert_equal(c.position, (1., 0.))
c.offset(0, 1)
- assert_equal(c.position,(1., 1.))
+ assert_equal(c.position, (1., 1.))
+
def test_ellipse_ctor():
""" Test ellipse creation
@@ -340,6 +363,7 @@ def test_ellipse_ctor():
assert_equal(e.width, 3)
assert_equal(e.height, 2)
+
def test_ellipse_bounds():
""" Test ellipse bounding box calculation
"""
@@ -352,10 +376,11 @@ def test_ellipse_bounds():
e = Ellipse((2, 2), 4, 2, rotation=270)
assert_equal(e.bounding_box, ((1, 3), (0, 4)))
+
def test_ellipse_conversion():
e = Ellipse((2.54, 25.4), 254.0, 2540., units='metric')
- #No effect
+ # No effect
e.to_metric()
assert_equal(e.position, (2.54, 25.4))
assert_equal(e.width, 254.)
@@ -366,7 +391,7 @@ def test_ellipse_conversion():
assert_equal(e.width, 10.)
assert_equal(e.height, 100.)
- #No effect
+ # No effect
e.to_inch()
assert_equal(e.position, (0.1, 1.))
assert_equal(e.width, 10.)
@@ -374,7 +399,7 @@ def test_ellipse_conversion():
e = Ellipse((0.1, 1.), 10.0, 100., units='inch')
- #no effect
+ # no effect
e.to_inch()
assert_equal(e.position, (0.1, 1.))
assert_equal(e.width, 10.)
@@ -391,17 +416,19 @@ def test_ellipse_conversion():
assert_equal(e.width, 254.)
assert_equal(e.height, 2540.)
+
def test_ellipse_offset():
e = Ellipse((0, 0), 1, 2)
e.offset(1, 0)
- assert_equal(e.position,(1., 0.))
+ assert_equal(e.position, (1., 0.))
e.offset(0, 1)
- assert_equal(e.position,(1., 1.))
+ assert_equal(e.position, (1., 1.))
+
def test_rectangle_ctor():
""" Test rectangle creation
"""
- test_cases = (((0,0), 1, 1), ((0, 0), 1, 2), ((1,1), 1, 2))
+ test_cases = (((0, 0), 1, 1), ((0, 0), 1, 2), ((1, 1), 1, 2))
for pos, width, height in test_cases:
r = Rectangle(pos, width, height)
assert_equal(r.position, pos)
@@ -417,18 +444,20 @@ def test_rectangle_hole_radius():
r = Rectangle((0,0), 2, 2, 1)
assert_equal(0.5, r.hole_radius)
+
def test_rectangle_bounds():
""" Test rectangle bounding box calculation
"""
- r = Rectangle((0,0), 2, 2)
+ r = Rectangle((0, 0), 2, 2)
xbounds, ybounds = r.bounding_box
assert_array_almost_equal(xbounds, (-1, 1))
assert_array_almost_equal(ybounds, (-1, 1))
- r = Rectangle((0,0), 2, 2, rotation=45)
+ r = Rectangle((0, 0), 2, 2, rotation=45)
xbounds, ybounds = r.bounding_box
assert_array_almost_equal(xbounds, (-math.sqrt(2), math.sqrt(2)))
assert_array_almost_equal(ybounds, (-math.sqrt(2), math.sqrt(2)))
+
def test_rectangle_conversion():
"""Test converting rectangles between units"""
@@ -436,7 +465,7 @@ def test_rectangle_conversion():
r = Rectangle((2.54, 25.4), 254.0, 2540.0, units='metric')
r.to_metric()
- assert_equal(r.position, (2.54,25.4))
+ assert_equal(r.position, (2.54, 25.4))
assert_equal(r.width, 254.0)
assert_equal(r.height, 2540.0)
@@ -479,12 +508,12 @@ def test_rectangle_conversion():
assert_equal(r.height, 100.0)
r.to_metric()
- assert_equal(r.position, (2.54,25.4))
+ assert_equal(r.position, (2.54, 25.4))
assert_equal(r.width, 254.0)
assert_equal(r.height, 2540.0)
r.to_metric()
- assert_equal(r.position, (2.54,25.4))
+ assert_equal(r.position, (2.54, 25.4))
assert_equal(r.width, 254.0)
assert_equal(r.height, 2540.0)
@@ -508,35 +537,39 @@ def test_rectangle_conversion():
assert_equal(r.height, 2540.0)
assert_equal(r.hole_diameter, 127.0)
+
def test_rectangle_offset():
r = Rectangle((0, 0), 1, 2)
r.offset(1, 0)
- assert_equal(r.position,(1., 0.))
+ assert_equal(r.position, (1., 0.))
r.offset(0, 1)
- assert_equal(r.position,(1., 1.))
+ assert_equal(r.position, (1., 1.))
+
def test_diamond_ctor():
""" Test diamond creation
"""
- test_cases = (((0,0), 1, 1), ((0, 0), 1, 2), ((1,1), 1, 2))
+ test_cases = (((0, 0), 1, 1), ((0, 0), 1, 2), ((1, 1), 1, 2))
for pos, width, height in test_cases:
d = Diamond(pos, width, height)
assert_equal(d.position, pos)
assert_equal(d.width, width)
assert_equal(d.height, height)
+
def test_diamond_bounds():
""" Test diamond bounding box calculation
"""
- d = Diamond((0,0), 2, 2)
+ d = Diamond((0, 0), 2, 2)
xbounds, ybounds = d.bounding_box
assert_array_almost_equal(xbounds, (-1, 1))
assert_array_almost_equal(ybounds, (-1, 1))
- d = Diamond((0,0), math.sqrt(2), math.sqrt(2), rotation=45)
+ d = Diamond((0, 0), math.sqrt(2), math.sqrt(2), rotation=45)
xbounds, ybounds = d.bounding_box
assert_array_almost_equal(xbounds, (-1, 1))
assert_array_almost_equal(ybounds, (-1, 1))
+
def test_diamond_conversion():
d = Diamond((2.54, 25.4), 254.0, 2540.0, units='metric')
@@ -572,19 +605,21 @@ def test_diamond_conversion():
assert_equal(d.width, 254.0)
assert_equal(d.height, 2540.0)
+
def test_diamond_offset():
d = Diamond((0, 0), 1, 2)
d.offset(1, 0)
- assert_equal(d.position,(1., 0.))
+ assert_equal(d.position, (1., 0.))
d.offset(0, 1)
- assert_equal(d.position,(1., 1.))
+ assert_equal(d.position, (1., 1.))
+
def test_chamfer_rectangle_ctor():
""" Test chamfer rectangle creation
"""
- test_cases = (((0,0), 1, 1, 0.2, (True, True, False, False)),
+ test_cases = (((0, 0), 1, 1, 0.2, (True, True, False, False)),
((0, 0), 1, 2, 0.3, (True, True, True, True)),
- ((1,1), 1, 2, 0.4, (False, False, False, False)))
+ ((1, 1), 1, 2, 0.4, (False, False, False, False)))
for pos, width, height, chamfer, corners in test_cases:
r = ChamferRectangle(pos, width, height, chamfer, corners)
assert_equal(r.position, pos)
@@ -593,23 +628,27 @@ def test_chamfer_rectangle_ctor():
assert_equal(r.chamfer, chamfer)
assert_array_almost_equal(r.corners, corners)
+
def test_chamfer_rectangle_bounds():
""" Test chamfer rectangle bounding box calculation
"""
- r = ChamferRectangle((0,0), 2, 2, 0.2, (True, True, False, False))
+ r = ChamferRectangle((0, 0), 2, 2, 0.2, (True, True, False, False))
xbounds, ybounds = r.bounding_box
assert_array_almost_equal(xbounds, (-1, 1))
assert_array_almost_equal(ybounds, (-1, 1))
- r = ChamferRectangle((0,0), 2, 2, 0.2, (True, True, False, False), rotation=45)
+ r = ChamferRectangle(
+ (0, 0), 2, 2, 0.2, (True, True, False, False), rotation=45)
xbounds, ybounds = r.bounding_box
assert_array_almost_equal(xbounds, (-math.sqrt(2), math.sqrt(2)))
assert_array_almost_equal(ybounds, (-math.sqrt(2), math.sqrt(2)))
+
def test_chamfer_rectangle_conversion():
- r = ChamferRectangle((2.54, 25.4), 254.0, 2540.0, 0.254, (True, True, False, False), units='metric')
+ r = ChamferRectangle((2.54, 25.4), 254.0, 2540.0, 0.254,
+ (True, True, False, False), units='metric')
r.to_metric()
- assert_equal(r.position, (2.54,25.4))
+ assert_equal(r.position, (2.54, 25.4))
assert_equal(r.width, 254.0)
assert_equal(r.height, 2540.0)
assert_equal(r.chamfer, 0.254)
@@ -626,7 +665,8 @@ def test_chamfer_rectangle_conversion():
assert_equal(r.height, 100.0)
assert_equal(r.chamfer, 0.01)
- r = ChamferRectangle((0.1, 1.0), 10.0, 100.0, 0.01, (True, True, False, False), units='inch')
+ r = ChamferRectangle((0.1, 1.0), 10.0, 100.0, 0.01,
+ (True, True, False, False), units='inch')
r.to_inch()
assert_equal(r.position, (0.1, 1.0))
assert_equal(r.width, 10.0)
@@ -634,30 +674,32 @@ def test_chamfer_rectangle_conversion():
assert_equal(r.chamfer, 0.01)
r.to_metric()
- assert_equal(r.position, (2.54,25.4))
+ assert_equal(r.position, (2.54, 25.4))
assert_equal(r.width, 254.0)
assert_equal(r.height, 2540.0)
assert_equal(r.chamfer, 0.254)
r.to_metric()
- assert_equal(r.position, (2.54,25.4))
+ assert_equal(r.position, (2.54, 25.4))
assert_equal(r.width, 254.0)
assert_equal(r.height, 2540.0)
assert_equal(r.chamfer, 0.254)
+
def test_chamfer_rectangle_offset():
r = ChamferRectangle((0, 0), 1, 2, 0.01, (True, True, False, False))
r.offset(1, 0)
- assert_equal(r.position,(1., 0.))
+ assert_equal(r.position, (1., 0.))
r.offset(0, 1)
- assert_equal(r.position,(1., 1.))
+ assert_equal(r.position, (1., 1.))
+
def test_round_rectangle_ctor():
""" Test round rectangle creation
"""
- test_cases = (((0,0), 1, 1, 0.2, (True, True, False, False)),
+ test_cases = (((0, 0), 1, 1, 0.2, (True, True, False, False)),
((0, 0), 1, 2, 0.3, (True, True, True, True)),
- ((1,1), 1, 2, 0.4, (False, False, False, False)))
+ ((1, 1), 1, 2, 0.4, (False, False, False, False)))
for pos, width, height, radius, corners in test_cases:
r = RoundRectangle(pos, width, height, radius, corners)
assert_equal(r.position, pos)
@@ -666,23 +708,27 @@ def test_round_rectangle_ctor():
assert_equal(r.radius, radius)
assert_array_almost_equal(r.corners, corners)
+
def test_round_rectangle_bounds():
""" Test round rectangle bounding box calculation
"""
- r = RoundRectangle((0,0), 2, 2, 0.2, (True, True, False, False))
+ r = RoundRectangle((0, 0), 2, 2, 0.2, (True, True, False, False))
xbounds, ybounds = r.bounding_box
assert_array_almost_equal(xbounds, (-1, 1))
assert_array_almost_equal(ybounds, (-1, 1))
- r = RoundRectangle((0,0), 2, 2, 0.2, (True, True, False, False), rotation=45)
+ r = RoundRectangle((0, 0), 2, 2, 0.2,
+ (True, True, False, False), rotation=45)
xbounds, ybounds = r.bounding_box
assert_array_almost_equal(xbounds, (-math.sqrt(2), math.sqrt(2)))
assert_array_almost_equal(ybounds, (-math.sqrt(2), math.sqrt(2)))
+
def test_round_rectangle_conversion():
- r = RoundRectangle((2.54, 25.4), 254.0, 2540.0, 0.254, (True, True, False, False), units='metric')
+ r = RoundRectangle((2.54, 25.4), 254.0, 2540.0, 0.254,
+ (True, True, False, False), units='metric')
r.to_metric()
- assert_equal(r.position, (2.54,25.4))
+ assert_equal(r.position, (2.54, 25.4))
assert_equal(r.width, 254.0)
assert_equal(r.height, 2540.0)
assert_equal(r.radius, 0.254)
@@ -699,7 +745,8 @@ def test_round_rectangle_conversion():
assert_equal(r.height, 100.0)
assert_equal(r.radius, 0.01)
- r = RoundRectangle((0.1, 1.0), 10.0, 100.0, 0.01, (True, True, False, False), units='inch')
+ r = RoundRectangle((0.1, 1.0), 10.0, 100.0, 0.01,
+ (True, True, False, False), units='inch')
r.to_inch()
assert_equal(r.position, (0.1, 1.0))
@@ -708,70 +755,76 @@ def test_round_rectangle_conversion():
assert_equal(r.radius, 0.01)
r.to_metric()
- assert_equal(r.position, (2.54,25.4))
+ assert_equal(r.position, (2.54, 25.4))
assert_equal(r.width, 254.0)
assert_equal(r.height, 2540.0)
assert_equal(r.radius, 0.254)
r.to_metric()
- assert_equal(r.position, (2.54,25.4))
+ assert_equal(r.position, (2.54, 25.4))
assert_equal(r.width, 254.0)
assert_equal(r.height, 2540.0)
assert_equal(r.radius, 0.254)
+
def test_round_rectangle_offset():
r = RoundRectangle((0, 0), 1, 2, 0.01, (True, True, False, False))
r.offset(1, 0)
- assert_equal(r.position,(1., 0.))
+ assert_equal(r.position, (1., 0.))
r.offset(0, 1)
- assert_equal(r.position,(1., 1.))
+ assert_equal(r.position, (1., 1.))
+
def test_obround_ctor():
""" Test obround creation
"""
- test_cases = (((0,0), 1, 1),
+ test_cases = (((0, 0), 1, 1),
((0, 0), 1, 2),
- ((1,1), 1, 2))
+ ((1, 1), 1, 2))
for pos, width, height in test_cases:
o = Obround(pos, width, height)
assert_equal(o.position, pos)
assert_equal(o.width, width)
assert_equal(o.height, height)
+
def test_obround_bounds():
""" Test obround bounding box calculation
"""
- o = Obround((2,2),2,4)
+ o = Obround((2, 2), 2, 4)
xbounds, ybounds = o.bounding_box
assert_array_almost_equal(xbounds, (1, 3))
assert_array_almost_equal(ybounds, (0, 4))
- o = Obround((2,2),4,2)
+ o = Obround((2, 2), 4, 2)
xbounds, ybounds = o.bounding_box
assert_array_almost_equal(xbounds, (0, 4))
assert_array_almost_equal(ybounds, (1, 3))
+
def test_obround_orientation():
o = Obround((0, 0), 2, 1)
assert_equal(o.orientation, 'horizontal')
o = Obround((0, 0), 1, 2)
assert_equal(o.orientation, 'vertical')
+
def test_obround_subshapes():
- o = Obround((0,0), 1, 4)
+ o = Obround((0, 0), 1, 4)
ss = o.subshapes
assert_array_almost_equal(ss['rectangle'].position, (0, 0))
assert_array_almost_equal(ss['circle1'].position, (0, 1.5))
assert_array_almost_equal(ss['circle2'].position, (0, -1.5))
- o = Obround((0,0), 4, 1)
+ o = Obround((0, 0), 4, 1)
ss = o.subshapes
assert_array_almost_equal(ss['rectangle'].position, (0, 0))
assert_array_almost_equal(ss['circle1'].position, (1.5, 0))
assert_array_almost_equal(ss['circle2'].position, (-1.5, 0))
+
def test_obround_conversion():
- o = Obround((2.54,25.4), 254.0, 2540.0, units='metric')
+ o = Obround((2.54, 25.4), 254.0, 2540.0, units='metric')
- #No effect
+ # No effect
o.to_metric()
assert_equal(o.position, (2.54, 25.4))
assert_equal(o.width, 254.0)
@@ -782,15 +835,15 @@ def test_obround_conversion():
assert_equal(o.width, 10.0)
assert_equal(o.height, 100.0)
- #No effect
+ # No effect
o.to_inch()
assert_equal(o.position, (0.1, 1.0))
assert_equal(o.width, 10.0)
assert_equal(o.height, 100.0)
- o= Obround((0.1, 1.0), 10.0, 100.0, units='inch')
+ o = Obround((0.1, 1.0), 10.0, 100.0, units='inch')
- #No effect
+ # No effect
o.to_inch()
assert_equal(o.position, (0.1, 1.0))
assert_equal(o.width, 10.0)
@@ -801,25 +854,27 @@ def test_obround_conversion():
assert_equal(o.width, 254.0)
assert_equal(o.height, 2540.0)
- #No effect
+ # No effect
o.to_metric()
assert_equal(o.position, (2.54, 25.4))
assert_equal(o.width, 254.0)
assert_equal(o.height, 2540.0)
+
def test_obround_offset():
o = Obround((0, 0), 1, 2)
o.offset(1, 0)
- assert_equal(o.position,(1., 0.))
+ assert_equal(o.position, (1., 0.))
o.offset(0, 1)
- assert_equal(o.position,(1., 1.))
+ assert_equal(o.position, (1., 1.))
+
def test_polygon_ctor():
""" Test polygon creation
"""
- test_cases = (((0,0), 3, 5, 0),
+ test_cases = (((0, 0), 3, 5, 0),
((0, 0), 5, 6, 0),
- ((1,1), 7, 7, 45))
+ ((1, 1), 7, 7, 45))
for pos, sides, radius, hole_diameter in test_cases:
p = Polygon(pos, sides, radius, hole_diameter)
assert_equal(p.position, pos)
@@ -827,73 +882,80 @@ def test_polygon_ctor():
assert_equal(p.radius, radius)
assert_equal(p.hole_diameter, hole_diameter)
+
def test_polygon_bounds():
""" Test polygon bounding box calculation
"""
- p = Polygon((2,2), 3, 2, 0)
+ p = Polygon((2, 2), 3, 2, 0)
xbounds, ybounds = p.bounding_box
assert_array_almost_equal(xbounds, (0, 4))
assert_array_almost_equal(ybounds, (0, 4))
- p = Polygon((2,2), 3, 4, 0)
+ p = Polygon((2, 2), 3, 4, 0)
xbounds, ybounds = p.bounding_box
assert_array_almost_equal(xbounds, (-2, 6))
assert_array_almost_equal(ybounds, (-2, 6))
+
def test_polygon_conversion():
p = Polygon((2.54, 25.4), 3, 254.0, 0, units='metric')
- #No effect
+ # No effect
p.to_metric()
assert_equal(p.position, (2.54, 25.4))
assert_equal(p.radius, 254.0)
-
+
p.to_inch()
assert_equal(p.position, (0.1, 1.0))
assert_equal(p.radius, 10.0)
-
- #No effect
+
+ # No effect
p.to_inch()
assert_equal(p.position, (0.1, 1.0))
assert_equal(p.radius, 10.0)
p = Polygon((0.1, 1.0), 3, 10.0, 0, units='inch')
-
- #No effect
+
+ # No effect
p.to_inch()
assert_equal(p.position, (0.1, 1.0))
assert_equal(p.radius, 10.0)
-
+
p.to_metric()
assert_equal(p.position, (2.54, 25.4))
assert_equal(p.radius, 254.0)
-
- #No effect
+
+ # No effect
p.to_metric()
assert_equal(p.position, (2.54, 25.4))
assert_equal(p.radius, 254.0)
+
def test_polygon_offset():
p = Polygon((0, 0), 5, 10, 0)
p.offset(1, 0)
- assert_equal(p.position,(1., 0.))
+ assert_equal(p.position, (1., 0.))
p.offset(0, 1)
- assert_equal(p.position,(1., 1.))
+ assert_equal(p.position, (1., 1.))
+
def test_region_ctor():
""" Test Region creation
"""
- apt = Circle((0,0), 0)
- lines = (Line((0,0), (1,0), apt), Line((1,0), (1,1), apt), Line((1,1), (0,1), apt), Line((0,1), (0,0), apt))
- points = ((0, 0), (1,0), (1,1), (0,1))
+ apt = Circle((0, 0), 0)
+ lines = (Line((0, 0), (1, 0), apt), Line((1, 0), (1, 1), apt),
+ Line((1, 1), (0, 1), apt), Line((0, 1), (0, 0), apt))
+ points = ((0, 0), (1, 0), (1, 1), (0, 1))
r = Region(lines)
for i, p in enumerate(lines):
assert_equal(r.primitives[i], p)
+
def test_region_bounds():
""" Test region bounding box calculation
"""
- apt = Circle((0,0), 0)
- lines = (Line((0,0), (1,0), apt), Line((1,0), (1,1), apt), Line((1,1), (0,1), apt), Line((0,1), (0,0), apt))
+ apt = Circle((0, 0), 0)
+ lines = (Line((0, 0), (1, 0), apt), Line((1, 0), (1, 1), apt),
+ Line((1, 1), (0, 1), apt), Line((0, 1), (0, 0), apt))
r = Region(lines)
xbounds, ybounds = r.bounding_box
assert_array_almost_equal(xbounds, (0, 1))
@@ -901,68 +963,76 @@ def test_region_bounds():
def test_region_offset():
- apt = Circle((0,0), 0)
- lines = (Line((0,0), (1,0), apt), Line((1,0), (1,1), apt), Line((1,1), (0,1), apt), Line((0,1), (0,0), apt))
+ apt = Circle((0, 0), 0)
+ lines = (Line((0, 0), (1, 0), apt), Line((1, 0), (1, 1), apt),
+ Line((1, 1), (0, 1), apt), Line((0, 1), (0, 0), apt))
r = Region(lines)
xlim, ylim = r.bounding_box
r.offset(0, 1)
- assert_array_almost_equal((xlim, tuple([y+1 for y in ylim])), r.bounding_box)
+ new_xlim, new_ylim = r.bounding_box
+ assert_array_almost_equal(new_xlim, xlim)
+ assert_array_almost_equal(new_ylim, tuple([y + 1 for y in ylim]))
+
def test_round_butterfly_ctor():
""" Test round butterfly creation
"""
- test_cases = (((0,0), 3), ((0, 0), 5), ((1,1), 7))
+ test_cases = (((0, 0), 3), ((0, 0), 5), ((1, 1), 7))
for pos, diameter in test_cases:
b = RoundButterfly(pos, diameter)
assert_equal(b.position, pos)
assert_equal(b.diameter, diameter)
- assert_equal(b.radius, diameter/2.)
+ assert_equal(b.radius, diameter / 2.)
+
def test_round_butterfly_ctor_validation():
""" Test RoundButterfly argument validation
"""
assert_raises(TypeError, RoundButterfly, 3, 5)
- assert_raises(TypeError, RoundButterfly, (3,4,5), 5)
+ assert_raises(TypeError, RoundButterfly, (3, 4, 5), 5)
+
def test_round_butterfly_conversion():
b = RoundButterfly((2.54, 25.4), 254.0, units='metric')
-
- #No Effect
+
+ # No Effect
b.to_metric()
assert_equal(b.position, (2.54, 25.4))
assert_equal(b.diameter, (254.0))
-
+
b.to_inch()
assert_equal(b.position, (0.1, 1.0))
assert_equal(b.diameter, 10.0)
-
- #No effect
+
+ # No effect
b.to_inch()
assert_equal(b.position, (0.1, 1.0))
assert_equal(b.diameter, 10.0)
b = RoundButterfly((0.1, 1.0), 10.0, units='inch')
-
- #No effect
+
+ # No effect
b.to_inch()
assert_equal(b.position, (0.1, 1.0))
assert_equal(b.diameter, 10.0)
-
+
b.to_metric()
assert_equal(b.position, (2.54, 25.4))
assert_equal(b.diameter, (254.0))
-
- #No Effect
+
+ # No Effect
b.to_metric()
assert_equal(b.position, (2.54, 25.4))
assert_equal(b.diameter, (254.0))
+
def test_round_butterfly_offset():
b = RoundButterfly((0, 0), 1)
b.offset(1, 0)
- assert_equal(b.position,(1., 0.))
+ assert_equal(b.position, (1., 0.))
b.offset(0, 1)
- assert_equal(b.position,(1., 1.))
+ assert_equal(b.position, (1., 1.))
+
def test_round_butterfly_bounds():
""" Test RoundButterfly bounding box calculation
@@ -972,20 +1042,23 @@ def test_round_butterfly_bounds():
assert_array_almost_equal(xbounds, (-1, 1))
assert_array_almost_equal(ybounds, (-1, 1))
+
def test_square_butterfly_ctor():
""" Test SquareButterfly creation
"""
- test_cases = (((0,0), 3), ((0, 0), 5), ((1,1), 7))
+ test_cases = (((0, 0), 3), ((0, 0), 5), ((1, 1), 7))
for pos, side in test_cases:
b = SquareButterfly(pos, side)
assert_equal(b.position, pos)
assert_equal(b.side, side)
+
def test_square_butterfly_ctor_validation():
""" Test SquareButterfly argument validation
"""
assert_raises(TypeError, SquareButterfly, 3, 5)
- assert_raises(TypeError, SquareButterfly, (3,4,5), 5)
+ assert_raises(TypeError, SquareButterfly, (3, 4, 5), 5)
+
def test_square_butterfly_bounds():
""" Test SquareButterfly bounding box calculation
@@ -995,51 +1068,54 @@ def test_square_butterfly_bounds():
assert_array_almost_equal(xbounds, (-1, 1))
assert_array_almost_equal(ybounds, (-1, 1))
+
def test_squarebutterfly_conversion():
b = SquareButterfly((2.54, 25.4), 254.0, units='metric')
-
- #No effect
+
+ # No effect
b.to_metric()
assert_equal(b.position, (2.54, 25.4))
assert_equal(b.side, (254.0))
-
+
b.to_inch()
assert_equal(b.position, (0.1, 1.0))
assert_equal(b.side, 10.0)
-
- #No effect
+
+ # No effect
b.to_inch()
assert_equal(b.position, (0.1, 1.0))
assert_equal(b.side, 10.0)
b = SquareButterfly((0.1, 1.0), 10.0, units='inch')
-
- #No effect
+
+ # No effect
b.to_inch()
assert_equal(b.position, (0.1, 1.0))
assert_equal(b.side, 10.0)
-
+
b.to_metric()
assert_equal(b.position, (2.54, 25.4))
assert_equal(b.side, (254.0))
-
- #No effect
+
+ # No effect
b.to_metric()
assert_equal(b.position, (2.54, 25.4))
assert_equal(b.side, (254.0))
+
def test_square_butterfly_offset():
b = SquareButterfly((0, 0), 1)
b.offset(1, 0)
- assert_equal(b.position,(1., 0.))
+ assert_equal(b.position, (1., 0.))
b.offset(0, 1)
- assert_equal(b.position,(1., 1.))
+ assert_equal(b.position, (1., 1.))
+
def test_donut_ctor():
""" Test Donut primitive creation
"""
- test_cases = (((0,0), 'round', 3, 5), ((0, 0), 'square', 5, 7),
- ((1,1), 'hexagon', 7, 9), ((2, 2), 'octagon', 9, 11))
+ test_cases = (((0, 0), 'round', 3, 5), ((0, 0), 'square', 5, 7),
+ ((1, 1), 'hexagon', 7, 9), ((2, 2), 'octagon', 9, 11))
for pos, shape, in_d, out_d in test_cases:
d = Donut(pos, shape, in_d, out_d)
assert_equal(d.position, pos)
@@ -1047,65 +1123,68 @@ def test_donut_ctor():
assert_equal(d.inner_diameter, in_d)
assert_equal(d.outer_diameter, out_d)
+
def test_donut_ctor_validation():
assert_raises(TypeError, Donut, 3, 'round', 5, 7)
assert_raises(TypeError, Donut, (3, 4, 5), 'round', 5, 7)
assert_raises(ValueError, Donut, (0, 0), 'triangle', 3, 5)
assert_raises(ValueError, Donut, (0, 0), 'round', 5, 3)
+
def test_donut_bounds():
d = Donut((0, 0), 'round', 0.0, 2.0)
- assert_equal(d.lower_left, (-1.0, -1.0))
- assert_equal(d.upper_right, (1.0, 1.0))
xbounds, ybounds = d.bounding_box
assert_equal(xbounds, (-1., 1.))
assert_equal(ybounds, (-1., 1.))
+
def test_donut_conversion():
d = Donut((2.54, 25.4), 'round', 254.0, 2540.0, units='metric')
-
- #No effect
+
+ # No effect
d.to_metric()
assert_equal(d.position, (2.54, 25.4))
assert_equal(d.inner_diameter, 254.0)
assert_equal(d.outer_diameter, 2540.0)
-
+
d.to_inch()
assert_equal(d.position, (0.1, 1.0))
assert_equal(d.inner_diameter, 10.0)
assert_equal(d.outer_diameter, 100.0)
-
- #No effect
+
+ # No effect
d.to_inch()
assert_equal(d.position, (0.1, 1.0))
assert_equal(d.inner_diameter, 10.0)
assert_equal(d.outer_diameter, 100.0)
d = Donut((0.1, 1.0), 'round', 10.0, 100.0, units='inch')
-
- #No effect
+
+ # No effect
d.to_inch()
assert_equal(d.position, (0.1, 1.0))
assert_equal(d.inner_diameter, 10.0)
assert_equal(d.outer_diameter, 100.0)
-
+
d.to_metric()
assert_equal(d.position, (2.54, 25.4))
assert_equal(d.inner_diameter, 254.0)
assert_equal(d.outer_diameter, 2540.0)
-
- #No effect
+
+ # No effect
d.to_metric()
assert_equal(d.position, (2.54, 25.4))
assert_equal(d.inner_diameter, 254.0)
assert_equal(d.outer_diameter, 2540.0)
+
def test_donut_offset():
d = Donut((0, 0), 'round', 1, 10)
d.offset(1, 0)
- assert_equal(d.position,(1., 0.))
+ assert_equal(d.position, (1., 0.))
d.offset(0, 1)
- assert_equal(d.position,(1., 1.))
+ assert_equal(d.position, (1., 1.))
+
def test_drill_ctor():
""" Test drill primitive creation
@@ -1115,7 +1194,8 @@ def test_drill_ctor():
d = Drill(position, diameter, None)
assert_equal(d.position, position)
assert_equal(d.diameter, diameter)
- assert_equal(d.radius, diameter/2.)
+ assert_equal(d.radius, diameter / 2.)
+
def test_drill_ctor_validation():
""" Test drill argument validation
@@ -1133,46 +1213,48 @@ def test_drill_bounds():
assert_array_almost_equal(xbounds, (0, 2))
assert_array_almost_equal(ybounds, (1, 3))
+
def test_drill_conversion():
d = Drill((2.54, 25.4), 254., None, units='metric')
-
- #No effect
+
+ # No effect
d.to_metric()
assert_equal(d.position, (2.54, 25.4))
assert_equal(d.diameter, 254.0)
-
+
d.to_inch()
assert_equal(d.position, (0.1, 1.0))
assert_equal(d.diameter, 10.0)
-
- #No effect
+
+ # No effect
d.to_inch()
assert_equal(d.position, (0.1, 1.0))
assert_equal(d.diameter, 10.0)
-
d = Drill((0.1, 1.0), 10., None, units='inch')
-
- #No effect
+
+ # No effect
d.to_inch()
assert_equal(d.position, (0.1, 1.0))
assert_equal(d.diameter, 10.0)
-
+
d.to_metric()
assert_equal(d.position, (2.54, 25.4))
assert_equal(d.diameter, 254.0)
-
- #No effect
+
+ # No effect
d.to_metric()
assert_equal(d.position, (2.54, 25.4))
assert_equal(d.diameter, 254.0)
+
def test_drill_offset():
d = Drill((0, 0), 1., None)
d.offset(1, 0)
- assert_equal(d.position,(1., 0.))
+ assert_equal(d.position, (1., 0.))
d.offset(0, 1)
- assert_equal(d.position,(1., 1.))
+ assert_equal(d.position, (1., 1.))
+
def test_drill_equality():
d = Drill((2.54, 25.4), 254., None)