def test_zero_wall(self):
with pytest.raises(ValueError):
zero_house = House()
zero_house.create_wall(0, 0)
def test_get_count_and_sum_of_walls(self):
wall = House()
wall_meter = [wall.create_wall(2, 2), wall.create_wall(2, 3), wall.create_wall(5, 2), wall.create_wall(5, 5)]
self.assertEqual(4, wall.get_count_of_walls())
self.assertEqual(45, wall.get_walls_square())