NPO-Accounting/conservancy_beancount File Diff

File diff c6dc2d83aca7 → d66ba8773f5e

tests/test_data_balance_of.py

➞

Show inline comments

@@ ... / @@ -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)