def test_fragility_function_already_fragile():
mesh_fn = os.path.join(os.path.dirname(__file__), 'test_meshes', 'fragility_test_1.stl')
mesh = trimesh.load(mesh_fn)
mesh = mesh.subdivide()
mesh = mesh.subdivide()
tree = bsp_tree.BSPTree(mesh)
origin = np.zeros(3)
normal = np.array([0., 0., 1.])
plane = (origin, normal)
trees = [bsp_tree.expand_node(tree, tree.nodes[0].path, plane)[0]]
objective_functions.evaluate_fragility_objective(trees, tree.nodes[0].path)
assert trees[0].objectives['fragility'] == 0
def test_edge_fragility(config):
config.connector_diameter = 3
config.connector_wall_distance = 1
mesh_fn = os.path.join(os.path.dirname(__file__), 'test_meshes', 'fragility_test_2.stl')
mesh = trimesh.load(mesh_fn)
mesh = mesh.subdivide()
tree = bsp_tree.BSPTree(mesh)
origin = np.zeros(3)
normal = np.array([0., 0., 1.])
plane = (origin, normal)
fragile_cut_tree, result = bsp_tree.expand_node(tree, tree.nodes[0].path, plane)
objective_functions.evaluate_fragility_objective([fragile_cut_tree], tree.nodes[0].path)
assert fragile_cut_tree.objectives['fragility'] == np.inf
def test_fragility_function_multiple_trees(config):
config.plane_spacing = 5
config.connector_diameter = 5
mesh_fn = os.path.join(os.path.dirname(__file__), 'test_meshes', 'fragility_test_1.stl')
mesh = trimesh.load(mesh_fn)
mesh = mesh.subdivide()
mesh = mesh.subdivide()
tree = bsp_tree.BSPTree(mesh)
normal = np.array([0., 0., 1.])
planes = bsp_tree.get_planes(mesh, normal)
trees = []
for plane in planes:
candidate, result = bsp_tree.expand_node(tree, tree.nodes[0].path, plane)
trees.append(candidate)
objective_functions.evaluate_fragility_objective(trees, tree.nodes[0].path)
assert trees[0].objectives['fragility'] == np.inf
assert trees[6].objectives['fragility'] == np.inf
assert trees[7].objectives['fragility'] == np.inf
assert trees[11].objectives['fragility'] == np.inf