def test_calculate_length_many_depots(self):
depots = self.depots[0:3]
route = Route(0, depots)
length = route.calculate_length(self.distance_matrix)
self.assertEqual(length, 21)
def test_calculate_length_one_depot(self):
depots = self.depots[0:1]
route = Route(0, depots)
length = route.calculate_length(self.distance_matrix)
self.assertTrue(length == 2)