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_small_house_with_plus():
h = House()
bedroom = Room('bedroom', 10)
kitchen = Room('kitchen', 9)
bathroom = Room('bathroom', 3)
h += bedroom
h += kitchen
h += bathroom
assert h.size() == 22
assert len(h.rooms) == 3
assert str(h) == '''House:
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
def test_empty_house():
h = House()
assert h.size() == 0
def test_empty_house():
h = House()
assert h.size() == 0
assert h.available_space == 100
