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
|
import os
import unittest
import pywavefront
from utils import fixture
class TestWavefront(unittest.TestCase):
def setUp(self):
self.mesh_names = ['Simple', 'SimpleB']
self.material_names = ['Material.simple', 'Material2.simple']
self.meshes = pywavefront.Wavefront(fixture('simple.obj'))
def testMaterials(self):
"""Ensure parsed wavefront materials match known values."""
self.assertEqual(len(self.meshes.materials), len(self.material_names))
self.assertEqual(self.meshes.materials[self.material_names[0]].__class__,
pywavefront.material.Material)
def testMeshes(self):
"""Ensure parsed wavefront meshes match known values."""
self.assertEqual(len(self.meshes.meshes), len(self.mesh_names))
self.assertEqual(self.meshes.meshes[self.mesh_names[0]].__class__,
pywavefront.mesh.Mesh)
def testMeshList(self):
"""Ensure parsed wavefront mesh list matches known values."""
self.assertEqual(len(self.meshes.mesh_list), len(self.mesh_names))
self.assertEqual(self.meshes.mesh_list[0].__class__,
pywavefront.mesh.Mesh)
def testAddDuplicateMesh(self):
"""Adding a duplicate mesh should increment the mesh list, but not the meshes hash."""
self.meshes.add_mesh(self.meshes.meshes[self.mesh_names[0]])
self.assertEqual(len(self.meshes.meshes), len(self.mesh_names))
self.assertEqual(len(self.meshes.mesh_list), len(self.mesh_names) + 1)
def testMeshMaterialVertices(self):
"""Mesh vertices should have known values."""
self.assertEqual(len(self.meshes.meshes[self.mesh_names[0]].materials[0].vertices), 24)
class TestBrokenWavefront(unittest.TestCase):
def testUnknownUsemtl(self):
"""Referencing an unknown material with usemtl should raise an exception."""
self.assertRaises(pywavefront.PywavefrontException,
pywavefront.Wavefront,
fixture('simple_unknown_usemtl.obj'))
class TestNoMaterial(TestWavefront):
def setUp(self):
# reset the obj file to new file with no mtl line
self.mesh_names = ['Simple', 'SimpleB']
self.material_names = ["default0"]
self.meshes = pywavefront.Wavefront(fixture('simple_no_mtl.obj'))
def testMeshMaterialVertices(self):
"""Mesh vertices should have known values."""
self.assertEqual(len(self.meshes.meshes[self.mesh_names[0]].materials[0].vertices), 48)
class TestNoObjectNoMaterial(TestNoMaterial):
def setUp(self):
# reset the obj file to new file with no mtl line
self.mesh_names = [None]
self.material_names = ["default0"]
self.meshes = pywavefront.Wavefront(fixture('simple_no_object_no_mtl.obj'))
class TestZeroFaceNormaIndex(TestWavefront):
def setUp(self):
# reset the obj file to new file with no mtl line
self.mesh_names = ['Simple', 'SimpleB']
self.material_names = ['Material.simple', 'Material2.simple']
self.meshes = pywavefront.Wavefront(fixture('simple_zero_indices.obj'))
class TestExtraSpace(TestWavefront):
def setUp(self):
# reset the obj file to new file with no mtl line
self.mesh_names = ['Simple', 'SimpleB']
self.material_names = ['Material.simple', 'Material2.simple']
self.meshes = pywavefront.Wavefront(fixture('simple_extra_empty_lines.obj'))
|
