diff --git a/tests/test_data_balance_of.py b/tests/test_data_balance_of.py index 91e63b7d68780c7cefe7986c57531eda5a93452f..6a44fd6b6f535a00bb31722d09f9efb22bf3a13b 100644 --- a/tests/test_data_balance_of.py +++ b/tests/test_data_balance_of.py @@ -24,6 +24,7 @@ from . import testutil from conservancy_beancount import data is_cash_eq = data.Account.is_cash_equivalent +USD = testutil.Amount @pytest.fixture def payable_payment_txn(): @@ -34,6 +35,14 @@ def payable_payment_txn(): ('Assets:Checking', -5), ]) +@pytest.fixture +def fx_donation_txn(): + return testutil.Transaction(postings=[ + ('Income:Donations', -500, 'EUR', ('.9', 'USD')), + ('Assets:Checking', 445), + ('Expenses:BankingFees', 5), + ]) + def balance_under(txn, *accts): pred = methodcaller('is_under', *accts) return data.balance_of(txn, pred) @@ -43,17 +52,18 @@ def test_balance_of_simple_txn(): ('Assets:Cash', 50), ('Income:Donations', -50), ]) - assert balance_under(txn, 'Assets') == 50 - assert balance_under(txn, 'Income') == -50 + assert balance_under(txn, 'Assets') == USD(50) + assert balance_under(txn, 'Income') == USD(-50) def test_zero_balance_of(payable_payment_txn): - assert balance_under(payable_payment_txn, 'Equity') == 0 - assert balance_under(payable_payment_txn, 'Assets:Cash') == 0 - assert balance_under(payable_payment_txn, 'Liabilities:CreditCard') == 0 + expected = testutil.Amount(0, '') + assert balance_under(payable_payment_txn, 'Equity') == expected + assert balance_under(payable_payment_txn, 'Assets:Cash') == expected + assert balance_under(payable_payment_txn, 'Liabilities:CreditCard') == expected def test_nonzero_balance_of(payable_payment_txn): - assert balance_under(payable_payment_txn, 'Assets', 'Expenses') == -50 - assert balance_under(payable_payment_txn, 'Assets', 'Liabilities') == -5 + assert balance_under(payable_payment_txn, 'Assets', 'Expenses') == USD(-50) + assert balance_under(payable_payment_txn, 'Assets', 'Liabilities') == USD(-5) def test_multiarg_balance_of(): txn = testutil.Transaction(postings=[ @@ -61,7 +71,12 @@ def test_multiarg_balance_of(): ('Expenses:BankingFees', 5), ('Assets:Checking', -655), ]) - assert data.balance_of(txn, is_cash_eq, data.Account.is_credit_card) == -5 + assert data.balance_of(txn, is_cash_eq, data.Account.is_credit_card) == USD(-5) def test_balance_of_multipost_txn(payable_payment_txn): - assert data.balance_of(payable_payment_txn, is_cash_eq) == -55 + assert data.balance_of(payable_payment_txn, is_cash_eq) == USD(-55) + +def test_balance_of_multicurrency_txn(fx_donation_txn): + assert balance_under(fx_donation_txn, 'Income') == USD(-450) + assert balance_under(fx_donation_txn, 'Income', 'Assets') == USD(-5) + assert balance_under(fx_donation_txn, 'Income', 'Expenses') == USD(-445)