NPO-Accounting/conservancy_beancount Changeset - 0581525c9824

Changeset - 0581525c9824

Parent rev.

Child rev.

[Not reviewed]

0 2 0

Brett Smith - 4 years ago 2020-06-09 13:04:27
brettcsmith@brettcsmith.org

reports: Add Balance.__pos__() method.

I did this while I was working on normalize_amount_func.
It turns out it's not immediately needed, but it's still nice to have.

2 files changed with 17 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

@@ ... / @@ -131,24 +131,27 @@ class Balance(Mapping[str, data.Amount]): @@
     def __eq__(self, other: Any) -> bool:
         if (self.is_zero()
             and isinstance(other, Balance)
             and other.is_zero()):
             return True
         else:
             return super().__eq__(other)
     def __neg__(self: BalanceType) -> BalanceType:
         return type(self)(-amt for amt in self.values())
     def __pos__(self: BalanceType) -> BalanceType:
         return self
     def __getitem__(self, key: str) -> data.Amount:
         return self._currency_map[key]
     def __iter__(self) -> Iterator[str]:
         return iter(self._currency_map)
     def __len__(self) -> int:
         return len(self._currency_map)
     def _all_amounts(self,
                      op_func: Callable[[DecimalCompat, DecimalCompat], bool],
                      operand: DecimalCompat,

tests/test_reports_balance.py

➞

Show inline comments

@@ ... / @@ -159,24 +159,38 @@ def test_abs(mapping): @@
     {'EUR': 10},
     {'JPY': 20, 'BRL': 30},
     {'EUR': -15},
     {'JPY': -25, 'BRL': -35},
     {'JPY': 40, 'USD': 0, 'EUR': -50},
 ])
 def test_neg(mapping):
     actual = -core.Balance(amounts_from_map(mapping))
     assert set(actual) == set(mapping)
     for key, number in mapping.items():
         assert actual[key] == testutil.Amount(-number, key)
 @pytest.mark.parametrize('mapping', [
     {},
     {'USD': 0},
     {'EUR': 10},
     {'JPY': 20, 'BRL': 30},
     {'EUR': -15},
     {'JPY': -25, 'BRL': -35},
     {'JPY': 40, 'USD': 0, 'EUR': -50},
 ])
 def test_pos(mapping):
     amounts = frozenset(amounts_from_map(mapping))
     actual = +core.Balance(amounts)
     assert set(actual.values()) == amounts
 @pytest.mark.parametrize('map1,map2,expected', [
     ({}, {}, True),
     ({}, {'USD': 0}, True),
     ({}, {'EUR': 1}, False),
     ({'USD': 1}, {'EUR': 1}, False),
     ({'USD': 1}, {'USD': '1.0'}, True),
     ({'USD': 1}, {'USD': '1.0', 'EUR': '2.0'}, False),
     ({'USD': 1, 'BRL': '2.0'}, {'USD': '1.0', 'EUR': '2.0'}, False),
     ({'USD': 1, 'EUR': 2, 'BRL': '3.0'}, {'USD': '1.0', 'EUR': '2.0'}, False),
     ({'USD': 1, 'EUR': 2}, {'USD': '1.0', 'EUR': '2.0'}, True),
 ])
 def test_eq(map1, map2, expected):

0 comments (0 inline, 0 general)