Adding tests in test_shaft.py

Author: ggurioli

tests/test_shaft.py

@@ -18,15 +18,19 @@ def test_generate_solid_02(self):
         shaft_solid = sh.generate_solid()
         self.assertIsInstance(shaft_solid, TopoDS_Solid)
 
+    def test_display_01(self):
+        sh = Shaft("tests/test_datasets/shaft.iges")
+        sh.display()
+
     def test_exception(self):
     	sh = Shaft("tests/test_datasets/parameters.prm")
     	with self.assertRaises(Exception):
             sh.generate_solid()
 
-    def test_display_01(self):
-        sh = Shaft("tests/test_datasets/shaft.iges")
-        sh.display()
-
     def test_display_02(self):
         sh = Shaft("tests/test_datasets/shaft.stl")
-        sh.display()
+        sh.display()
+
+    def test_init(self):
+        sh = Shaft("tests/test_datasets/shaft.iges")
+        assert sh.filename == "data/shaft.iges"