def test_zero_size_house_add_room(): h = House(0) assert h.available_space == 0 bedroom = Room('bedroom', 10) with pytest.raises(NotEnoughSpaceError): h.add_rooms(bedroom)
def test_house_tax(): h = House() bedroom = Room('bedroom', 10) kitchen = Room('kitchen', 9) bathroom = Room('bathroom', 3) h.add_rooms(bedroom, kitchen, bathroom) assert h.calculate_tax() == 2200
def test_small_house(): h = House() bedroom = Room('bedroom', 10) kitchen = Room('kitchen', 9) bathroom = Room('bathroom', 3) h.add_rooms(bedroom, kitchen, bathroom) assert h.size() == 22 assert len(h.rooms) == 3 assert str(h) == '''House:
def test_small_house_add_equal_then_more(): h = House(15) assert h.available_space == 15 bedroom = Room('bedroom', 15) h.add_rooms(bedroom) assert h.size() == 15 assert h.available_space == 15 tiny_room = Room('very small closet', 0.01) with pytest.raises(NotEnoughSpaceError): h.add_rooms(tiny_room)
def test_one_neighborhood(): Neighborhood.total_size = 0 n = Neighborhood() houses = [] for i in range(3): h = House() bedroom = Room('bedroom', 10) kitchen = Room('kitchen', 9) bathroom = Room('bathroom', 3) h.add_rooms(bedroom, kitchen, bathroom) houses.append(h) n.add_houses(*houses) assert n.size() == Neighborhood.total_size
def test_palace(): h = House(175) for i in range(10): h.add_rooms(Room(f'bedroom {i}', 15)) for i in range(5): h.add_rooms(Room(f'bathroom {i}', 3)) for i in range(3): h.add_rooms(Room(f'kitchen {i}', 3)) assert h.size() == 174 assert len(h.rooms) == 18