def test_add_income(self): account = Account() income = account.add_income(amount=1000, description="Gift", timestamp=datetime(2012, 1, 1)) self.assertIn(income, account.transactions) self.assertEqual(income.amount, Decimal("1000.00")) self.assertEqual(income.description, "Gift") self.assertEqual(income.timestamp, datetime(2012, 1, 1))
def test_balance(self): account = Account() account.add_income(amount=100) result = account.balance expected = Decimal(100) self.assertEqual(expected, result)