def read_entities(
input_dir: str
) -> Tuple[Dict[str, Rider], Dict[str, Vehicle], Dict[str, Depot], Params]:
"""Method to parse the Riders, Vehicles and Depots from JSON to Dict"""
riders_file = RIDERS_FILE.format(input_dir=input_dir)
with open(riders_file) as f:
logging.info(f'Read riders from {riders_file}.')
riders_dicts = json.load(f)
riders = {
r_dict['rider_id']: Rider.from_dict(r_dict)
for r_dict in riders_dicts
}
logging.info(f'Successfully parsed {len(riders)} riders.')
vehicles_file = VEHICLES_FILE.format(input_dir=input_dir)
with open(vehicles_file) as f:
vehicles_dicts = json.load(f)
logging.info(f'Read vehicles from {vehicles_file}.')
vehicles = {
v_dict['vehicle_id']: Vehicle.from_dict(v_dict)
for v_dict in vehicles_dicts
}
logging.info(f'Successfully parsed {len(vehicles)} vehicles.')
depots_file = DEPOTS_FILE.format(input_dir=input_dir)
with open(depots_file) as f:
depots_dicts = json.load(f)
logging.info(f'Read depots from {depots_file}.')
depots = {
d_dict['depot_id']: Depot.from_dict(d_dict)
for d_dict in depots_dicts
}
logging.info(f'Successfully parsed {len(depots)} depots.')
params_file = PARAMS_FILE.format(input_dir=input_dir)
with open(params_file) as f:
logging.info(f'Read params from {params_file}.')
params_dict = json.load(f)
params = Params.from_dict(params_dict)
logging.info(f'Successfully parsed {len(params_dict)} params.')
return riders, vehicles, depots, params
def _assert_vehicle_fields(self, vehicle_dict: Dict[str, Any]):
"""Auxiliary method to assert fields in the Vehicle are correct"""
vehicle = Vehicle.from_dict(vehicle_dict)
self.assertEqual(
vehicle.capacity,
vehicle_dict['capacity'],
msg='The Vehicle is instantiated with incorrect capacity.')
self.assertEqual(
vehicle.start,
vehicle_dict['start'],
msg='The Vehicle is instantiated with incorrect start Depot.')
self.assertEqual(
vehicle.end,
vehicle_dict['end'],
msg='The Vehicle is instantiated with incorrect end Depot.')
self.assertEqual(
vehicle.vehicle_id,
vehicle_dict['vehicle_id'],
msg='The Vehicle is instantiated with incorrect vehicle_id.')
