src/PythonAPI/Test/TestSketcherAddArc.py

   1 import unittest
   2
   3 import model
   4 import geom
   5
   6 from TestSketcher import SketcherTestCase
   7
   8 class SketcherAddArc(SketcherTestCase):
   9     def test_arc_by_coords(self):
  10         arc = self.sketch.addArc(0, 1, 0, 0, 1, 1, 0)
  11         model.do()
  12         self.assertEqual(arc.startPoint().x(), 0)
  13         self.assertEqual(arc.startPoint().y(), 0)
  14
  15     def test_arc_by_points(self):
  16         center = geom.Pnt2d(0, 1)
  17         start = geom.Pnt2d(0, 0)
  18         end = geom.Pnt2d(1, 1)
  19         arc = self.sketch.addArc(center, start, end, 0)
  20         model.do()
  21         self.assertEqual(arc.startPoint().x(), 0)
  22         self.assertEqual(arc.startPoint().y(), 0)
  23
  24     def test_modify_arc(self):
  25         # Note: arc will modify startPoint and endPoint to be in circle
  26         arc = self.sketch.addArc(0, 1, 0, 0, 1, 1, 0)
  27         arc.setByCenterStartEnd(0, 0, 1, 1, -1, -1, 0)
  28         model.do()
  29         self.assertEqual(arc.center().x(), 0)
  30         self.assertEqual(arc.center().y(), 0)
  31         self.assertEqual(arc.startPoint().x(), 1)
  32         self.assertEqual(arc.startPoint().y(), 1)
  33         self.assertEqual(arc.endPoint().x(), -1)
  34         self.assertEqual(arc.endPoint().y(), -1)
  35
  36
  37 if __name__ == "__main__":
  38     unittest.main(verbosity=2)