def test_viz_to_binary_json(test_dir): components = [ Beam("bm1", n1=[0, 0, 0], n2=[2, 0, 0], sec="IPE220", colour="red"), Beam("bm2", n1=[0, 0, 1], n2=[2, 0, 1], sec="HP220x10", colour="blue"), Beam("bm3", n1=[0, 0, 2], n2=[2, 0, 2], sec="BG800x400x20x40", colour="green"), Beam("bm4", n1=[0, 0, 3], n2=[2, 0, 3], sec="CIRC200", colour="green"), Beam("bm5", n1=[0, 0, 4], n2=[2, 0, 4], sec="TUB200x10", colour="green"), Plate( "pl1", [(0, 0, 0), (0, 0, 1), (0, 1, 1), (0, 1, 0)], 0.01, use3dnodes=True, ), ] a = Assembly("my_test_assembly") / components res = a.to_vis_mesh() res.to_binary_and_json(test_dir / "viz/binjson/beams") res.to_custom_json(test_dir / "viz/binjson/beams.json")
def test_viz_structural(): components = [ Beam("bm1", n1=[0, 0, 0], n2=[2, 0, 0], sec="IPE220", colour="red"), Beam("bm2", n1=[0, 0, 1], n2=[2, 0, 1], sec="HP220x10", colour="blue"), Beam("bm3", n1=[0, 0, 2], n2=[2, 0, 2], sec="BG800x400x20x40", colour="green"), Beam("bm4", n1=[0, 0, 3], n2=[2, 0, 3], sec="CIRC200", colour="green"), Beam("bm5", n1=[0, 0, 4], n2=[2, 0, 4], sec="TUB200x10", colour="green"), Plate( "pl1", [(0, 0, 0), (0, 0, 1), (0, 1, 1), (0, 1, 0)], 0.01, use3dnodes=True, ), ] a = Assembly("my_test_assembly") / components res = a.to_vis_mesh(auto_merge_by_color=False) merged = res.merge_objects_in_parts_by_color() assert res.num_polygons == 416 assert len(res.world[0].id_map.values()) == 6 assert merged.num_polygons == 416 assert len(merged.world[0].id_map.values()) == 4