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 print_status(self):
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()