diff --git a/Lib/test/test_argparse.py b/Lib/test/test_argparse.py index e1fe42b505daf1..42998d5c6ea06d 100644 --- a/Lib/test/test_argparse.py +++ b/Lib/test/test_argparse.py @@ -5383,7 +5383,9 @@ def test_zero_or_more_optional(self): args = parser.parse_args([]) self.assertEqual(NS(x=[]), args) - def test_double_dash(self): + +class TestDoubleDash(TestCase): + def test_single_argument_option(self): parser = argparse.ArgumentParser(exit_on_error=False) parser.add_argument('-f', '--foo') parser.add_argument('bar', nargs='*') @@ -5407,6 +5409,7 @@ def test_double_dash(self): args = parser.parse_args(['a', '--', 'b', '--', 'c', '--foo', 'd']) self.assertEqual(NS(foo=None, bar=['a', 'b', '--', 'c', '--foo', 'd']), args) + def test_multiple_argument_option(self): parser = argparse.ArgumentParser(exit_on_error=False) parser.add_argument('-f', '--foo', nargs='*') parser.add_argument('bar', nargs='*') @@ -5429,6 +5432,7 @@ def test_double_dash(self): self.assertEqual(NS(foo=['c'], bar=['a', 'b']), args) self.assertEqual(argv, ['--', 'd']) + def test_multiple_double_dashes(self): parser = argparse.ArgumentParser(exit_on_error=False) parser.add_argument('foo') parser.add_argument('bar', nargs='*') @@ -5444,9 +5448,10 @@ def test_double_dash(self): args = parser.parse_args(['--', '--', 'a', '--', 'b', 'c']) self.assertEqual(NS(foo='--', bar=['a', '--', 'b', 'c']), args) + def test_remainder(self): parser = argparse.ArgumentParser(exit_on_error=False) parser.add_argument('foo') - parser.add_argument('bar', nargs=argparse.REMAINDER) + parser.add_argument('bar', nargs='...') args = parser.parse_args(['--', 'a', 'b', 'c']) self.assertEqual(NS(foo='a', bar=['b', 'c']), args) @@ -5457,6 +5462,40 @@ def test_double_dash(self): args = parser.parse_args(['a', '--', 'b', '--', 'c']) self.assertEqual(NS(foo='a', bar=['b', '--', 'c']), args) + parser = argparse.ArgumentParser(exit_on_error=False) + parser.add_argument('--foo') + parser.add_argument('bar', nargs='...') + args = parser.parse_args(['--foo', 'a', '--', 'b', '--', 'c']) + self.assertEqual(NS(foo='a', bar=['--', 'b', '--', 'c']), args) + + def test_subparser(self): + parser = argparse.ArgumentParser(exit_on_error=False) + parser.add_argument('foo') + subparsers = parser.add_subparsers() + parser1 = subparsers.add_parser('run') + parser1.add_argument('-f') + parser1.add_argument('bar', nargs='*') + + args = parser.parse_args(['x', 'run', 'a', 'b', '-f', 'c']) + self.assertEqual(NS(foo='x', f='c', bar=['a', 'b']), args) + args = parser.parse_args(['x', 'run', 'a', 'b', '--', '-f', 'c']) + self.assertEqual(NS(foo='x', f=None, bar=['a', 'b', '-f', 'c']), args) + args = parser.parse_args(['x', 'run', 'a', '--', 'b', '-f', 'c']) + self.assertEqual(NS(foo='x', f=None, bar=['a', 'b', '-f', 'c']), args) + args = parser.parse_args(['x', 'run', '--', 'a', 'b', '-f', 'c']) + self.assertEqual(NS(foo='x', f=None, bar=['a', 'b', '-f', 'c']), args) + args = parser.parse_args(['x', '--', 'run', 'a', 'b', '-f', 'c']) + self.assertEqual(NS(foo='x', f='c', bar=['a', 'b']), args) + args = parser.parse_args(['--', 'x', 'run', 'a', 'b', '-f', 'c']) + self.assertEqual(NS(foo='x', f='c', bar=['a', 'b']), args) + args = parser.parse_args(['x', 'run', '--', 'a', '--', 'b']) + self.assertEqual(NS(foo='x', f=None, bar=['a', '--', 'b']), args) + args = parser.parse_args(['x', '--', 'run', '--', 'a', '--', 'b']) + self.assertEqual(NS(foo='x', f=None, bar=['a', '--', 'b']), args) + self.assertRaisesRegex(argparse.ArgumentError, + "invalid choice: '--'", + parser.parse_args, ['--', 'x', '--', 'run', 'a', 'b']) + # =========================== # parse_intermixed_args tests