NPO-Accounting/conservancy_beancount Changeset - 1cbc9d3dc933

Changeset - 1cbc9d3dc933

Parent rev.

Child rev.

[Not reviewed]

0 2 0

Brett Smith - 4 years ago 2020-06-11 17:07:14
brettcsmith@brettcsmith.org

tests: Add _meta_type kwarg to testutil.Posting.

2 files changed with 16 insertions and 33 deletions:

tests/test_reports_related_postings.py

tests/testutil.py

0 comments (0 inline, 0 general)

tests/test_reports_related_postings.py

➞

Show inline comments

@@ ... / @@ -219,102 +219,83 @@ def test_meta_values_all_match(): @@
     ])
     assert related.meta_values('key') == {'1', '2'}
 def test_meta_values_all_match_one_value():
     related = core.RelatedPostings([
         testutil.Posting('Income:Donations', -1, key='1'),
         testutil.Posting('Income:Donations', -2, key='1'),
     ])
     assert related.meta_values('key') == {'1'}
 def test_meta_values_all_match_default_given():
     related = core.RelatedPostings([
         testutil.Posting('Income:Donations', -1, key='1'),
         testutil.Posting('Income:Donations', -2, key='2'),
     ])
     assert related.meta_values('key', '') == {'1', '2'}
 def test_meta_values_many_types():
     expected = {
         datetime.date(2020, 4, 1),
         Decimal(42),
         testutil.Amount(5),
         'rt:42',
+    }
     related = core.RelatedPostings(
         testutil.Posting('Income:Donations', -index, key=value)
         for index, value in enumerate(expected)
+    )
     assert related.meta_values('key') == expected
 @pytest.mark.parametrize('count', range(3))
 def test_all_meta_links_zero(count):
     postings = (
         testutil.Posting('Income:Donations', -n, testkey=str(n))
         for n in range(count)
+    )
     related = core.RelatedPostings(
         post._replace(meta=data.Metadata(post.meta))
         for post in postings
+    )
     related = core.RelatedPostings(testutil.Posting(
         'Income:Donations', -n, testkey=str(n), _meta_type=data.Metadata,
     ) for n in range(count))
     assert next(related.all_meta_links('approval'), None) is None
 def test_all_meta_links_singletons():
     postings = (
         testutil.Posting('Income:Donations', -10, statement=value)
         for value in itertools.chain(
     related = core.RelatedPostings(testutil.Posting(
         'Income:Donations', -10, statement=value, _meta_type=data.Metadata,
     ) for value in itertools.chain(
         testutil.NON_LINK_METADATA_STRINGS,
         testutil.LINK_METADATA_STRINGS,
         testutil.NON_STRING_METADATA_VALUES,
     ))
     related = core.RelatedPostings(
         post._replace(meta=data.Metadata(post.meta))
         for post in postings
+    )
     assert set(related.all_meta_links('statement')) == testutil.LINK_METADATA_STRINGS
 def test_all_meta_links_multiples():
     postings = (
         testutil.Posting('Income:Donations', -10, approval=' '.join(value))
         for value in itertools.permutations(testutil.LINK_METADATA_STRINGS, 2)
+    )
     related = core.RelatedPostings(
         post._replace(meta=data.Metadata(post.meta))
         for post in postings
+    )
     related = core.RelatedPostings(testutil.Posting(
         'Income:Donations', -10, approval=' '.join(value), _meta_type=data.Metadata,
     ) for value in itertools.permutations(testutil.LINK_METADATA_STRINGS, 2))
     assert set(related.all_meta_links('approval')) == testutil.LINK_METADATA_STRINGS
 def test_all_meta_links_preserves_order():
     postings = (
         testutil.Posting('Income:Donations', -10, approval=c)
         for c in '121323'
+    )
     related = core.RelatedPostings(
         post._replace(meta=data.Metadata(post.meta))
         for post in postings
+    )
     related = core.RelatedPostings(testutil.Posting(
         'Income:Donations', -10, approval=c, _meta_type=data.Metadata,
     ) for c in '121323')
     assert list(related.all_meta_links('approval')) == list('123')
 def test_group_by_meta_zero():
     assert not list(core.RelatedPostings.group_by_meta([], 'metacurrency'))
 def test_group_by_meta_one(credit_card_cycle):
     posting = next(post for post in data.Posting.from_entries(credit_card_cycle)
                    if post.account.is_credit_card())
     actual = core.RelatedPostings.group_by_meta([posting], 'metacurrency')
     assert set(key for key, _ in actual) == {'USD'}
 def test_group_by_meta_many(two_accruals_three_payments):
     postings = [post for post in data.Posting.from_entries(two_accruals_three_payments)
                 if post.account == 'Assets:Receivable:Accounts']
     actual = dict(core.RelatedPostings.group_by_meta(postings, 'metacurrency'))
     assert set(actual) == {'USD', 'EUR'}
     for key, group in actual.items():
         assert 2 <= len(group) <= 3
         assert group.balance().is_zero()
 def test_group_by_meta_many_single_posts(two_accruals_three_payments):
     postings = [post for post in data.Posting.from_entries(two_accruals_three_payments)
                 if post.account == 'Assets:Receivable:Accounts']
     actual = dict(core.RelatedPostings.group_by_meta(postings, 'metanumber'))
     assert set(actual) == {post.units.number for post in postings}
     assert len(actual) == len(postings)

tests/testutil.py

➞

Show inline comments

@@ ... / @@ -87,70 +87,72 @@ def combine_values(*value_seqs): @@
             stop = max(stop, len(seq))
         except TypeError:
             pass
     return itertools.islice(
         zip(*(itertools.cycle(seq) for seq in value_seqs)),
         stop,
+    )
 def date_seq(date=FY_MID_DATE, step=1):
     while True:
         yield date
         date += datetime.timedelta(days=step)
 def parse_date(s, fmt='%Y-%m-%d'):
     return datetime.datetime.strptime(s, fmt).date()
 def test_path(s):
     if s is None:
         return s
     s = Path(s)
     if not s.is_absolute():
         s = TESTS_DIR / s
     return s
 def Amount(number, currency='USD'):
     return bc_amount.Amount(Decimal(number), currency)
 def Cost(number, currency='USD', date=FY_MID_DATE, label=None):
     return bc_data.Cost(Decimal(number), currency, date, label)
 def Posting(account, number,
             currency='USD', cost=None, price=None, flag=None,
-            type_=bc_data.Posting, **meta):
+            _post_type=bc_data.Posting, _meta_type=None, **meta):
     if cost is not None:
         cost = Cost(*cost)
     if not meta:
         meta = None
     return type_(
     elif _meta_type:
         meta = _meta_type(meta)
     return _post_type(
         account,
         Amount(number, currency),
         cost,
         price,
         flag,
         meta,
+    )
 def Transaction(date=FY_MID_DATE, flag='*', payee=None,
                 narration='', tags=None, links=None, postings=(),
                 **meta):
     if isinstance(date, str):
         date = parse_date(date)
     meta.setdefault('filename', '<test>')
     meta.setdefault('lineno', 0)
     real_postings = []
     for post in postings:
         try:
             post.account
         except AttributeError:
             if isinstance(post[-1], dict):
                 args = post[:-1]
                 kwargs = post[-1]
             else:
                 args = post
                 kwargs = {}
             post = Posting(*args, **kwargs)
         real_postings.append(post)
     return bc_data.Transaction(
         meta,
         date,
         flag,

0 comments (0 inline, 0 general)