def test_budget_attribute_is_set_to_None(self): """testing if a TypeError will ve raised when the budget attribute is set to None """ from stalker import Invoice test_invoice = Invoice(budget=self.test_budget, client=self.test_client, amount=1500, unit='TRY') with pytest.raises(TypeError) as cm: test_invoice.budget = None assert str(cm.value) == \ 'Invoice.budget should be a Budget instance, not NoneType'
def test_budget_attribute_is_set_to_a_value_other_than_a_budget_instance( self): """testing if a TypeError will be raised when the Budget attribute is set to a value other than a Budget instance """ from stalker import Invoice test_invoice = Invoice(budget=self.test_budget, client=self.test_client, amount=1500, unit='TRY') with pytest.raises(TypeError) as cm: test_invoice.budget = 'Not a budget instance' assert str(cm.value) == \ 'Invoice.budget should be a Budget instance, not str'