62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from collections import defaultdict
|
|
|
|
import simplecadapi as scad
|
|
|
|
|
|
def _box_edge_occurrences():
|
|
box = scad.make_box_rsolid(1.0, 1.0, 1.0)
|
|
occurrences = []
|
|
for face in box.get_faces():
|
|
occurrences.extend(face.get_outer_wire().get_edges())
|
|
return box, occurrences
|
|
|
|
|
|
def test_box_edge_occurrences_share_canonical_topo_ids():
|
|
_, occurrences = _box_edge_occurrences()
|
|
|
|
assert len(occurrences) == 24
|
|
assert len({edge.topo_id for edge in occurrences}) == 12
|
|
|
|
|
|
def test_shared_edge_occurrences_share_tags_and_metadata():
|
|
_, occurrences = _box_edge_occurrences()
|
|
groups = defaultdict(list)
|
|
for edge in occurrences:
|
|
groups[edge.topo_id].append(edge)
|
|
|
|
shared = next(group for group in groups.values() if len(group) >= 2)
|
|
edge_a, edge_b = shared[0], shared[1]
|
|
|
|
scad.apply_tag(edge_a, "shared.edge")
|
|
edge_a.set_metadata("name", "same-topological-edge")
|
|
|
|
assert "shared.edge" in scad.list_tags(edge_b)
|
|
assert edge_b.get_metadata("name") == "same-topological-edge"
|
|
|
|
|
|
def test_solid_get_edges_returns_unique_topological_edges():
|
|
box, _ = _box_edge_occurrences()
|
|
|
|
edges = box.get_edges()
|
|
|
|
assert len(edges) == 12
|
|
assert len({edge.topo_id for edge in edges}) == 12
|
|
|
|
|
|
def test_edge_incident_faces_and_face_adjacency_are_available():
|
|
box, _ = _box_edge_occurrences()
|
|
|
|
edge = box.get_edges(0)
|
|
incident_faces = edge.get_incident_faces()
|
|
|
|
assert len(incident_faces) == 2
|
|
assert len({face.topo_id for face in incident_faces}) == 2
|
|
|
|
face = box.get_faces(0)
|
|
adjacent_faces = face.get_adjacent_faces()
|
|
|
|
assert len(adjacent_faces) == 4
|
|
assert face.topo_id not in {adjacent.topo_id for adjacent in adjacent_faces}
|