diff --git a/tests/testutil.py b/tests/testutil.py index 7fc38156f5e148911f003ddabf181d25ebb74bc2..187b53ea7221706bd6ac4e42f46a78494832d1b6 100644 --- a/tests/testutil.py +++ b/tests/testutil.py @@ -27,6 +27,15 @@ FY_START_DATE = datetime.date(2020, 3, 1) FY_MID_DATE = datetime.date(2020, 9, 1) PAST_DATE = datetime.date(2000, 1, 1) +def check_post_meta(txn, *expected_meta, default=None): + assert len(txn.postings) == len(expected_meta) + for post, expected in zip(txn.postings, expected_meta): + if not expected: + assert not post.meta + else: + assert all(post.meta.get(key, default) == value + for key, value in expected.items()) + def parse_date(s, fmt='%Y-%m-%d'): return datetime.datetime.strptime(s, fmt).date()