class RouteTestCase(unittest.TestCase):
def setUp(self):
ids = ['a', 'b', 'c', 'd', 'e']
lat = [10, 20, 15, 40, 50]
lon = [15, 30, 22.5, 60, 75]
self.routes = [1, 1, 2, 2, 3]
self.stops = [Stop(ids[i], lat[i], lon[i]) for i in range(len(ids))]
s1 = [self.stops[0], self.stops[1]]
print(s1)
s2 = [self.stops[2], self.stops[3]]
self.m1 = MegaStop(ids[0], s1)
self.m2 = MegaStop(ids[1], s2)
self.route = Route(1, [self.m1, self.m2])
def test_routes_build(self):
self.assertEqual(self.route.stops[self.m1.id],
self.m1,
msg="Built Stop Dictionary Error")
def test_get_stop(self):
self.assertEqual(self.m1, self.route.get_stop(self.m1.id))
