Skip to content

Commit

Permalink
pythongh-95468: Add more tests for "--" (double dash) in test_argparse (
Browse files Browse the repository at this point in the history
pythonGH-124274)

(cherry picked from commit baa3550)

Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
  • Loading branch information
serhiy-storchaka authored and miss-islington committed Sep 20, 2024
1 parent 6c6b044 commit 429b57b
Showing 1 changed file with 41 additions and 2 deletions.
43 changes: 41 additions & 2 deletions Lib/test/test_argparse.py
Original file line number Diff line number Diff line change
Expand Up @@ -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='*')
Expand All @@ -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='*')
Expand All @@ -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='*')
Expand All @@ -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)
Expand All @@ -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
Expand Down

0 comments on commit 429b57b

Please sign in to comment.