src/PythonAPI/Test/TestSketcherAddArc.py

   1 import unittest
   2
   3 import model
   4 import geom
   5 from model import WrongNumberOfArguments
   6
   7 from TestSketcher import SketcherTestCase
   8
   9 class SketcherAddArc(SketcherTestCase):
  10     def test_arc_by_coords(self):
  11         arc = self.sketch.addArc(0, 1, 0, 0, 1, 1)
  12         model.do()
  13         self.assertEqual(arc.startPointData().x(), 0)
  14         self.assertEqual(arc.startPointData().y(), 0)
  15
  16     def test_arc_by_points(self):
  17         center = geom.Pnt2d(0, 1)
  18         start = geom.Pnt2d(0, 0)
  19         end = geom.Pnt2d(1, 1)
  20         arc = self.sketch.addArc(center, start, end)
  21         model.do()
  22         self.assertEqual(arc.startPointData().x(), 0)
  23         self.assertEqual(arc.startPointData().y(), 0)
  24
  25     def test_number_of_args(self):
  26         with self.assertRaises(WrongNumberOfArguments):
  27             self.sketch.addArc(0, 1, 1, 1)
  28         with self.assertRaises(WrongNumberOfArguments):
  29             self.sketch.addArc(0, 1)
  30
  31     def test_modify_arc(self):
  32         arc = self.sketch.addArc(0, 1, 0, 0, 1, 1)
  33         arc.setCenter(0, 0)
  34         arc.setStartPoint(-1.0, 0)
  35         arc.setEndPoint(0, 1.0)
  36         model.do()
  37         self.assertEqual(arc.centerData().x(), 0)
  38         self.assertEqual(arc.centerData().y(), 0)
  39         self.assertEqual(arc.startPointData().x(), -1)
  40         self.assertEqual(arc.startPointData().y(), 0)
  41         self.assertEqual(arc.endPointData().x(), 0)
  42         self.assertEqual(arc.endPointData().y(), 1)
  43
  44
  45 if __name__ == "__main__":
  46     unittest.main(verbosity=2)