def test_add_segment_after(self): c = (5, 5) d = (10, 10) a = (0, 0) b = (4, 4) segment = LineSegment(a, b) to_add = LineSegment(c, d) segment.add_segment(to_add) self.assertEqual(segment.a, a) self.assertEqual(segment.b, d)
def test_add_segment_middle(self): a = (0, 0) b = (10, 10) c = (2, 2) d = (4, 4) segment = LineSegment(a, b) to_add = LineSegment(c, d) segment.add_segment(to_add) self.assertEqual(segment.a, a) self.assertEqual(segment.b, b)
def test_add_segment_before(self): a = (5, 5) b = (10, 10) c = (0, 0) d = (4, 4) segment = LineSegment(a, b) to_add = LineSegment(c, d) segment.add_segment(to_add) self.assertEqual(segment.a, c) self.assertEqual(segment.b, b)