def test_smart_travel():
lift = Lift(queues, capacity)
lift.direction_of_travel = 'up'
lift.smart_travel()
assert lift.current_floor == 4
lift.direction_of_travel = 'down'
lift.smart_travel()
assert lift.current_floor == 2
def test_passengers_pickup_no_queue():
lift = Lift(queues, 2)
lift.direction_of_travel = 'down'
lift.current_floor = 1
floor_queue = lift.queues[1]
lift.passengers_pickup()
assert lift.occupants == []
def test_passengers_pickup_going_down():
lift = Lift(queues, 2)
lift.direction_of_travel = 'down'
lift.current_floor = 4
floor_queue = lift.queues[4]
lift.passengers_pickup()
assert lift.occupants == [3]
assert lift.queues == {0:(), 1:(), 2: (5,5,5), 3:(), 4:(), 5:(), 6:()}
