NPO-Accounting/conservancy_beancount
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

data.Account: Add is_real_asset method.

Browse source
This commit is contained in:
Brett Smith 2020-03-18 08:59:03 -04:00
parent 661fe676e1
commit 28238643a3
2 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
conservancy_beancount/data.py
View file

@ -35,6 +35,12 @@ from .beancount_types import (
class Account(str): class Account(str):
SEP = bc_account.sep SEP = bc_account.sep
def is_real_asset(self) -> bool:
return (
self.is_under('Assets:')
and not self.is_under('Assets:PrepaidExpenses')
)
def is_under(self, acct_s: str) -> bool: def is_under(self, acct_s: str) -> bool:
return self.startswith(acct_s) and ( return self.startswith(acct_s) and (
acct_s.endswith(self.SEP) acct_s.endswith(self.SEP)

13
tests/test_data_account.py
View file

@ -30,3 +30,16 @@ from conservancy_beancount import data
]) ])
def test_is_under(acct_name, under_arg, expected): def test_is_under(acct_name, under_arg, expected):
assert data.Account(acct_name).is_under(under_arg) == expected assert data.Account(acct_name).is_under(under_arg) == expected
@pytest.mark.parametrize('acct_name,expected', [
('Accrued:AccountsPayable', False),
('Accrued:AccountsReceivable', False),
('Assets:Cash', True),
('Assets:Cash:EUR', True),
('Assets:PrepaidExpenses', False),
('Assets:Bank:Checking', True),
('Expenses:General', False),
('Income:Donations', False),
])
def test_is_real_asset(acct_name, expected):
assert data.Account(acct_name).is_real_asset() == expected