NPO-Accounting/conservancy_beancount Changeset - cd1766adcf14

Changeset - cd1766adcf14

Parent rev.

Child rev.

[Not reviewed]

0 2 0

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

reports: Balance.__eq__ respects tolerance.

2 files changed with 56 insertions and 4 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

@@ ... / @@ -134,6 +134,8 @@ 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
         if isinstance(other, Balance):
             clean_self = self.clean_copy()
             clean_other = other.clean_copy()
             return len(clean_self) == len(clean_other) and all(
                 clean_self[key] == clean_other.get(key) for key in clean_self
+            )
         else:
@@ ... / @@ -162,2 +164,13 @@ class Balance(Mapping[str, data.Amount]): @@
     def copy(self: BalanceType) -> BalanceType:
         return type(self)(self.values())
     def clean_copy(self: BalanceType, tolerance: Optional[Decimal]=None) -> BalanceType:
         if tolerance is None:
             tolerance = self.tolerance
         return type(self)(
             amount for amount in self.values()
             if abs(amount.number) >= tolerance
+        )
     @staticmethod

tests/test_reports_balance.py

➞

Show inline comments

@@ ... / @@ -36,2 +36,4 @@ DEFAULT_STRINGS = [ @@
 TOLERANCES = [Decimal(n) for n in ['.1', '.01', '.001', 0]]
 def amounts_from_map(currency_map):
@@ ... / @@ -221,2 +223,11 @@ def test_eq(map1, map2, expected): @@
 @pytest.mark.parametrize('tolerance', TOLERANCES)
 def test_eq_considers_tolerance(tolerance):
     tolerance = Decimal(tolerance)
     mapping = {'EUR': 100, 'USD': '.002'}
     bal1 = core.Balance(amounts_from_map(mapping))
     mapping['USD'] = '.004'
     bal2 = core.Balance(amounts_from_map(mapping), tolerance)
     assert (bal1 == bal2) == (tolerance > Decimal('.002'))
 @pytest.mark.parametrize('number,currency', {
@@ ... / @@ -296,2 +307,30 @@ def test_iadd_balance(mapping): @@
 def test_copy():
     amounts = frozenset(amounts_from_map({'USD': 10, 'EUR': '.001'}))
     # Use a ridiculous tolerance to test it doesn't matter.
     actual = core.Balance(amounts, 100).copy()
     assert frozenset(actual.values()) == amounts
 @pytest.mark.parametrize('tolerance', TOLERANCES)
 def test_clean_copy(tolerance):
     usd = testutil.Amount(10)
     eur = testutil.Amount('.002', 'EUR')
     actual = core.Balance([usd, eur], tolerance).clean_copy()
     if tolerance < eur.number:
         expected = {usd, eur}
     else:
         expected = {usd}
     assert frozenset(actual.values()) == expected
 @pytest.mark.parametrize('tolerance', TOLERANCES)
 def test_clean_copy_arg(tolerance):
     usd = testutil.Amount(10)
     eur = testutil.Amount('.002', 'EUR')
     actual = core.Balance([usd, eur], 0).clean_copy(tolerance)
     if tolerance < eur.number:
         expected = {usd, eur}
     else:
         expected = {usd}
     assert frozenset(actual.values()) == expected
 @pytest.mark.parametrize('mapping,expected', DEFAULT_STRINGS)

0 comments (0 inline, 0 general)