Fix coding style problems.

[modules/shaper.git] / src / PythonAPI / Test / TestFeaturesExtrusion.py

diff --git a/src/PythonAPI/Test/TestFeaturesExtrusion.py b/src/PythonAPI/Test/TestFeaturesExtrusion.py
index a58c876629fe17bea43eca2dbcc92eb4582470f3..f561711182d85a73fa4d4b9ccc3dca80419c95bf 100644 (file)
--- a/src/PythonAPI/Test/TestFeaturesExtrusion.py
+++ b/src/PythonAPI/Test/TestFeaturesExtrusion.py
@@ -1,8 +1,27 @@
+# Copyright (C) 2014-2019  CEA/DEN, EDF R&D
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+#
+# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+#
+
 import unittest
 
 import ModelAPI
 
-import model
+from salome.shaper import model
 
 #-----------------------------------------------------------------------------
 # Fixtures
@@ -56,9 +75,9 @@ class FeaturesAddExtrusionTestCase(FeaturesAddExtrusionFixture):
         self.assertEqual(extrusion.toSize().value(), 10)
         self.assertEqual(extrusion.fromSize().value(), 0)
         self.assertEqual(extrusion.toObject().context(), None)
-        self.assertEqual(extrusion.toOffset().value(), 0)
+        self.assertEqual(extrusion.toOffset().isInitialized(), False)
         self.assertEqual(extrusion.fromObject().context(), None)
-        self.assertEqual(extrusion.fromOffset().value(), 0)
+        self.assertEqual(extrusion.fromOffset().isInitialized(), False)
 
     def test_add_extrusion_by_face_and_planes(self):
         # base
@@ -85,8 +104,8 @@ class FeaturesAddExtrusionTestCase(FeaturesAddExtrusionFixture):
                                        from_object, 20)
 
         self.assertEqual(extrusion.creationMethod().value(), "ByPlanesAndOffsets")
-        self.assertEqual(extrusion.toSize().value(), 0)
-        self.assertEqual(extrusion.fromSize().value(), 0)
+        self.assertEqual(extrusion.toSize().isInitialized(), False)
+        self.assertEqual(extrusion.fromSize().isInitialized(), False)
 #         self.assertEqual(extrusion.getToObject().context(),
 #                          to_sketch.result())
         self.assertEqual(extrusion.toOffset().value(), 15)
@@ -126,9 +145,9 @@ class FeaturesExtrusionTestCase(FeaturesExtrusionFixture):
         self.assertEqual(self.extrusion.toSize().value(), 15)
         self.assertEqual(self.extrusion.fromSize().value(), 20)
         self.assertEqual(self.extrusion.toObject().context(), None)
-        self.assertEqual(self.extrusion.toOffset().value(), 0)
+        self.assertEqual(self.extrusion.toOffset().isInitialized(), False)
         self.assertEqual(self.extrusion.fromObject().context(), None)
-        self.assertEqual(self.extrusion.fromOffset().value(), 0)
+        self.assertEqual(self.extrusion.fromOffset().isInitialized(), False)
 
     def test_extrusion_set_planes_and_offsets(self):
         # to
@@ -146,8 +165,12 @@ class FeaturesExtrusionTestCase(FeaturesExtrusionFixture):
 
         to_object = to_sketch.selectFace()[0]
         from_object = from_sketch.selectFace()[0]
+
+        self.part.moveFeature(self.extrusion.feature(), from_sketch.feature())
+
         self.extrusion.setPlanesAndOffsets(to_object, 15, from_object, 20)
 
 
 if __name__ == "__main__":
-    unittest.main()
+    test_program = unittest.main(exit=False)
+    assert test_program.result.wasSuccessful(), "Test failed"