[3.13] gh-95468: Add more tests for "--" (double dash) in test_argparse (GH-124274) (GH-125068)

(cherry picked from commit baa3550bc3)

Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
This commit is contained in:
Miss Islington (bot) 2024-10-07 23:40:47 +02:00 committed by GitHub
parent 970f3b73d2
commit 2a40dda89d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -5993,7 +5993,9 @@ class TestParseKnownArgs(TestCase):
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='*')
@ -6017,6 +6019,7 @@ class TestParseKnownArgs(TestCase):
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='*')
@ -6039,6 +6042,7 @@ class TestParseKnownArgs(TestCase):
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='*')
@ -6054,9 +6058,10 @@ class TestParseKnownArgs(TestCase):
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)
@ -6067,6 +6072,40 @@ class TestParseKnownArgs(TestCase):
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