NPO-Accounting/conservancy_beancount Changeset - 1b8137529472

Changeset - 1b8137529472

Parent rev.

Child rev.

[Not reviewed]

0 2 0

Brett Smith - 4 years ago 2020-04-29 18:35:20
brettcsmith@brettcsmith.org

reports: Add Balance.__neg__() method.

2 files changed with 21 insertions and 0 deletions:

conservancy_beancount/reports/core.py

tests/test_reports_balance.py

0 comments (0 inline, 0 general)

conservancy_beancount/reports/core.py

➞

Show inline comments

@@ ... / @@ -64,12 +64,17 @@ class Balance(Mapping[str, data.Amount]): @@
         amounts = [amount for amount in self.values() if amount.number]
         if not amounts:
             return "Zero balance"
         amounts.sort(key=lambda amt: abs(amt.number), reverse=True)
         return ', '.join(str(amount) for amount in amounts)
     def __neg__(self) -> 'Balance':
         return type(self)(
             (key, -amt) for key, amt in self.items()
+        )
     def __getitem__(self, key: str) -> data.Amount:
         return data.Amount(self._currency_map[key], key)
     def __iter__(self) -> Iterator[str]:
         return iter(self._currency_map)

tests/test_reports_balance.py

➞

Show inline comments

@@ ... / @@ -64,12 +64,28 @@ def test_mixed_balance(): @@
     balance = core.Balance(amounts.items())
     assert balance
     assert len(balance) == 2
     assert not balance.is_zero()
     assert all(balance[key] == amt for key, amt in amounts.items())
 @pytest.mark.parametrize('balance_map_kwargs', [
     {},
     {'USD': 0},
     {'EUR': 10},
     {'JPY': 20, 'BRL': 30},
     {'EUR': -15},
     {'JPY': -25, 'BRL': -35},
     {'JPY': 40, 'USD': 0, 'EUR': -50},
 ])
 def test_neg(balance_map_kwargs):
     amounts = testutil.balance_map(**balance_map_kwargs)
     actual = -core.Balance(amounts.items())
     assert set(actual) == set(balance_map_kwargs)
     for key in balance_map_kwargs:
         assert actual[key] == -amounts[key]
 @pytest.mark.parametrize('balance_map_kwargs,expected', [
     ({}, "Zero balance"),
     ({'JPY': 0, 'BRL': 0}, "Zero balance"),
     ({'USD': '20.00'}, "20.00 USD"),
     ({'EUR': '50.00', 'GBP': '80.00'}, "80.00 GBP, 50.00 EUR"),
     ({'JPY': '-55.00', 'BRL': '-85.00'}, "-85.00 BRL, -55.00 JPY"),

0 comments (0 inline, 0 general)