NPO-Accounting/conservancy_beancount Changeset - 7660700e6c8f

Changeset - 7660700e6c8f

Parent rev.

Child rev.

[Not reviewed]

0 2 0

Brett Smith - 4 years ago 2020-06-15 22:11:39
brettcsmith@brettcsmith.org

cliutil: Add date_arg() function.

Meant to be used as an argument type.

2 files changed with 25 insertions and 0 deletions:

conservancy_beancount/cliutil.py

tests/test_cliutil.py

0 comments (0 inline, 0 general)

conservancy_beancount/cliutil.py

➞

Show inline comments

@@ ... @@
 GNU Affero General Public License for more details.
 You should have received a copy of the GNU Affero General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>."""
 import argparse
 import datetime
 import enum
 import io
 import logging
 import operator
 import os
 import pkg_resources
@@ ... @@
         action=InfoAction,
         nargs=0,
         const=f"{progname} version {VERSION}\n{LICENSE}",
         help="Show program version and license information",
+    )
 def date_arg(arg: str) -> datetime.date:
     return datetime.datetime.strptime(arg, '%Y-%m-%d').date()
 def make_entry_point(mod_name: str, prog_name: str=sys.argv[0]) -> Callable[[], int]:
     """Create an entry_point function for a tool
     The returned function is suitable for use as an entry_point in setup.py.
     It sets up the root logger and excepthook, then calls the module's main
     function.

tests/test_cliutil.py

➞

Show inline comments

@@ ... / @@ -12,12 +12,13 @@ @@
 # GNU Affero General Public License for more details.
+#
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 import argparse
 import datetime
 import errno
 import io
 import inspect
 import logging
 import os
 import re
@@ ... / @@ -71,12 +72,32 @@ def test_bytes_output_path(path_name, tmp_path): @@
 @pytest.mark.parametrize('path', STREAM_PATHS)
 def test_bytes_output_stream(path):
     stream = io.BytesIO()
     actual = cliutil.bytes_output(path, stream)
     assert actual is stream
 @pytest.mark.parametrize('year,month,day', [
     (2000, 1, 1),
     (2016, 2, 29),
     (2020, 12, 31),
 ])
 def test_date_arg_valid(year, month, day):
     arg = f'{year}-{month}-{day}'
     expected = datetime.date(year, month, day)
     assert cliutil.date_arg(arg) == expected
 @pytest.mark.parametrize('arg', [
     '2000',
     '20-02-12',
     '2019-02-29',
     'two thousand',
 ])
 def test_date_arg_invalid(arg):
     with pytest.raises(ValueError):
         cliutil.date_arg(arg)
 @pytest.mark.parametrize('func_name', [
     'bytes_output',
     'text_output',
 ])
 def test_default_output(func_name):
     actual = getattr(cliutil, func_name)()

0 comments (0 inline, 0 general)