diff options
Diffstat (limited to 'test/test_argparse.py')
-rw-r--r-- | test/test_argparse.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/test/test_argparse.py b/test/test_argparse.py index c9ec289..fadc1af 100644 --- a/test/test_argparse.py +++ b/test/test_argparse.py @@ -2466,6 +2466,29 @@ class TestSetDefaults(TestCase): self.assertEqual(NS(w='WW', x='X', y='YY', z='Z'), parser.parse_args([])) +# ================= +# Get default tests +# ================= + +class TestGetDefault(TestCase): + + def test_get_default(self): + parser = ErrorRaisingArgumentParser() + self.assertEqual(None, parser.get_default("foo")) + self.assertEqual(None, parser.get_default("bar")) + + parser.add_argument("--foo") + self.assertEqual(None, parser.get_default("foo")) + self.assertEqual(None, parser.get_default("bar")) + + parser.add_argument("--bar", type=int, default=42) + self.assertEqual(None, parser.get_default("foo")) + self.assertEqual(42, parser.get_default("bar")) + + parser.set_defaults(foo="badger") + self.assertEqual("badger", parser.get_default("foo")) + self.assertEqual(42, parser.get_default("bar")) + # ===================== # Help formatting tests # ===================== |