Merge pull request #70 from ggurioli/branch_tests

Adding tests in test_shaft.py

Author: mtezzele

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_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_exception(self):
+    	sh = Shaft("tests/test_datasets/parameters.prm")
+    	with self.assertRaises(Exception):
+            sh.generate_solid()
+
+    def test_init(self):
+        sh = Shaft("tests/test_datasets/shaft.iges")
+        assert sh.filename == "tests/test_datasets/shaft.iges"