class BuildingTestCase(unittest.TestCase):
building = None
def setUp(self):
self.building = Building(5, 3)
def test_get_elevator(self):
direction = 1
self.assertIsInstance(
self.building.get_elevator(self.building.elevators, direction),
Elevator)
direction = -1
self.assertIsInstance(
self.building.get_elevator(self.building.elevators, direction),
Elevator)
def test_call_elevator(self):
from_floor = 3
direction = 1
# Nearest elevator arrives to floor 'from_floor"
self.assertIsInstance(
self.building.call_elevator(from_floor, direction), Elevator)
def test_move_elevator(self):
to_floor = 5
elevator = self.building.elevators[0]
self.building.move_elevator(elevator, to_floor)
self.assertEqual(to_floor, elevator.floor)
to_floor = 6
self.assertNotEqual(to_floor, elevator.floor)
def test_validate_floor_elevator_direction(self):
to_floor = 3
elevator = self.building.elevators[0]
elevator.floor = 4
elevator.direction = 1
self.assertRaises(ValueError,
self.building.validate_floor_elevator_direction,
elevator, to_floor)
to_floor = 5
self.assertEqual(
None,
self.building.validate_floor_elevator_direction(
elevator, to_floor))
def test_validate_direction(self):
from_floor = 0
direction = -1
self.assertRaises(ValueError, self.building.validate_direction,
from_floor, direction)
direction = 1
self.assertEqual(
None, self.building.validate_direction(from_floor, direction))
from_floor = 5
self.assertRaises(ValueError, self.building.validate_direction,
from_floor, direction)
direction = -1
self.assertEqual(
None, self.building.validate_direction(from_floor, direction))
def test_add_request_elevators(self):
from_floor = 4
to_floor = 5
self.building.add_request(from_floor, to_floor)
self.assertEqual(1, len(self.building.requests))
from_floor = 4
to_floor = 1
self.building.add_request(from_floor, to_floor)
self.assertEqual(2, len(self.building.requests))
from_floor = 3
to_floor = 4
self.building.add_request(from_floor, to_floor)
self.assertEqual(3, len(self.building.requests))
def test_move_passengers(self):
for _ in range(6):
from_floor, to_floor = random.sample(
range(len(self.building.floors)), 2)
request = Requests_Elevators(from_floor, to_floor)
print(request.__str__())
self.building.requests.append(request)
self.building.move_passengers()
self.assertEqual(0, len(self.building.requests))
def test_move_passenger_demo(self):
from_floor = 0
to_floor = 4
self.building.add_request(from_floor, to_floor)
from_floor = 0
to_floor = 4
self.building.add_request(from_floor, to_floor)
from_floor = 0
to_floor = 4
self.building.add_request(from_floor, to_floor)
from_floor = 0
to_floor = 3
self.building.add_request(from_floor, to_floor)
from_floor = 4
to_floor = 0
self.building.add_request(from_floor, to_floor)
from_floor = 2
to_floor = 3
self.building.add_request(from_floor, to_floor)
from_floor = 2
to_floor = 3
self.building.add_request(from_floor, to_floor)
self.building.move_passengers()
self.assertEqual(0, len(self.building.requests))
def tearDown(self):
self.building = None
