Add a unit test for get_train_routes func.