https://github.com/python/cpython/commit/baa3550bc3a119f41cc4eaed5373f9d695208e8e
commit: baa3550bc3a119f41cc4eaed5373f9d695208e8e
branch: main
author: Serhiy Storchaka <[email protected]>
committer: serhiy-storchaka <[email protected]>
date: 2024-09-20T09:54:20Z
summary:
gh-95468: Add more tests for "--" (double dash) in test_argparse (GH-124274)
files:
M Lib/test/test_argparse.py
diff --git a/Lib/test/test_argparse.py b/Lib/test/test_argparse.py
index 609881879551b6..138ff19e86acf4 100644
--- a/Lib/test/test_argparse.py
+++ b/Lib/test/test_argparse.py
@@ -5720,7 +5720,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='*')
@@ -5744,6 +5746,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='*')
@@ -5766,6 +5769,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='*')
@@ -5781,9 +5785,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)
@@ -5794,6 +5799,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
_______________________________________________
Python-checkins mailing list -- [email protected]
To unsubscribe send an email to [email protected]
https://mail.python.org/mailman3/lists/python-checkins.python.org/
Member address: [email protected]
