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