NPO-Accounting/conservancy_beancount Changeset - 9fef177d2dc6

Changeset - 9fef177d2dc6

Parent rev.

Child rev.

[Not reviewed]

0 2 0

Brett Smith - 4 years ago 2020-04-29 14:12:08
brettcsmith@brettcsmith.org

reports: Add RelatedPostings.all_meta_links() method.

2 files changed with 46 insertions and 0 deletions:

conservancy_beancount/reports/core.py

tests/test_reports_related_postings.py

0 comments (0 inline, 0 general)

conservancy_beancount/reports/core.py

➞

Show inline comments

@@ ... / @@ -131,24 +131,33 @@ class RelatedPostings(Sequence[data.Posting]): @@
     ) -> Union[data.Posting, Sequence[data.Posting]]:
         if isinstance(index, slice):
             raise NotImplementedError("RelatedPostings[slice]")
         else:
             return self._postings[index]
     def __len__(self) -> int:
         return len(self._postings)
     def add(self, post: data.Posting) -> None:
         self._postings.append(post)
     def all_meta_links(self, key: MetaKey) -> Set[str]:
         retval: Set[str] = set()
         for post in self:
             try:
                 retval.update(post.meta.get_links(key))
             except TypeError:
                 pass
         return retval
     def clear(self) -> None:
         self._postings.clear()
     def iter_with_balance(self) -> Iterator[Tuple[data.Posting, Balance]]:
         balance = MutableBalance()
         for post in self:
             balance.add_amount(post.units)
             yield post, balance
     def balance(self) -> Balance:
         for _, balance in self.iter_with_balance():
             pass

tests/test_reports_related_postings.py

➞

Show inline comments

@@ ... / @@ -187,24 +187,61 @@ def test_meta_values_all_match_default_given(): @@
 def test_meta_values_many_types():
     expected = {
         datetime.date(2020, 4, 1),
         Decimal(42),
         testutil.Amount(5),
         'rt:42',
+    }
     related = core.RelatedPostings()
     for index, value in enumerate(expected):
         related.add(testutil.Posting('Income:Donations', -index, key=value))
     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
+    )
     assert related.all_meta_links('approval') == set()
 def test_all_meta_links_singletons():
     postings = (
         testutil.Posting('Income:Donations', -10, statement=value)
         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 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
+    )
     assert related.all_meta_links('approval') == testutil.LINK_METADATA_STRINGS
 def test_group_by_meta_zero():
     assert len(core.RelatedPostings.group_by_meta([], 'metacurrency')) == 0
 def test_group_by_meta_key_error():
     # Make sure the return value doesn't act like a defaultdict.
     with pytest.raises(KeyError):
         core.RelatedPostings.group_by_meta([], 'metakey')['metavalue']
 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')

0 comments (0 inline, 0 general)