Changeset - 1b8137529472
[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:
0 comments (0 inline, 0 general)
conservancy_beancount/reports/core.py
Show inline comments
...
 
@@ -69,2 +69,7 @@ class Balance(Mapping[str, data.Amount]):
 

	
 
    def __neg__(self) -> 'Balance':
 
        return type(self)(
 
            (key, -amt) for key, amt in self.items()
 
        )
 

	
 
    def __getitem__(self, key: str) -> data.Amount:
tests/test_reports_balance.py
Show inline comments
...
 
@@ -69,2 +69,18 @@ def test_mixed_balance():
 

	
 
@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', [
0 comments (0 inline, 0 general)