def test_get_budget_totals(self): account = Account() account.add_budget("Groceries", interval=MONTHLY, limit=1000) account.add_purchase(100, category="Groceries") result = account.get_budget_totals() expected = {"Groceries": Decimal(100)} self.assertEqual(result, expected)
def test_add_purchase(self): account = Account() purchase = account.add_purchase(1000, description="Morning Coffee") self.assertIn(purchase, account.transactions) self.assertEqual(purchase.amount, Decimal("1000.00")) self.assertEqual("Morning Coffee", purchase.description)