surfaces/surf/test/test_polymesh.py
Stephen M. McQuay 285c6c7254 First functioning pass at connectivity.
- added unittests (passing)
    - updated run script to take int for number of subds
2012-07-13 15:33:24 -06:00

57 lines
1.8 KiB
Python

import json
import os
import unittest
from surf.geometry import PolygonMesh
class TestPM(unittest.TestCase):
def setUp(self):
path, file_name = os.path.split(__file__)
self.samples_dir = os.path.join(path, os.pardir, os.pardir,
'blender', 'samples')
full_cube_file_name = os.path.join(self.samples_dir, 'cube.json')
with open(full_cube_file_name) as cube_file:
self.full_cube = json.load(cube_file)
cube_file_name = os.path.join(self.samples_dir, 'cube-no-connectivity.json')
with open(cube_file_name) as cube_file:
self.skeleton_cube = json.load(cube_file)
self.skel = PolygonMesh(**self.skeleton_cube)
self.cube = PolygonMesh(**self.full_cube)
def test_cube_load(self):
p = PolygonMesh(**self.full_cube)
v = p.vertices[0]
self.assertAlmostEqual(v.x, -1.0)
self.assertAlmostEqual(v.y, -1.0)
self.assertAlmostEqual(v.z, -1.0)
def test_connectivity(self):
self.skel.faces_for_vert
self.skel.edges_for_vert
self.skel.edges_for_face
self.skel.faces_for_edge
def test_faces_for_vert(self):
for pi, qi in zip(self.skel.faces_for_vert, self.cube.faces_for_vert):
self.assertEqual(sorted(pi), sorted(qi))
def test_edges_for_vert(self):
for pi, qi in zip(self.skel.edges_for_vert, self.cube.edges_for_vert):
self.assertEqual(sorted(pi), sorted(qi))
def test_edges_for_face(self):
for pi, qi in zip(self.skel.edges_for_face, self.cube.edges_for_face):
self.assertEqual(sorted(pi), sorted(qi))
def test_faces_for_edge(self):
for pi, qi in zip(self.skel.faces_for_edge, self.cube.faces_for_edge):
self.assertEqual(sorted(pi), sorted(qi))
if __name__ == '__main__':
unittest.main(verbosity=3)