NPO-Accounting/conservancy_beancount Files · tests/test_data_balance_of.py

Files @ 5784068904e8

Branch filter:

Location: NPO-Accounting/conservancy_beancount/tests/test_data_balance_of.py

5784068904e8 2.4 KiB text/x-python Show Annotation Show as Raw Download as Raw

bkuhn

payroll-type — US:403b:Employee:Roth — needed separate since taxable

Since Roth contributions are taxable, there are some reports that
need to include these amounts in total salary (i.e., when running a
report that seeks to show total taxable income for an employee). As
such, we need a `payroll-type` specifically for Roth 403(b)
contributions.

"""Test data.balance_of function"""
# Copyright © 2020  Brett Smith
# License: AGPLv3-or-later WITH Beancount-Plugin-Additional-Permission-1.0
#
# Full copyright and licensing details can be found at toplevel file
# LICENSE.txt in the repository.

from decimal import Decimal
from operator import methodcaller

import pytest

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():
    return testutil.Transaction(postings=[
        ('Liabilities:Payable:Accounts', 50),
        ('Assets:Checking', -50),
        ('Expenses:BankingFees', 5),
        ('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)

def test_balance_of_simple_txn():
    txn = testutil.Transaction(postings=[
        ('Assets:Cash', 50),
        ('Income:Donations', -50),
    ])
    assert balance_under(txn, 'Assets') == USD(50)
    assert balance_under(txn, 'Income') == USD(-50)

def test_zero_balance_of(payable_payment_txn):
    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') == USD(-50)
    assert balance_under(payable_payment_txn, 'Assets', 'Liabilities') == USD(-5)

def test_multiarg_balance_of():
    txn = testutil.Transaction(postings=[
        ('Liabilities:CreditCard', 650),
        ('Expenses:BankingFees', 5),
        ('Assets:Checking', -655),
    ])
    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) == 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)