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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
try:
from . import generic as g
except BaseException:
import generic as g
try:
import collada
except ImportError:
collada = None
except BaseException:
# TODO : REMOVE WHEN UPSTREAM RELEASE FIXED
# https://github.com/pycollada/pycollada/pull/92
g.log.error("DAE fix not pushed yet!")
collada = None
class DAETest(g.unittest.TestCase):
def test_duck(self):
"""
Load a collada scene with pycollada.
"""
if collada is None:
g.log.error("no pycollada to test!")
return
scene = g.get_mesh("duck.dae")
assert len(scene.geometry) == 1
assert len(scene.graph.nodes_geometry) == 1
conv = scene.convert_units("inch")
assert conv.units == "inch"
def test_shoulder(self):
if collada is None:
g.log.error("no pycollada to test!")
return
scene = g.get_mesh("shoulder.zae")
assert len(scene.geometry) == 3
assert len(scene.graph.nodes_geometry) == 3
assert scene.units != "mm"
conv = scene.convert_units("mm")
assert conv.units == "mm"
def test_export(self):
if collada is None:
g.log.error("no pycollada to test!")
return
a = g.get_mesh("ballA.off")
r = a.export(file_type="dae")
assert len(r) > 0
def test_obj_roundtrip(self):
# get a zipped-DAE scene
s = g.get_mesh("duck.zae", force="mesh")
with g.TemporaryDirectory() as root:
# export using a file path so it can auto-create
# a FilePathResolver to write the stupid assets
path = g.os.path.join(root, "duck.obj")
s.export(path)
# bring it back from outer space
rec = g.trimesh.load(path, force="mesh")
assert g.np.ptp(g.np.ptp(rec.visual.uv, axis=0)) > 1e-5
assert s.visual.material.baseColorTexture.size == rec.visual.material.image.size
conv = s.convert_units("inch")
assert conv.units == "inch"
def test_material_round(self):
"""
Test to make sure materials survive a roundtrip
with an actually identical result
"""
s = g.get_mesh("blue_cube.dae")
assert len(s.geometry) == 1
m = next(iter(s.geometry.values()))
rs = g.trimesh.load(
file_obj=g.trimesh.util.wrap_as_stream(m.export(file_type="dae")),
file_type="dae",
)
assert len(rs.geometry) == 1
r = next(iter(rs.geometry.values()))
# this will compare everything in `material._data`
assert hash(m.visual.material) == hash(r.visual.material)
if __name__ == "__main__":
g.trimesh.util.attach_to_log()
g.unittest.main()
|
