[PythonAPI / sketcher] Added methods for sketch features modification

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

diff --git a/src/PythonAPI/Test/TestSketcherAddLine.py b/src/PythonAPI/Test/TestSketcherAddLine.py
index 41322e80fbbe59852ab3de39e38b4ed3ea6b2a54..fa4ed94c666e029451f39d47be6711abb3617b35 100644 (file)
--- a/src/PythonAPI/Test/TestSketcherAddLine.py
+++ b/src/PythonAPI/Test/TestSketcherAddLine.py
@@ -1,13 +1,24 @@
 import unittest
-import modeler
+import model
 from TestSketcher import SketcherTestCase
 
-class SketcherAddLineTestCase(SketcherTestCase):    
-    def runTest(self):
+class SketcherAddLine(SketcherTestCase):    
+    def test_add_line(self):
         line = self.sketch.addLine(0, 0, 0, 1)
-        self.assertEqual(line.startPointData().x(), line.endPointData().x())        
-        self.assertNotEqual(line.startPointData().y(), line.endPointData().y())
-        
-    
+        model.do()
+        self.assertEqual(line.getStartPoint().x(), line.getEndPoint().x())        
+        self.assertNotEqual(line.getStartPoint().y(), line.getEndPoint().y())
+
+    def test_modify_line(self):
+        line = self.sketch.addLine(0, 0, 0, 1)
+        model.do()
+        line.setStartPoint(0, 1)
+        line.setEndPoint(1, 1)
+        self.assertEqual(line.getStartPoint().x(), 0)
+        self.assertEqual(line.getStartPoint().y(), 1)
+        self.assertEqual(line.getEndPoint().x(), 1)
+        self.assertEqual(line.getEndPoint().y(), 1)
+
+
 if __name__ == "__main__":
-    unittest.main()
\ No newline at end of file
+    unittest.main(verbosity=2)
\ No newline at end of file