def test_move_elevator_with_two_items(self):
start = SimulatorState(0)
start.add_item_to_floor('HG', 2)
start.add_item_to_floor('HM', 0)
start.add_item_to_floor('NG', 2)
start.add_item_to_floor('NM', 0)
expected_floor_map = [
{
'generator': {
'floor': 2,
'name': 'HG'
},
'microchip': {
'floor': 1,
'name': 'HM'
}
},
{
'generator': {
'floor': 2,
'name': 'NG'
},
'microchip': {
'floor': 1,
'name': 'NM'
}
}
]
rad_sim = RadiationContainmentSimulator(start)
rad_sim.elevator_up(['NM', 'HM'])
self.assert_simulator_state(rad_sim.get_simulator_state(), 1, expected_floor_map)
def advance_simulator_down(state, cargo) -> SimulatorState:
rad_sim = RadiationContainmentSimulator(state)
rad_sim.elevator_down(cargo)
return rad_sim.get_simulator_state()