1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
try:
from . import generic as g
except BaseException:
import generic as g
class MedialTests(g.unittest.TestCase):
def test_medial(self):
p = g.get_mesh("2D/wrench.dxf")
assert p.is_closed
assert len(p.polygons_full) == 1
poly = p.polygons_full[0]
medial = p.medial_axis()
points = medial.vertices.view(g.np.ndarray)
assert all(poly.contains(g.Point(v)) for v in points)
# circles are a special case for medial axis
poly = g.Point([0, 0]).buffer(1.0)
# construct a Path2D from the polygon medial axis
med = g.trimesh.path.Path2D(
**g.trimesh.path.exchange.misc.edges_to_path(
*g.trimesh.path.polygons.medial_axis(poly)
)
)
# should have returned a single tiny line
# with midpoint at origin
assert len(med.vertices) == 2
assert len(med.entities) == 1
assert float(g.np.ptp(med.vertices.mean(axis=0))) < 1e-8
if __name__ == "__main__":
g.trimesh.util.attach_to_log()
g.unittest.main()
|
