Updated unittest after inheritance fix

tpaviot · tpaviot · commit 122701fcb502 · 2015-08-15T07:21:04.000+02:00
diff --git a/test/occutils_test.py b/test/occutils_test.py
@@ -22,17 +22,26 @@
 
 sys.path.append('../OCCUtils')
 
-from OCC.BRepPrimAPI import BRepPrimAPI_MakeBox
+from OCC.BRepPrimAPI import BRepPrimAPI_MakeBox, BRepPrimAPI_MakeSphere
 from OCC.TopoDS import TopoDS_Face, TopoDS_Edge
 
 from Topology import Topo
 from edge import Edge
+from face import Face
+from wire import Wire
+from vertex import Vertex
+from shell import Shell
+from solid import Solid
 
 
 def get_test_box_shape():
     return BRepPrimAPI_MakeBox(10, 20, 30).Shape()
 
 
+def get_test_sphere_shape():
+    return BRepPrimAPI_MakeSphere(10.).Shape()
+
+
 class TestTopo(unittest.TestCase):
     def test_init(self):
         b = get_test_box_shape()
@@ -79,15 +88,58 @@ def test_creat_edge(self):
         edge_0 = t.edges().next()  # it's a TopoDS_Edge
         assert not edge_0.IsNull()
         # then create an edge
-        my_Edge = Edge(edge_0)
-        assert not my_Edge.IsNull()
-        assert my_Edge.tolerance == 1e-06
+        my_edge = Edge(edge_0)
+        assert not my_edge.IsNull()
+        assert my_edge.tolerance == 1e-06
+        assert my_edge.type == 'line'
+        assert my_edge.length() == 30.
+
+
+class TestFace(unittest.TestCase):
+    def test_creat_face(self):
+        # create a box
+        my_face = Face(BRepPrimAPI_MakeSphere(1., 1.).Face())
+        assert not my_face.IsNull()
+        assert my_face.tolerance == 1e-06
+        assert not my_face.is_planar()
+        assert my_face.is_trimmed()
+
+
+class TestWire(unittest.TestCase):
+    def test_creat_face(self):
+        # create a box
+        b = get_test_box_shape()
+        # take the first edge
+        t = Topo(b)
+        wire = t.wires().next()
+        my_wire = Wire(wire)
+        assert not my_wire.IsNull()
+        assert my_wire.tolerance == 1e-06
+
+
+class TestVertex(unittest.TestCase):
+    def test_creat_vertex(self):
+        my_vertex = Vertex(1., 2., -2.6)
+        assert my_vertex.tolerance == 1e-06
+        assert my_vertex.x == 1.
+        assert my_vertex.y == 2.
+        assert my_vertex.z == -2.6
+
+
+class TestShell(unittest.TestCase):
+    def test_creat_shell(self):
+        my_shell = Shell(BRepPrimAPI_MakeBox(10, 20, 30).Shell())
+        assert my_shell.tolerance == 1e-06
+
+
+class TestSolid(unittest.TestCase):
+    def test_creat_solid(self):
+        my_solid = Solid(BRepPrimAPI_MakeBox(10, 20, 30).Solid())
+        assert my_solid.tolerance == 1e-06
 
 
 def suite():
     test_suite = unittest.TestSuite()
-    test_suite.addTest(unittest.makeSuite(TestTopo))
-    test_suite.addTest(unittest.makeSuite(TestEdge))
     return test_suite
 
 if __name__ == "__main__":
