def setUp(self):
self.test_shape = ExtrudeSplineShape(points=[(50, 0),
(50, 20), (70, 80),
(90, 50), (70, 0),
(90, -50), (70, -80),
(50, -20)],
distance=30)
def test_absolute_shape_volume(self):
"""creates an extruded shape with one placement angle using spline \
connections and checks the volume is correct"""
test_shape = ExtrudeSplineShape(points=[(0, 0), (0, 20), (20, 20),
(20, 0)],
distance=30)
test_shape.create_solid()
assert test_shape.solid is not None
assert test_shape.volume > 20 * 20 * 30
def test_hash_value_update(self):
"""tests that the hash_value of the shape is not updated until a new solid has been created"""
test_shape = ExtrudeSplineShape(points=[(0, 0), (0, 20), (20, 20)],
distance=20)
test_shape.solid
assert test_shape.hash_value is not None
initial_hash_value = test_shape.hash_value
test_shape.distance = 30
assert test_shape.hash_value == initial_hash_value
test_shape.solid
assert test_shape.hash_value != initial_hash_value
def test_rotation_angle(self):
"""creates an extruded shape with a rotation_angle < 360 and checks that the
correct cut is performed and the volume is correct"""
test_shape = ExtrudeSplineShape(
points=[(50, 0), (50, 20), (70, 20), (70, 0)],
distance=50,
azimuth_placement_angle=[45, 135, 225, 315]
)
test_volume = test_shape.volume
test_shape.rotation_angle = 180
assert test_shape.volume == pytest.approx(test_volume * 0.5)
def test_conditional_solid_reconstruction(self):
"""tests that a new cadquery solid with a new unique hash is constructed when .solid is called again after changes have been made to the shape"""
test_shape = ExtrudeSplineShape(points=[(0, 0), (0, 20), (20, 20)],
distance=20)
assert test_shape.solid is not None
assert test_shape.hash_value is not None
initial_hash_value = test_shape.hash_value
test_shape.distance = 30
assert test_shape.solid is not None
assert test_shape.hash_value is not None
assert initial_hash_value != test_shape.hash_value
def test_cut_volume(self):
"""creates an extruded shape using spline connections with another shape cut
out and checks that the volume is correct"""
inner_shape = ExtrudeSplineShape(
points=[(5, 5), (5, 10), (10, 10), (10, 5)], distance=30
)
outer_shape = ExtrudeSplineShape(
points=[(3, 3), (3, 12), (12, 12), (12, 3)], distance=30
)
outer_shape_with_cut = ExtrudeSplineShape(
points=[(3, 3), (3, 12), (12, 12), (12, 3)], cut=inner_shape, distance=30,
)
assert inner_shape.volume == pytest.approx(1165, abs=2)
assert outer_shape.volume == pytest.approx(3775, abs=2)
assert outer_shape_with_cut.volume == pytest.approx(3775 - 1165, abs=2)
def test_initial_solid_construction(self):
"""tests that a cadquery solid with a unique hash is constructed when .solid is called"""
test_shape = ExtrudeSplineShape(points=[(0, 0), (0, 20), (20, 20),
(20, 0)],
distance=20)
assert test_shape.hash_value is None
assert test_shape.solid is not None
assert type(test_shape.solid).__name__ == "Workplane"
assert test_shape.hash_value is not None
def test_solid_return(self):
"""tests that the same cadquery solid with the same unique hash is returned when shape.solid is called again when no changes have been made to the shape"""
test_shape = ExtrudeSplineShape(points=[(0, 0), (0, 20), (20, 20),
(20, 0)],
distance=20)
assert test_shape.solid is not None
assert test_shape.hash_value is not None
initial_hash_value = test_shape.hash_value
assert test_shape.solid is not None
assert test_shape.hash_value is not None
assert initial_hash_value == test_shape.hash_value
def test_extruded_shape_relative_volume(self):
"""creates two extruded shapes with different placement angles using spline \
connections and checks their relative volumes are correct"""
test_shape_1 = ExtrudeSplineShape(points=[(13, 0), (13, 20), (16, 20),
(20, 10), (16, 0)],
distance=5)
test_shape_1.azimuth_placement_angle = 0
# test_shape_2 is test_shape_1 extruded 4 times
test_shape_2 = ExtrudeSplineShape(points=[(13, 0), (13, 20), (16, 20),
(20, 10), (16, 0)],
distance=5)
test_shape_2.azimuth_placement_angle = [0, 90, 180, 270]
assert test_shape_1.volume * 4 == pytest.approx(test_shape_2.volume,
rel=0.01)
class TestExtrudeSplineShape(unittest.TestCase):
def setUp(self):
self.test_shape = ExtrudeSplineShape(points=[(50, 0),
(50, 20), (70, 80),
(90, 50), (70, 0),
(90, -50), (70, -80),
(50, -20)],
distance=30)
def test_default_parameters(self):
"""Checks that the default parameters of an ExtrudeSplineShape are correct."""
assert self.test_shape.rotation_angle == 360
assert self.test_shape.stp_filename == "ExtrudeSplineShape.stp"
assert self.test_shape.stl_filename == "ExtrudeSplineShape.stl"
assert self.test_shape.extrude_both
def test_absolute_shape_volume(self):
"""Creates an ExtrudeSplineShape and checks that the volume is correct."""
assert self.test_shape.solid is not None
assert self.test_shape.volume > 20 * 20 * 30
def test_shape_face_areas(self):
"""Creates an ExtrudeSplineShape and checks that the face areas are expected."""
self.test_shape.extrude_both = False
assert len(self.test_shape.areas) == 3
assert len(set([round(i) for i in self.test_shape.areas])) == 2
def test_relative_shape_volume(self):
"""Creates two ExtrudeSplineShapes and checks that their relative volumes
are correct."""
test_volume = self.test_shape.volume
self.test_shape.azimuth_placement_angle = [0, 180]
assert self.test_shape.volume == pytest.approx(test_volume * 2,
rel=0.01)
def test_cut_volume(self):
"""Creates an ExtrudeSplineShape with another ExtrudeSplineShape cut out
and checks that the volume is correct."""
inner_shape = ExtrudeSplineShape(points=[(5, 5), (5, 10), (10, 10),
(10, 5)],
distance=30)
outer_shape = ExtrudeSplineShape(points=[(3, 3), (3, 12), (12, 12),
(12, 3)],
distance=30)
outer_shape_with_cut = ExtrudeSplineShape(
points=[(3, 3), (3, 12), (12, 12), (12, 3)],
cut=inner_shape,
distance=30,
)
assert inner_shape.volume == pytest.approx(1165, abs=2)
assert outer_shape.volume == pytest.approx(3775, abs=2)
assert outer_shape_with_cut.volume == pytest.approx(3775 - 1165, abs=2)
def test_rotation_angle(self):
"""Creates an ExtrudeStraightShape with a rotation_angle < 360 and checks that
the correct cut is performed and the volume is correct."""
self.test_shape.azimuth_placement_angle = [45, 135, 225, 315]
test_volume = self.test_shape.volume
self.test_shape.rotation_angle = 180
assert self.test_shape.volume == pytest.approx(test_volume * 0.5,
rel=0.01)
def test_extrude_both(self):
"""Creates an ExtrudeSplineShape with extrude_both = True and False and checks
that the volumes are correct."""
test_volume_extrude_both = self.test_shape.volume
self.test_shape.extrude_both = False
assert self.test_shape.volume == pytest.approx(
test_volume_extrude_both)
def test_export_stp(self):
"""Exports and stp file with mode = solid and wire and checks
that the outputs exist and relative file sizes are correct."""
os.system("rm test_solid.stp test_solid2.stp test_wire.stp")
self.test_shape.export_stp('test_solid.stp', mode='solid')
self.test_shape.export_stp('test_solid2.stp')
self.test_shape.export_stp('test_wire.stp', mode='wire')
assert Path("test_solid.stp").exists() is True
assert Path("test_solid2.stp").exists() is True
assert Path("test_wire.stp").exists() is True
assert Path("test_solid.stp").stat().st_size == \
Path("test_solid2.stp").stat().st_size
assert Path("test_wire.stp").stat().st_size < \
Path("test_solid2.stp").stat().st_size
os.system("rm test_solid.stp test_solid2.stp test_wire.stp")
def test_incorrect_points_input(self):
"""Checks that an error is raised when the points are input with the
connection"""
def incorrect_points_definition():
self.test_shape.points = [(10, 10, 'spline'), (10, 30, 'spline'),
(30, 30, 'spline'), (30, 10, 'spline')]
self.assertRaises(ValueError, incorrect_points_definition)