def test_get_roof_square(self):
actual_res = self.house.get_roof_square()
expected_res = 60
assert actual_res == expected_res
house_with_other_roof = House()
house_with_other_roof.create_roof(10, 6, "gable")
actual_res = house_with_other_roof.get_roof_square()
expected_res = 120
assert actual_res == expected_res
def test_roof_square(self):
roof = House()
roof.create_roof(2, 2, "gable")
self.assertEqual(8, roof.get_roof_square())