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)
