NPO-Accounting/conservancy_beancount Changeset - d8df34ebaf6f

Changeset - d8df34ebaf6f

Parent rev.

Child rev.

[Not reviewed]

0 2 0

Brett Smith - 4 years ago 2020-05-28 13:03:23
brettcsmith@brettcsmith.org

reports: Balance.format() accepts None as a format.

2 files changed with 12 insertions and 1 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

@@ ... / @@ -112,17 +112,20 @@ class Balance(Mapping[str, data.Amount]): @@
         """Returns true if all amounts in the balance <= 0."""
         return self._all_amounts(operator.le, 0)
     def format(self,
                fmt: str='#,#00.00 ¤¤',
+               fmt: Optional[str]='#,#00.00 ¤¤',
                sep: str=', ',
                empty: str="Zero balance",
     ) -> str:
         """Formats the balance as a string with the given parameters
         If the balance is zero, returns ``empty``. Otherwise, returns a string
         with each amount in the balance formatted as ``fmt``, separated by
         ``sep``.
         If you set ``fmt`` to None, amounts will be formatted according to the
         user's locale. The default format is Beancount's input format.
         """
         amounts = [amount for amount in self.values() if amount.number]
         if not amounts:
             return empty

tests/test_reports_balance.py

➞

Show inline comments

@@ ... / @@ -21,8 +21,10 @@ from decimal import Decimal @@
 import pytest
 from . import testutil
 import babel.numbers
 from conservancy_beancount.reports import core
 DEFAULT_STRINGS = [
     ({}, "Zero balance"),
@@ ... / @@ -190,8 +192,14 @@ def test_format_sep(sep): @@
     amounts = testutil.balance_map(**bal_kwargs)
     balance = core.Balance(amounts)
     assert balance.format(sep=sep) == expected
 def test_format_none():
     amounts = testutil.balance_map(BRL=65000)
     balance = core.Balance(amounts)
     expected = babel.numbers.format_currency(65000, 'BRL')
     assert balance.format(None) == expected
 @pytest.mark.parametrize('empty', [
     "N/A",
     "Zero",
     "ø",

0 comments (0 inline, 0 general)