def test_raises_get_route_exceptions(self): """ raises get_route exceptions from transport network class :return: """ with self.assertRaises(exceptions.RouteIdNotFoundException): graph_obj = graph.Graph.build_from_parameters(5, 1000, 0.5, 2) network = TransportNetwork(graph_obj) network.get_route("r1")
def test_add_route_transport_network(self): """ to test add_route method of transport_network class :return: """ graph_obj = graph.Graph.build_from_parameters(5, 1000, 0.5, 2) mode_manager = TransportModeManager() bus_obj = mode_manager.get_mode("bus") network = TransportNetwork(graph_obj) route = Route("r1", bus_obj, "1,2,0,4,3", "3,4,0,2,1", "1,0,3", "3,0,1") network.add_route(route) self.assertTrue(isinstance(network.get_route("r1"), Route)) self.assertEqual(len(network.get_routes()), 1) with self.assertRaises(exceptions.RouteIsNotvalidException): network.add_route("route") with self.assertRaises(exceptions.NodeSequencesIsNotValidException): route = Route("r2", bus_obj, "1,3,0,4,3", "3,4,0,2,1", "1,0,3", "3,0,1") network.add_route(route) with self.assertRaises(exceptions.RouteIdDuplicatedException): route = Route("r1", bus_obj, "1,2,0,4,3", "3,4,0,2,1", "1,0,3", "3,0,1") network.add_route(route)