def test_should_not_stop(self):
lift = LiftLogic()
self.assertEqual(False, lift.shouldStop(3, 2, [1, 2]))
def test_should_stop_for_called_floors(self):
lift = LiftLogic()
self.assertEqual(True, lift.shouldStop(1, None, [3, 1]))
def test_should_stop_for_both(self):
lift = LiftLogic()
self.assertEqual(True, lift.shouldStop(2, 2, [3, 2]))
def test_should_stop_for_asked_floor(self):
lift = LiftLogic()
self.assertEqual(True, lift.shouldStop(1, 1, []))