def test_charge_for_three():
charge = ChargeFactory()
for_three = charge.get_charge_for_amount_of_buildings(3)
assert for_three == charge.three_apartments
def test_charge_for_one():
charge = ChargeFactory()
for_one = charge.get_charge_for_amount_of_buildings(1)
assert for_one == charge.one_apartments
