update tests to unittest2 (or unittest py2.7)
This commit is contained in:
@@ -0,0 +1,47 @@
|
||||
import json
|
||||
import os
|
||||
import unittest
|
||||
|
||||
from surf.geometry import PolygonMesh
|
||||
from surf.subd import cc
|
||||
|
||||
|
||||
class TestCC(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
path, file_name = os.path.split(__file__)
|
||||
self.samples_dir = os.path.join(path, os.pardir, os.pardir, os.pardir,
|
||||
'blender', 'samples')
|
||||
cube_file_name = os.path.join(self.samples_dir, 'cube.json')
|
||||
with open(cube_file_name, 'r') as cube_file:
|
||||
self.cube = json.load(cube_file)
|
||||
cube_file_name = os.path.join(self.samples_dir,
|
||||
'cube-blender-cc-1.json')
|
||||
with open(cube_file_name, 'r') as cube_file:
|
||||
self.cube1 = json.load(cube_file)
|
||||
|
||||
def test_refined_vertex_location(self):
|
||||
p = PolygonMesh(**self.cube)
|
||||
p2 = cc.refine(p)
|
||||
v2 = [list(i) for i in p2.vertices]
|
||||
for a, b in zip(sorted(self.cube1['vertices']), sorted(v2)):
|
||||
for i in range(len(a)):
|
||||
self.assertAlmostEqual(a[i], b[i])
|
||||
|
||||
def test_vert_count(self):
|
||||
p = PolygonMesh(**self.cube)
|
||||
p2 = cc.refine(p)
|
||||
self.assertEqual(len(p2.vertices), 26)
|
||||
|
||||
def test_edge_count(self):
|
||||
p = PolygonMesh(**self.cube)
|
||||
p2 = cc.refine(p)
|
||||
self.assertEqual(len(p2.edges), 48)
|
||||
|
||||
def test_face_count(self):
|
||||
p = PolygonMesh(**self.cube)
|
||||
p2 = cc.refine(p)
|
||||
self.assertEqual(len(p2.faces), 24)
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main(verbosity=3)
|
||||
Reference in New Issue
Block a user