def setUp(self):
self.test_shape = RotateSplineShape(
points=[(50, 0), (50, 20), (70,
80), (90,
50), (70,
0), (90,
-50), (70,
-80), (50,
-20)])
def test_absolute_shape_volume(self):
"""creates a rotated shape using spline connections and checks \
the volume is correct"""
test_shape = RotateSplineShape(points=[(0, 0), (0, 20), (20,
20), (20, 0)])
test_shape.rotation_angle = 360
test_shape.create_solid()
assert test_shape.solid is not None
assert test_shape.volume > 100
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 = RotateSplineShape(points=[(0, 0), (0, 20), (20, 20)],
rotation_angle=360)
test_shape.solid
assert test_shape.hash_value is not None
initial_hash_value = test_shape.hash_value
test_shape.rotation_angle = 180
assert test_shape.hash_value == initial_hash_value
test_shape.solid
assert test_shape.hash_value != initial_hash_value
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 = RotateSplineShape(points=[(0, 0), (0, 20), (20, 20)],
rotation_angle=360)
assert test_shape.solid is not None
assert test_shape.hash_value is not None
initial_hash_value = test_shape.hash_value
test_shape.rotation_angle = 180
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_initial_solid_construction(self):
"""tests that a cadquery solid with a unique hash is constructed when .solid is called"""
test_shape = RotateSplineShape(points=[(0, 0), (0, 20), (20, 20),
(20, 0)],
rotation_angle=360)
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_cut_volume(self):
"""creates a rotated shape using spline connections with another shape \
cut out and checks the volume is correct"""
inner_shape = RotateSplineShape(points=[(5, 5), (5, 10), (10, 10),
(10, 5)],
rotation_angle=180)
outer_shape = RotateSplineShape(points=[(3, 3), (3, 12), (12, 12),
(12, 3)],
rotation_angle=180)
outer_shape_with_cut = RotateSplineShape(
points=[(3, 3), (3, 12), (12, 12), (12, 3)],
cut=inner_shape,
rotation_angle=180,
)
assert inner_shape.volume == pytest.approx(900.88, abs=0.1)
assert outer_shape.volume == pytest.approx(2881.76, abs=0.1)
assert outer_shape_with_cut.volume == pytest.approx(2881.76 - 900.88,
abs=0.2)
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 = RotateSplineShape(points=[(0, 0), (0, 20), (20, 20),
(20, 0)],
rotation_angle=360)
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
class TestRotateSplineShape(unittest.TestCase):
def setUp(self):
self.test_shape = RotateSplineShape(
points=[(50, 0), (50, 20), (70, 80), (90, 50), (70, 0),
(90, -50), (70, -80), (50, -20)])
def test_default_parameters(self):
"""Checks that the default parameters of a RotateSplineShape are correct."""
assert self.test_shape.rotation_angle == 360
assert self.test_shape.stp_filename == "RotateSplineShape.stp"
assert self.test_shape.stl_filename == "RotateSplineShape.stl"
assert self.test_shape.azimuth_placement_angle == 0
def test_absolute_shape_volume(self):
"""creates a rotated shape using spline connections and checks the volume
is correct"""
self.test_shape.rotation_angle = 360
volume_360 = self.test_shape.volume
assert self.test_shape.solid is not None
assert volume_360 > 100
self.test_shape.rotation_angle = 180
assert self.test_shape.solid is not None
assert self.test_shape.volume == pytest.approx(volume_360 * 0.5)
def test_cut_volume(self):
"""Creates a RotateSplineShape with another RotateSplineShape cut out
and checks that the volume is correct."""
inner_shape = RotateSplineShape(
points=[(5, 5), (5, 10), (10, 10), (10, 5)], rotation_angle=180
)
outer_shape = RotateSplineShape(
points=[(3, 3), (3, 12), (12, 12), (12, 3)], rotation_angle=180
)
outer_shape_with_cut = RotateSplineShape(
points=[(3, 3), (3, 12), (12, 12), (12, 3)],
cut=inner_shape,
rotation_angle=180,
)
assert inner_shape.volume == pytest.approx(900.88, abs=0.1)
assert outer_shape.volume == pytest.approx(2881.76, abs=0.1)
assert outer_shape_with_cut.volume == pytest.approx(
2881.76 - 900.88, abs=0.2)
def test_shape_face_areas(self):
"""Creates RotateSplineShapes and checks that the face areas are expected."""
assert len(self.test_shape.areas) == 1
assert len(set(self.test_shape.areas)) == 1
self.test_shape.rotation_angle = 180
assert len(self.test_shape.areas) == 3
assert len(set([round(i) for i in self.test_shape.areas])) == 2
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
)