5 from model import WrongNumberOfArguments
7 from TestSketcher import SketcherTestCase
9 class SketcherAddArc(SketcherTestCase):
10 def test_arc_by_coords(self):
11 arc = self.sketch.addArc(0, 1, 0, 0, 1, 1)
13 self.assertEqual(arc.startPoint().x(), 0)
14 self.assertEqual(arc.startPoint().y(), 0)
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)
22 self.assertEqual(arc.startPoint().x(), 0)
23 self.assertEqual(arc.startPoint().y(), 0)
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)
31 def test_modify_arc(self):
32 # Note: arc will modify startPoint and endPoint to be in circle
33 arc = self.sketch.addArc(0, 1, 0, 0, 1, 1)
35 arc.setStartPoint(1, 1)
36 arc.setEndPoint(-1, -1)
38 self.assertEqual(arc.center().x(), 0)
39 self.assertEqual(arc.center().y(), 0)
40 self.assertEqual(arc.startPoint().x(), 1)
41 self.assertEqual(arc.startPoint().y(), 1)
42 self.assertEqual(arc.endPoint().x(), -1)
43 self.assertEqual(arc.endPoint().y(), -1)
46 if __name__ == "__main__":
47 unittest.main(verbosity=2)