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_budget(self):
account = Account()
budget = account.add_budget(name="Test Budget", interval=MONTHLY, limit=1000)
self.assertIn(budget, account.budgets)
self.assertEqual(budget.name, "Test Budget")
self.assertEqual(budget.interval, MONTHLY)
self.assertEqual(budget.limit, Decimal("1000.00"))
