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()