def test_print_status(self):
room = Room(1, 3)
self.hotel.add_room(room)
self.hotel.take_room(1, 3)
res = self.hotel.status().strip()
actual = 'Hotel Some Hotel has 3 total guests\nFree rooms: \nTaken rooms: 1'
self.assertEqual(res, actual)
def test_free_room(self):
room = Room(1, 3)
self.hotel.add_room(room)
self.hotel.take_room(1, 3)
self.hotel.free_room(1)
self.assertEqual(self.hotel.guests, 0)
self.assertEqual(self.hotel.rooms[0].is_taken, False)
self.assertEqual(self.hotel.rooms[0].guests, 0)
def setUp(self):
self.room = Room(1, 3)
self.hotel = Hotel("Some Hotel")
class Tests(unittest.TestCase):
def setUp(self):
self.room = Room(1, 3)
self.hotel = Hotel("Some Hotel")
def test_init_creates_all_attributes(self):
self.assertEqual(self.room.number, 1)
self.assertEqual(self.room.capacity, 3)
self.assertEqual(self.room.guests, 0)
self.assertEqual(self.room.is_taken, False)
def test_take_room_success(self):
self.room.take_room(2)
self.assertEqual(self.room.is_taken, True)
self.assertEqual(self.room.guests, 2)
def test_take_room_not_enough_capacity(self):
result = self.room.take_room(4)
self.assertEqual(self.room.is_taken, False)
self.assertEqual(self.room.guests, 0)
self.assertEqual(result, "Room number 1 cannot be taken")
def test_take_room_not_free(self):
self.room.take_room(1)
result = self.room.take_room(1)
self.assertEqual(self.room.is_taken, True)
self.assertEqual(self.room.guests, 1)
self.assertEqual(result, "Room number 1 cannot be taken")
def test_free_room_success(self):
self.room.take_room(1)
self.room.free_room()
self.assertEqual(self.room.is_taken, False)
self.assertEqual(self.room.guests, 0)
def test_free_room_not_taken(self):
result = self.room.free_room()
self.assertEqual(self.room.is_taken, False)
self.assertEqual(self.room.guests, 0)
self.assertEqual(result, "Room number 1 is not taken")
def test_init_creates_all_attributes(self):
self.assertEqual(self.hotel.name, "Some Hotel")
self.assertEqual(self.hotel.rooms, [])
self.assertEqual(self.hotel.guests, 0)
def test_class_methods_creates_a_hotel(self):
hotel = Hotel.from_stars(3)
self.assertEqual(hotel.name, "3 stars Hotel")
self.assertEqual(self.hotel.rooms, [])
self.assertEqual(self.hotel.guests, 0)
def test_add_room(self):
room = Room(1, 3)
self.hotel.add_room(room)
self.assertEqual(self.hotel.rooms, [room])
def test_take_room(self):
room = Room(1, 3)
self.hotel.add_room(room)
self.hotel.take_room(1, 3)
self.assertEqual(self.hotel.rooms[0].is_taken, True)
def test_free_room(self):
room = Room(1, 3)
self.hotel.add_room(room)
self.hotel.take_room(1, 3)
self.hotel.free_room(1)
self.assertEqual(self.hotel.guests, 0)
self.assertEqual(self.hotel.rooms[0].is_taken, False)
self.assertEqual(self.hotel.rooms[0].guests, 0)
def test_print_status(self):
room = Room(1, 3)
self.hotel.add_room(room)
self.hotel.take_room(1, 3)
res = self.hotel.status().strip()
actual = 'Hotel Some Hotel has 3 total guests\nFree rooms: \nTaken rooms: 1'
self.assertEqual(res, actual)
def test_take_room(self):
room = Room(1, 3)
self.hotel.add_room(room)
self.hotel.take_room(1, 3)
self.assertEqual(self.hotel.rooms[0].is_taken, True)
def test_add_room(self):
room = Room(1, 3)
self.hotel.add_room(room)
self.assertEqual(self.hotel.rooms, [room])
rooms_free = ', '.join(
str(room.number) for room in self.rooms if room.is_taken)
rooms_taken = ', '.join(
str(room.number) for room in self.rooms if not room.is_taken)
message = f'''Hotel {self.name} has {self.guests} total guests
Free rooms: {rooms_free}
Taken rooms: {rooms_taken}'''
print(message)
from project.hotel import Hotel
from project.room import Room
hotel = Hotel.from_stars(5)
first_room = Room(1, 3)
second_room = Room(2, 2)
third_room = Room(3, 1)
hotel.add_room(first_room)
hotel.add_room(second_room)
hotel.add_room(third_room)
hotel.take_room(1, 4)
hotel.take_room(1, 2)
hotel.take_room(3, 1)
hotel.take_room(3, 1)
hotel.print_status()
def test_print_status(self, mock_print):
room = Room(1, 3)
self.hotel.add_room(room)
self.hotel.take_room(1, 3)
self.hotel.print_status()
mock_print.assert_called_with("Taken rooms: 1")
class Tests(unittest.TestCase):
def setUp(self):
self.room = Room(1, 3)
self.hotel = Hotel("Some Hotel")
def test_init_creates_all_attributes(self):
self.assertEqual(self.room.number, 1)
self.assertEqual(self.room.capacity, 3)
self.assertEqual(self.room.guests, 0)
self.assertEqual(self.room.is_taken, False)
def test_take_room_success(self):
self.room.take_room(2)
self.assertEqual(self.room.is_taken, True)
self.assertEqual(self.room.guests, 2)
def test_take_room_not_enough_capacity(self):
result = self.room.take_room(4)
self.assertEqual(self.room.is_taken, False)
self.assertEqual(self.room.guests, 0)
self.assertEqual(result, "Room number 1 cannot be taken")
def test_take_room_not_free(self):
self.room.take_room(1)
result = self.room.take_room(1)
self.assertEqual(self.room.is_taken, True)
self.assertEqual(self.room.guests, 1)
self.assertEqual(result, "Room number 1 cannot be taken")
def test_free_room_success(self):
self.room.take_room(1)
self.room.free_room()
self.assertEqual(self.room.is_taken, False)
self.assertEqual(self.room.guests, 0)
def test_free_room_not_taken(self):
result = self.room.free_room()
self.assertEqual(self.room.is_taken, False)
self.assertEqual(self.room.guests, 0)
self.assertEqual(result, "Room number 1 is not taken")
def test_init_creates_all_attributes(self):
self.assertEqual(self.hotel.name, "Some Hotel")
self.assertEqual(self.hotel.rooms, [])
self.assertEqual(self.hotel.guests, 0)
def test_class_methods_creates_a_hotel(self):
hotel = Hotel.from_stars(3)
self.assertEqual(hotel.name, "3 stars Hotel")
self.assertEqual(self.hotel.rooms, [])
self.assertEqual(self.hotel.guests, 0)
def test_add_room(self):
room = Room(1, 3)
self.hotel.add_room(room)
self.assertEqual(self.hotel.rooms, [room])
def test_take_room(self):
room = Room(1, 3)
self.hotel.add_room(room)
self.hotel.take_room(1, 3)
self.assertEqual(self.hotel.rooms[0].is_taken, True)
def test_free_room(self):
room = Room(1, 3)
self.hotel.add_room(room)
self.hotel.take_room(1, 3)
self.hotel.free_room(1)
self.assertEqual(self.hotel.guests, 0)
self.assertEqual(self.hotel.rooms[0].is_taken, False)
self.assertEqual(self.hotel.rooms[0].guests, 0)
@mock.patch('builtins.print')
def test_print_status(self, mock_print):
room = Room(1, 3)
self.hotel.add_room(room)
self.hotel.take_room(1, 3)
self.hotel.print_status()
mock_print.assert_called_with("Taken rooms: 1")
