diff options
| author | Kevin Van Brunt <kmvanbrunt@gmail.com> | 2020-08-25 16:11:49 -0400 |
|---|---|---|
| committer | Kevin Van Brunt <kmvanbrunt@gmail.com> | 2020-08-26 17:52:34 -0400 |
| commit | 478ea83336a4d1659ad06ef365db3dc5e051e46d (patch) | |
| tree | cbaff855e2d73f1d86ae1230f32770e31c62ad6b /tests/test_argparse_custom.py | |
| parent | 97c348c599d8fa963553593e5c19fb100b85e313 (diff) | |
| download | cmd2-git-478ea83336a4d1659ad06ef365db3dc5e051e46d.tar.gz | |
The functions cmd2 adds to Namespaces (get_statement() and get_handler()) are now
Cmd2AttributeWrapper objects named cmd2_statement and cmd2_handler. This makes it
easy to filter out which attributes in an argparse.Namespace were added by cmd2.
Diffstat (limited to 'tests/test_argparse_custom.py')
| -rw-r--r-- | tests/test_argparse_custom.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tests/test_argparse_custom.py b/tests/test_argparse_custom.py index 3ce90118..e2b3bb97 100644 --- a/tests/test_argparse_custom.py +++ b/tests/test_argparse_custom.py @@ -267,3 +267,13 @@ def test_apcustom_metavar_tuple(): parser = Cmd2ArgumentParser() parser.add_argument('--aflag', nargs=2, metavar=('foo', 'bar'), help='This is a test') assert '[--aflag foo bar]' in parser.format_help() + + +def test_cmd2_attribute_wrapper(): + initial_val = 5 + wrapper = cmd2.Cmd2AttributeWrapper(initial_val) + assert wrapper.get() == initial_val + + new_val = 22 + wrapper.set(new_val) + assert wrapper.get() == new_val |