PageRenderTime 205ms CodeModel.GetById 19ms RepoModel.GetById 3ms app.codeStats 1ms

/lib-python/modified-2.7/test/test_argparse.py

https://bitbucket.org/dac_io/pypy
Python | 4646 lines | 4639 code | 6 blank | 1 comment | 2 complexity | f31dd679868b0aab7953a6b7fab8a988 MD5 | raw file
  1. # Author: Steven J. Bethard <steven.bethard@gmail.com>.
  2. import codecs
  3. import inspect
  4. import os
  5. import shutil
  6. import stat
  7. import sys
  8. import textwrap
  9. import tempfile
  10. import unittest
  11. import argparse
  12. from StringIO import StringIO
  13. class StdIOBuffer(StringIO):
  14. pass
  15. from test import test_support
  16. class TestCase(unittest.TestCase):
  17. def assertEqual(self, obj1, obj2):
  18. if obj1 != obj2:
  19. print('')
  20. print(repr(obj1))
  21. print(repr(obj2))
  22. print(obj1)
  23. print(obj2)
  24. super(TestCase, self).assertEqual(obj1, obj2)
  25. def setUp(self):
  26. # The tests assume that line wrapping occurs at 80 columns, but this
  27. # behaviour can be overridden by setting the COLUMNS environment
  28. # variable. To ensure that this assumption is true, unset COLUMNS.
  29. env = test_support.EnvironmentVarGuard()
  30. env.unset("COLUMNS")
  31. self.addCleanup(env.__exit__)
  32. class TempDirMixin(object):
  33. def setUp(self):
  34. self.temp_dir = tempfile.mkdtemp()
  35. self.old_dir = os.getcwd()
  36. os.chdir(self.temp_dir)
  37. def tearDown(self):
  38. os.chdir(self.old_dir)
  39. shutil.rmtree(self.temp_dir, True)
  40. def create_readonly_file(self, filename):
  41. file_path = os.path.join(self.temp_dir, filename)
  42. with open(file_path, 'w') as file:
  43. file.write(filename)
  44. os.chmod(file_path, stat.S_IREAD)
  45. class Sig(object):
  46. def __init__(self, *args, **kwargs):
  47. self.args = args
  48. self.kwargs = kwargs
  49. class NS(object):
  50. def __init__(self, **kwargs):
  51. self.__dict__.update(kwargs)
  52. def __repr__(self):
  53. sorted_items = sorted(self.__dict__.items())
  54. kwarg_str = ', '.join(['%s=%r' % tup for tup in sorted_items])
  55. return '%s(%s)' % (type(self).__name__, kwarg_str)
  56. __hash__ = None
  57. def __eq__(self, other):
  58. return vars(self) == vars(other)
  59. def __ne__(self, other):
  60. return not (self == other)
  61. class ArgumentParserError(Exception):
  62. def __init__(self, message, stdout=None, stderr=None, error_code=None):
  63. Exception.__init__(self, message, stdout, stderr)
  64. self.message = message
  65. self.stdout = stdout
  66. self.stderr = stderr
  67. self.error_code = error_code
  68. def stderr_to_parser_error(parse_args, *args, **kwargs):
  69. # if this is being called recursively and stderr or stdout is already being
  70. # redirected, simply call the function and let the enclosing function
  71. # catch the exception
  72. if isinstance(sys.stderr, StdIOBuffer) or isinstance(sys.stdout, StdIOBuffer):
  73. return parse_args(*args, **kwargs)
  74. # if this is not being called recursively, redirect stderr and
  75. # use it as the ArgumentParserError message
  76. old_stdout = sys.stdout
  77. old_stderr = sys.stderr
  78. sys.stdout = StdIOBuffer()
  79. sys.stderr = StdIOBuffer()
  80. try:
  81. try:
  82. result = parse_args(*args, **kwargs)
  83. for key in list(vars(result)):
  84. if getattr(result, key) is sys.stdout:
  85. setattr(result, key, old_stdout)
  86. if getattr(result, key) is sys.stderr:
  87. setattr(result, key, old_stderr)
  88. return result
  89. except SystemExit:
  90. code = sys.exc_info()[1].code
  91. stdout = sys.stdout.getvalue()
  92. stderr = sys.stderr.getvalue()
  93. raise ArgumentParserError("SystemExit", stdout, stderr, code)
  94. finally:
  95. sys.stdout = old_stdout
  96. sys.stderr = old_stderr
  97. class ErrorRaisingArgumentParser(argparse.ArgumentParser):
  98. def parse_args(self, *args, **kwargs):
  99. parse_args = super(ErrorRaisingArgumentParser, self).parse_args
  100. return stderr_to_parser_error(parse_args, *args, **kwargs)
  101. def exit(self, *args, **kwargs):
  102. exit = super(ErrorRaisingArgumentParser, self).exit
  103. return stderr_to_parser_error(exit, *args, **kwargs)
  104. def error(self, *args, **kwargs):
  105. error = super(ErrorRaisingArgumentParser, self).error
  106. return stderr_to_parser_error(error, *args, **kwargs)
  107. class ParserTesterMetaclass(type):
  108. """Adds parser tests using the class attributes.
  109. Classes of this type should specify the following attributes:
  110. argument_signatures -- a list of Sig objects which specify
  111. the signatures of Argument objects to be created
  112. failures -- a list of args lists that should cause the parser
  113. to fail
  114. successes -- a list of (initial_args, options, remaining_args) tuples
  115. where initial_args specifies the string args to be parsed,
  116. options is a dict that should match the vars() of the options
  117. parsed out of initial_args, and remaining_args should be any
  118. remaining unparsed arguments
  119. """
  120. def __init__(cls, name, bases, bodydict):
  121. if name == 'ParserTestCase':
  122. return
  123. # default parser signature is empty
  124. if not hasattr(cls, 'parser_signature'):
  125. cls.parser_signature = Sig()
  126. if not hasattr(cls, 'parser_class'):
  127. cls.parser_class = ErrorRaisingArgumentParser
  128. # ---------------------------------------
  129. # functions for adding optional arguments
  130. # ---------------------------------------
  131. def no_groups(parser, argument_signatures):
  132. """Add all arguments directly to the parser"""
  133. for sig in argument_signatures:
  134. parser.add_argument(*sig.args, **sig.kwargs)
  135. def one_group(parser, argument_signatures):
  136. """Add all arguments under a single group in the parser"""
  137. group = parser.add_argument_group('foo')
  138. for sig in argument_signatures:
  139. group.add_argument(*sig.args, **sig.kwargs)
  140. def many_groups(parser, argument_signatures):
  141. """Add each argument in its own group to the parser"""
  142. for i, sig in enumerate(argument_signatures):
  143. group = parser.add_argument_group('foo:%i' % i)
  144. group.add_argument(*sig.args, **sig.kwargs)
  145. # --------------------------
  146. # functions for parsing args
  147. # --------------------------
  148. def listargs(parser, args):
  149. """Parse the args by passing in a list"""
  150. return parser.parse_args(args)
  151. def sysargs(parser, args):
  152. """Parse the args by defaulting to sys.argv"""
  153. old_sys_argv = sys.argv
  154. sys.argv = [old_sys_argv[0]] + args
  155. try:
  156. return parser.parse_args()
  157. finally:
  158. sys.argv = old_sys_argv
  159. # class that holds the combination of one optional argument
  160. # addition method and one arg parsing method
  161. class AddTests(object):
  162. def __init__(self, tester_cls, add_arguments, parse_args):
  163. self._add_arguments = add_arguments
  164. self._parse_args = parse_args
  165. add_arguments_name = self._add_arguments.__name__
  166. parse_args_name = self._parse_args.__name__
  167. for test_func in [self.test_failures, self.test_successes]:
  168. func_name = test_func.__name__
  169. names = func_name, add_arguments_name, parse_args_name
  170. test_name = '_'.join(names)
  171. def wrapper(self, test_func=test_func):
  172. test_func(self)
  173. try:
  174. wrapper.__name__ = test_name
  175. except TypeError:
  176. pass
  177. setattr(tester_cls, test_name, wrapper)
  178. def _get_parser(self, tester):
  179. args = tester.parser_signature.args
  180. kwargs = tester.parser_signature.kwargs
  181. parser = tester.parser_class(*args, **kwargs)
  182. self._add_arguments(parser, tester.argument_signatures)
  183. return parser
  184. def test_failures(self, tester):
  185. parser = self._get_parser(tester)
  186. for args_str in tester.failures:
  187. args = args_str.split()
  188. raises = tester.assertRaises
  189. raises(ArgumentParserError, parser.parse_args, args)
  190. def test_successes(self, tester):
  191. parser = self._get_parser(tester)
  192. for args, expected_ns in tester.successes:
  193. if isinstance(args, str):
  194. args = args.split()
  195. result_ns = self._parse_args(parser, args)
  196. tester.assertEqual(expected_ns, result_ns)
  197. # add tests for each combination of an optionals adding method
  198. # and an arg parsing method
  199. for add_arguments in [no_groups, one_group, many_groups]:
  200. for parse_args in [listargs, sysargs]:
  201. AddTests(cls, add_arguments, parse_args)
  202. bases = TestCase,
  203. ParserTestCase = ParserTesterMetaclass('ParserTestCase', bases, {})
  204. # ===============
  205. # Optionals tests
  206. # ===============
  207. class TestOptionalsSingleDash(ParserTestCase):
  208. """Test an Optional with a single-dash option string"""
  209. argument_signatures = [Sig('-x')]
  210. failures = ['-x', 'a', '--foo', '-x --foo', '-x -y']
  211. successes = [
  212. ('', NS(x=None)),
  213. ('-x a', NS(x='a')),
  214. ('-xa', NS(x='a')),
  215. ('-x -1', NS(x='-1')),
  216. ('-x-1', NS(x='-1')),
  217. ]
  218. class TestOptionalsSingleDashCombined(ParserTestCase):
  219. """Test an Optional with a single-dash option string"""
  220. argument_signatures = [
  221. Sig('-x', action='store_true'),
  222. Sig('-yyy', action='store_const', const=42),
  223. Sig('-z'),
  224. ]
  225. failures = ['a', '--foo', '-xa', '-x --foo', '-x -z', '-z -x',
  226. '-yx', '-yz a', '-yyyx', '-yyyza', '-xyza']
  227. successes = [
  228. ('', NS(x=False, yyy=None, z=None)),
  229. ('-x', NS(x=True, yyy=None, z=None)),
  230. ('-za', NS(x=False, yyy=None, z='a')),
  231. ('-z a', NS(x=False, yyy=None, z='a')),
  232. ('-xza', NS(x=True, yyy=None, z='a')),
  233. ('-xz a', NS(x=True, yyy=None, z='a')),
  234. ('-x -za', NS(x=True, yyy=None, z='a')),
  235. ('-x -z a', NS(x=True, yyy=None, z='a')),
  236. ('-y', NS(x=False, yyy=42, z=None)),
  237. ('-yyy', NS(x=False, yyy=42, z=None)),
  238. ('-x -yyy -za', NS(x=True, yyy=42, z='a')),
  239. ('-x -yyy -z a', NS(x=True, yyy=42, z='a')),
  240. ]
  241. class TestOptionalsSingleDashLong(ParserTestCase):
  242. """Test an Optional with a multi-character single-dash option string"""
  243. argument_signatures = [Sig('-foo')]
  244. failures = ['-foo', 'a', '--foo', '-foo --foo', '-foo -y', '-fooa']
  245. successes = [
  246. ('', NS(foo=None)),
  247. ('-foo a', NS(foo='a')),
  248. ('-foo -1', NS(foo='-1')),
  249. ('-fo a', NS(foo='a')),
  250. ('-f a', NS(foo='a')),
  251. ]
  252. class TestOptionalsSingleDashSubsetAmbiguous(ParserTestCase):
  253. """Test Optionals where option strings are subsets of each other"""
  254. argument_signatures = [Sig('-f'), Sig('-foobar'), Sig('-foorab')]
  255. failures = ['-f', '-foo', '-fo', '-foo b', '-foob', '-fooba', '-foora']
  256. successes = [
  257. ('', NS(f=None, foobar=None, foorab=None)),
  258. ('-f a', NS(f='a', foobar=None, foorab=None)),
  259. ('-fa', NS(f='a', foobar=None, foorab=None)),
  260. ('-foa', NS(f='oa', foobar=None, foorab=None)),
  261. ('-fooa', NS(f='ooa', foobar=None, foorab=None)),
  262. ('-foobar a', NS(f=None, foobar='a', foorab=None)),
  263. ('-foorab a', NS(f=None, foobar=None, foorab='a')),
  264. ]
  265. class TestOptionalsSingleDashAmbiguous(ParserTestCase):
  266. """Test Optionals that partially match but are not subsets"""
  267. argument_signatures = [Sig('-foobar'), Sig('-foorab')]
  268. failures = ['-f', '-f a', '-fa', '-foa', '-foo', '-fo', '-foo b']
  269. successes = [
  270. ('', NS(foobar=None, foorab=None)),
  271. ('-foob a', NS(foobar='a', foorab=None)),
  272. ('-foor a', NS(foobar=None, foorab='a')),
  273. ('-fooba a', NS(foobar='a', foorab=None)),
  274. ('-foora a', NS(foobar=None, foorab='a')),
  275. ('-foobar a', NS(foobar='a', foorab=None)),
  276. ('-foorab a', NS(foobar=None, foorab='a')),
  277. ]
  278. class TestOptionalsNumeric(ParserTestCase):
  279. """Test an Optional with a short opt string"""
  280. argument_signatures = [Sig('-1', dest='one')]
  281. failures = ['-1', 'a', '-1 --foo', '-1 -y', '-1 -1', '-1 -2']
  282. successes = [
  283. ('', NS(one=None)),
  284. ('-1 a', NS(one='a')),
  285. ('-1a', NS(one='a')),
  286. ('-1-2', NS(one='-2')),
  287. ]
  288. class TestOptionalsDoubleDash(ParserTestCase):
  289. """Test an Optional with a double-dash option string"""
  290. argument_signatures = [Sig('--foo')]
  291. failures = ['--foo', '-f', '-f a', 'a', '--foo -x', '--foo --bar']
  292. successes = [
  293. ('', NS(foo=None)),
  294. ('--foo a', NS(foo='a')),
  295. ('--foo=a', NS(foo='a')),
  296. ('--foo -2.5', NS(foo='-2.5')),
  297. ('--foo=-2.5', NS(foo='-2.5')),
  298. ]
  299. class TestOptionalsDoubleDashPartialMatch(ParserTestCase):
  300. """Tests partial matching with a double-dash option string"""
  301. argument_signatures = [
  302. Sig('--badger', action='store_true'),
  303. Sig('--bat'),
  304. ]
  305. failures = ['--bar', '--b', '--ba', '--b=2', '--ba=4', '--badge 5']
  306. successes = [
  307. ('', NS(badger=False, bat=None)),
  308. ('--bat X', NS(badger=False, bat='X')),
  309. ('--bad', NS(badger=True, bat=None)),
  310. ('--badg', NS(badger=True, bat=None)),
  311. ('--badge', NS(badger=True, bat=None)),
  312. ('--badger', NS(badger=True, bat=None)),
  313. ]
  314. class TestOptionalsDoubleDashPrefixMatch(ParserTestCase):
  315. """Tests when one double-dash option string is a prefix of another"""
  316. argument_signatures = [
  317. Sig('--badger', action='store_true'),
  318. Sig('--ba'),
  319. ]
  320. failures = ['--bar', '--b', '--ba', '--b=2', '--badge 5']
  321. successes = [
  322. ('', NS(badger=False, ba=None)),
  323. ('--ba X', NS(badger=False, ba='X')),
  324. ('--ba=X', NS(badger=False, ba='X')),
  325. ('--bad', NS(badger=True, ba=None)),
  326. ('--badg', NS(badger=True, ba=None)),
  327. ('--badge', NS(badger=True, ba=None)),
  328. ('--badger', NS(badger=True, ba=None)),
  329. ]
  330. class TestOptionalsSingleDoubleDash(ParserTestCase):
  331. """Test an Optional with single- and double-dash option strings"""
  332. argument_signatures = [
  333. Sig('-f', action='store_true'),
  334. Sig('--bar'),
  335. Sig('-baz', action='store_const', const=42),
  336. ]
  337. failures = ['--bar', '-fbar', '-fbaz', '-bazf', '-b B', 'B']
  338. successes = [
  339. ('', NS(f=False, bar=None, baz=None)),
  340. ('-f', NS(f=True, bar=None, baz=None)),
  341. ('--ba B', NS(f=False, bar='B', baz=None)),
  342. ('-f --bar B', NS(f=True, bar='B', baz=None)),
  343. ('-f -b', NS(f=True, bar=None, baz=42)),
  344. ('-ba -f', NS(f=True, bar=None, baz=42)),
  345. ]
  346. class TestOptionalsAlternatePrefixChars(ParserTestCase):
  347. """Test an Optional with option strings with custom prefixes"""
  348. parser_signature = Sig(prefix_chars='+:/', add_help=False)
  349. argument_signatures = [
  350. Sig('+f', action='store_true'),
  351. Sig('::bar'),
  352. Sig('/baz', action='store_const', const=42),
  353. ]
  354. failures = ['--bar', '-fbar', '-b B', 'B', '-f', '--bar B', '-baz', '-h', '--help', '+h', '::help', '/help']
  355. successes = [
  356. ('', NS(f=False, bar=None, baz=None)),
  357. ('+f', NS(f=True, bar=None, baz=None)),
  358. ('::ba B', NS(f=False, bar='B', baz=None)),
  359. ('+f ::bar B', NS(f=True, bar='B', baz=None)),
  360. ('+f /b', NS(f=True, bar=None, baz=42)),
  361. ('/ba +f', NS(f=True, bar=None, baz=42)),
  362. ]
  363. class TestOptionalsAlternatePrefixCharsAddedHelp(ParserTestCase):
  364. """When ``-`` not in prefix_chars, default operators created for help
  365. should use the prefix_chars in use rather than - or --
  366. http://bugs.python.org/issue9444"""
  367. parser_signature = Sig(prefix_chars='+:/', add_help=True)
  368. argument_signatures = [
  369. Sig('+f', action='store_true'),
  370. Sig('::bar'),
  371. Sig('/baz', action='store_const', const=42),
  372. ]
  373. failures = ['--bar', '-fbar', '-b B', 'B', '-f', '--bar B', '-baz']
  374. successes = [
  375. ('', NS(f=False, bar=None, baz=None)),
  376. ('+f', NS(f=True, bar=None, baz=None)),
  377. ('::ba B', NS(f=False, bar='B', baz=None)),
  378. ('+f ::bar B', NS(f=True, bar='B', baz=None)),
  379. ('+f /b', NS(f=True, bar=None, baz=42)),
  380. ('/ba +f', NS(f=True, bar=None, baz=42))
  381. ]
  382. class TestOptionalsAlternatePrefixCharsMultipleShortArgs(ParserTestCase):
  383. """Verify that Optionals must be called with their defined prefixes"""
  384. parser_signature = Sig(prefix_chars='+-', add_help=False)
  385. argument_signatures = [
  386. Sig('-x', action='store_true'),
  387. Sig('+y', action='store_true'),
  388. Sig('+z', action='store_true'),
  389. ]
  390. failures = ['-w',
  391. '-xyz',
  392. '+x',
  393. '-y',
  394. '+xyz',
  395. ]
  396. successes = [
  397. ('', NS(x=False, y=False, z=False)),
  398. ('-x', NS(x=True, y=False, z=False)),
  399. ('+y -x', NS(x=True, y=True, z=False)),
  400. ('+yz -x', NS(x=True, y=True, z=True)),
  401. ]
  402. class TestOptionalsShortLong(ParserTestCase):
  403. """Test a combination of single- and double-dash option strings"""
  404. argument_signatures = [
  405. Sig('-v', '--verbose', '-n', '--noisy', action='store_true'),
  406. ]
  407. failures = ['--x --verbose', '-N', 'a', '-v x']
  408. successes = [
  409. ('', NS(verbose=False)),
  410. ('-v', NS(verbose=True)),
  411. ('--verbose', NS(verbose=True)),
  412. ('-n', NS(verbose=True)),
  413. ('--noisy', NS(verbose=True)),
  414. ]
  415. class TestOptionalsDest(ParserTestCase):
  416. """Tests various means of setting destination"""
  417. argument_signatures = [Sig('--foo-bar'), Sig('--baz', dest='zabbaz')]
  418. failures = ['a']
  419. successes = [
  420. ('--foo-bar f', NS(foo_bar='f', zabbaz=None)),
  421. ('--baz g', NS(foo_bar=None, zabbaz='g')),
  422. ('--foo-bar h --baz i', NS(foo_bar='h', zabbaz='i')),
  423. ('--baz j --foo-bar k', NS(foo_bar='k', zabbaz='j')),
  424. ]
  425. class TestOptionalsDefault(ParserTestCase):
  426. """Tests specifying a default for an Optional"""
  427. argument_signatures = [Sig('-x'), Sig('-y', default=42)]
  428. failures = ['a']
  429. successes = [
  430. ('', NS(x=None, y=42)),
  431. ('-xx', NS(x='x', y=42)),
  432. ('-yy', NS(x=None, y='y')),
  433. ]
  434. class TestOptionalsNargsDefault(ParserTestCase):
  435. """Tests not specifying the number of args for an Optional"""
  436. argument_signatures = [Sig('-x')]
  437. failures = ['a', '-x']
  438. successes = [
  439. ('', NS(x=None)),
  440. ('-x a', NS(x='a')),
  441. ]
  442. class TestOptionalsNargs1(ParserTestCase):
  443. """Tests specifying the 1 arg for an Optional"""
  444. argument_signatures = [Sig('-x', nargs=1)]
  445. failures = ['a', '-x']
  446. successes = [
  447. ('', NS(x=None)),
  448. ('-x a', NS(x=['a'])),
  449. ]
  450. class TestOptionalsNargs3(ParserTestCase):
  451. """Tests specifying the 3 args for an Optional"""
  452. argument_signatures = [Sig('-x', nargs=3)]
  453. failures = ['a', '-x', '-x a', '-x a b', 'a -x', 'a -x b']
  454. successes = [
  455. ('', NS(x=None)),
  456. ('-x a b c', NS(x=['a', 'b', 'c'])),
  457. ]
  458. class TestOptionalsNargsOptional(ParserTestCase):
  459. """Tests specifying an Optional arg for an Optional"""
  460. argument_signatures = [
  461. Sig('-w', nargs='?'),
  462. Sig('-x', nargs='?', const=42),
  463. Sig('-y', nargs='?', default='spam'),
  464. Sig('-z', nargs='?', type=int, const='42', default='84'),
  465. ]
  466. failures = ['2']
  467. successes = [
  468. ('', NS(w=None, x=None, y='spam', z=84)),
  469. ('-w', NS(w=None, x=None, y='spam', z=84)),
  470. ('-w 2', NS(w='2', x=None, y='spam', z=84)),
  471. ('-x', NS(w=None, x=42, y='spam', z=84)),
  472. ('-x 2', NS(w=None, x='2', y='spam', z=84)),
  473. ('-y', NS(w=None, x=None, y=None, z=84)),
  474. ('-y 2', NS(w=None, x=None, y='2', z=84)),
  475. ('-z', NS(w=None, x=None, y='spam', z=42)),
  476. ('-z 2', NS(w=None, x=None, y='spam', z=2)),
  477. ]
  478. class TestOptionalsNargsZeroOrMore(ParserTestCase):
  479. """Tests specifying an args for an Optional that accepts zero or more"""
  480. argument_signatures = [
  481. Sig('-x', nargs='*'),
  482. Sig('-y', nargs='*', default='spam'),
  483. ]
  484. failures = ['a']
  485. successes = [
  486. ('', NS(x=None, y='spam')),
  487. ('-x', NS(x=[], y='spam')),
  488. ('-x a', NS(x=['a'], y='spam')),
  489. ('-x a b', NS(x=['a', 'b'], y='spam')),
  490. ('-y', NS(x=None, y=[])),
  491. ('-y a', NS(x=None, y=['a'])),
  492. ('-y a b', NS(x=None, y=['a', 'b'])),
  493. ]
  494. class TestOptionalsNargsOneOrMore(ParserTestCase):
  495. """Tests specifying an args for an Optional that accepts one or more"""
  496. argument_signatures = [
  497. Sig('-x', nargs='+'),
  498. Sig('-y', nargs='+', default='spam'),
  499. ]
  500. failures = ['a', '-x', '-y', 'a -x', 'a -y b']
  501. successes = [
  502. ('', NS(x=None, y='spam')),
  503. ('-x a', NS(x=['a'], y='spam')),
  504. ('-x a b', NS(x=['a', 'b'], y='spam')),
  505. ('-y a', NS(x=None, y=['a'])),
  506. ('-y a b', NS(x=None, y=['a', 'b'])),
  507. ]
  508. class TestOptionalsChoices(ParserTestCase):
  509. """Tests specifying the choices for an Optional"""
  510. argument_signatures = [
  511. Sig('-f', choices='abc'),
  512. Sig('-g', type=int, choices=range(5))]
  513. failures = ['a', '-f d', '-fad', '-ga', '-g 6']
  514. successes = [
  515. ('', NS(f=None, g=None)),
  516. ('-f a', NS(f='a', g=None)),
  517. ('-f c', NS(f='c', g=None)),
  518. ('-g 0', NS(f=None, g=0)),
  519. ('-g 03', NS(f=None, g=3)),
  520. ('-fb -g4', NS(f='b', g=4)),
  521. ]
  522. class TestOptionalsRequired(ParserTestCase):
  523. """Tests the an optional action that is required"""
  524. argument_signatures = [
  525. Sig('-x', type=int, required=True),
  526. ]
  527. failures = ['a', '']
  528. successes = [
  529. ('-x 1', NS(x=1)),
  530. ('-x42', NS(x=42)),
  531. ]
  532. class TestOptionalsActionStore(ParserTestCase):
  533. """Tests the store action for an Optional"""
  534. argument_signatures = [Sig('-x', action='store')]
  535. failures = ['a', 'a -x']
  536. successes = [
  537. ('', NS(x=None)),
  538. ('-xfoo', NS(x='foo')),
  539. ]
  540. class TestOptionalsActionStoreConst(ParserTestCase):
  541. """Tests the store_const action for an Optional"""
  542. argument_signatures = [Sig('-y', action='store_const', const=object)]
  543. failures = ['a']
  544. successes = [
  545. ('', NS(y=None)),
  546. ('-y', NS(y=object)),
  547. ]
  548. class TestOptionalsActionStoreFalse(ParserTestCase):
  549. """Tests the store_false action for an Optional"""
  550. argument_signatures = [Sig('-z', action='store_false')]
  551. failures = ['a', '-za', '-z a']
  552. successes = [
  553. ('', NS(z=True)),
  554. ('-z', NS(z=False)),
  555. ]
  556. class TestOptionalsActionStoreTrue(ParserTestCase):
  557. """Tests the store_true action for an Optional"""
  558. argument_signatures = [Sig('--apple', action='store_true')]
  559. failures = ['a', '--apple=b', '--apple b']
  560. successes = [
  561. ('', NS(apple=False)),
  562. ('--apple', NS(apple=True)),
  563. ]
  564. class TestOptionalsActionAppend(ParserTestCase):
  565. """Tests the append action for an Optional"""
  566. argument_signatures = [Sig('--baz', action='append')]
  567. failures = ['a', '--baz', 'a --baz', '--baz a b']
  568. successes = [
  569. ('', NS(baz=None)),
  570. ('--baz a', NS(baz=['a'])),
  571. ('--baz a --baz b', NS(baz=['a', 'b'])),
  572. ]
  573. class TestOptionalsActionAppendWithDefault(ParserTestCase):
  574. """Tests the append action for an Optional"""
  575. argument_signatures = [Sig('--baz', action='append', default=['X'])]
  576. failures = ['a', '--baz', 'a --baz', '--baz a b']
  577. successes = [
  578. ('', NS(baz=['X'])),
  579. ('--baz a', NS(baz=['X', 'a'])),
  580. ('--baz a --baz b', NS(baz=['X', 'a', 'b'])),
  581. ]
  582. class TestOptionalsActionAppendConst(ParserTestCase):
  583. """Tests the append_const action for an Optional"""
  584. argument_signatures = [
  585. Sig('-b', action='append_const', const=Exception),
  586. Sig('-c', action='append', dest='b'),
  587. ]
  588. failures = ['a', '-c', 'a -c', '-bx', '-b x']
  589. successes = [
  590. ('', NS(b=None)),
  591. ('-b', NS(b=[Exception])),
  592. ('-b -cx -b -cyz', NS(b=[Exception, 'x', Exception, 'yz'])),
  593. ]
  594. class TestOptionalsActionAppendConstWithDefault(ParserTestCase):
  595. """Tests the append_const action for an Optional"""
  596. argument_signatures = [
  597. Sig('-b', action='append_const', const=Exception, default=['X']),
  598. Sig('-c', action='append', dest='b'),
  599. ]
  600. failures = ['a', '-c', 'a -c', '-bx', '-b x']
  601. successes = [
  602. ('', NS(b=['X'])),
  603. ('-b', NS(b=['X', Exception])),
  604. ('-b -cx -b -cyz', NS(b=['X', Exception, 'x', Exception, 'yz'])),
  605. ]
  606. class TestOptionalsActionCount(ParserTestCase):
  607. """Tests the count action for an Optional"""
  608. argument_signatures = [Sig('-x', action='count')]
  609. failures = ['a', '-x a', '-x b', '-x a -x b']
  610. successes = [
  611. ('', NS(x=None)),
  612. ('-x', NS(x=1)),
  613. ]
  614. # ================
  615. # Positional tests
  616. # ================
  617. class TestPositionalsNargsNone(ParserTestCase):
  618. """Test a Positional that doesn't specify nargs"""
  619. argument_signatures = [Sig('foo')]
  620. failures = ['', '-x', 'a b']
  621. successes = [
  622. ('a', NS(foo='a')),
  623. ]
  624. class TestPositionalsNargs1(ParserTestCase):
  625. """Test a Positional that specifies an nargs of 1"""
  626. argument_signatures = [Sig('foo', nargs=1)]
  627. failures = ['', '-x', 'a b']
  628. successes = [
  629. ('a', NS(foo=['a'])),
  630. ]
  631. class TestPositionalsNargs2(ParserTestCase):
  632. """Test a Positional that specifies an nargs of 2"""
  633. argument_signatures = [Sig('foo', nargs=2)]
  634. failures = ['', 'a', '-x', 'a b c']
  635. successes = [
  636. ('a b', NS(foo=['a', 'b'])),
  637. ]
  638. class TestPositionalsNargsZeroOrMore(ParserTestCase):
  639. """Test a Positional that specifies unlimited nargs"""
  640. argument_signatures = [Sig('foo', nargs='*')]
  641. failures = ['-x']
  642. successes = [
  643. ('', NS(foo=[])),
  644. ('a', NS(foo=['a'])),
  645. ('a b', NS(foo=['a', 'b'])),
  646. ]
  647. class TestPositionalsNargsZeroOrMoreDefault(ParserTestCase):
  648. """Test a Positional that specifies unlimited nargs and a default"""
  649. argument_signatures = [Sig('foo', nargs='*', default='bar')]
  650. failures = ['-x']
  651. successes = [
  652. ('', NS(foo='bar')),
  653. ('a', NS(foo=['a'])),
  654. ('a b', NS(foo=['a', 'b'])),
  655. ]
  656. class TestPositionalsNargsOneOrMore(ParserTestCase):
  657. """Test a Positional that specifies one or more nargs"""
  658. argument_signatures = [Sig('foo', nargs='+')]
  659. failures = ['', '-x']
  660. successes = [
  661. ('a', NS(foo=['a'])),
  662. ('a b', NS(foo=['a', 'b'])),
  663. ]
  664. class TestPositionalsNargsOptional(ParserTestCase):
  665. """Tests an Optional Positional"""
  666. argument_signatures = [Sig('foo', nargs='?')]
  667. failures = ['-x', 'a b']
  668. successes = [
  669. ('', NS(foo=None)),
  670. ('a', NS(foo='a')),
  671. ]
  672. class TestPositionalsNargsOptionalDefault(ParserTestCase):
  673. """Tests an Optional Positional with a default value"""
  674. argument_signatures = [Sig('foo', nargs='?', default=42)]
  675. failures = ['-x', 'a b']
  676. successes = [
  677. ('', NS(foo=42)),
  678. ('a', NS(foo='a')),
  679. ]
  680. class TestPositionalsNargsOptionalConvertedDefault(ParserTestCase):
  681. """Tests an Optional Positional with a default value
  682. that needs to be converted to the appropriate type.
  683. """
  684. argument_signatures = [
  685. Sig('foo', nargs='?', type=int, default='42'),
  686. ]
  687. failures = ['-x', 'a b', '1 2']
  688. successes = [
  689. ('', NS(foo=42)),
  690. ('1', NS(foo=1)),
  691. ]
  692. class TestPositionalsNargsNoneNone(ParserTestCase):
  693. """Test two Positionals that don't specify nargs"""
  694. argument_signatures = [Sig('foo'), Sig('bar')]
  695. failures = ['', '-x', 'a', 'a b c']
  696. successes = [
  697. ('a b', NS(foo='a', bar='b')),
  698. ]
  699. class TestPositionalsNargsNone1(ParserTestCase):
  700. """Test a Positional with no nargs followed by one with 1"""
  701. argument_signatures = [Sig('foo'), Sig('bar', nargs=1)]
  702. failures = ['', '--foo', 'a', 'a b c']
  703. successes = [
  704. ('a b', NS(foo='a', bar=['b'])),
  705. ]
  706. class TestPositionalsNargs2None(ParserTestCase):
  707. """Test a Positional with 2 nargs followed by one with none"""
  708. argument_signatures = [Sig('foo', nargs=2), Sig('bar')]
  709. failures = ['', '--foo', 'a', 'a b', 'a b c d']
  710. successes = [
  711. ('a b c', NS(foo=['a', 'b'], bar='c')),
  712. ]
  713. class TestPositionalsNargsNoneZeroOrMore(ParserTestCase):
  714. """Test a Positional with no nargs followed by one with unlimited"""
  715. argument_signatures = [Sig('foo'), Sig('bar', nargs='*')]
  716. failures = ['', '--foo']
  717. successes = [
  718. ('a', NS(foo='a', bar=[])),
  719. ('a b', NS(foo='a', bar=['b'])),
  720. ('a b c', NS(foo='a', bar=['b', 'c'])),
  721. ]
  722. class TestPositionalsNargsNoneOneOrMore(ParserTestCase):
  723. """Test a Positional with no nargs followed by one with one or more"""
  724. argument_signatures = [Sig('foo'), Sig('bar', nargs='+')]
  725. failures = ['', '--foo', 'a']
  726. successes = [
  727. ('a b', NS(foo='a', bar=['b'])),
  728. ('a b c', NS(foo='a', bar=['b', 'c'])),
  729. ]
  730. class TestPositionalsNargsNoneOptional(ParserTestCase):
  731. """Test a Positional with no nargs followed by one with an Optional"""
  732. argument_signatures = [Sig('foo'), Sig('bar', nargs='?')]
  733. failures = ['', '--foo', 'a b c']
  734. successes = [
  735. ('a', NS(foo='a', bar=None)),
  736. ('a b', NS(foo='a', bar='b')),
  737. ]
  738. class TestPositionalsNargsZeroOrMoreNone(ParserTestCase):
  739. """Test a Positional with unlimited nargs followed by one with none"""
  740. argument_signatures = [Sig('foo', nargs='*'), Sig('bar')]
  741. failures = ['', '--foo']
  742. successes = [
  743. ('a', NS(foo=[], bar='a')),
  744. ('a b', NS(foo=['a'], bar='b')),
  745. ('a b c', NS(foo=['a', 'b'], bar='c')),
  746. ]
  747. class TestPositionalsNargsOneOrMoreNone(ParserTestCase):
  748. """Test a Positional with one or more nargs followed by one with none"""
  749. argument_signatures = [Sig('foo', nargs='+'), Sig('bar')]
  750. failures = ['', '--foo', 'a']
  751. successes = [
  752. ('a b', NS(foo=['a'], bar='b')),
  753. ('a b c', NS(foo=['a', 'b'], bar='c')),
  754. ]
  755. class TestPositionalsNargsOptionalNone(ParserTestCase):
  756. """Test a Positional with an Optional nargs followed by one with none"""
  757. argument_signatures = [Sig('foo', nargs='?', default=42), Sig('bar')]
  758. failures = ['', '--foo', 'a b c']
  759. successes = [
  760. ('a', NS(foo=42, bar='a')),
  761. ('a b', NS(foo='a', bar='b')),
  762. ]
  763. class TestPositionalsNargs2ZeroOrMore(ParserTestCase):
  764. """Test a Positional with 2 nargs followed by one with unlimited"""
  765. argument_signatures = [Sig('foo', nargs=2), Sig('bar', nargs='*')]
  766. failures = ['', '--foo', 'a']
  767. successes = [
  768. ('a b', NS(foo=['a', 'b'], bar=[])),
  769. ('a b c', NS(foo=['a', 'b'], bar=['c'])),
  770. ]
  771. class TestPositionalsNargs2OneOrMore(ParserTestCase):
  772. """Test a Positional with 2 nargs followed by one with one or more"""
  773. argument_signatures = [Sig('foo', nargs=2), Sig('bar', nargs='+')]
  774. failures = ['', '--foo', 'a', 'a b']
  775. successes = [
  776. ('a b c', NS(foo=['a', 'b'], bar=['c'])),
  777. ]
  778. class TestPositionalsNargs2Optional(ParserTestCase):
  779. """Test a Positional with 2 nargs followed by one optional"""
  780. argument_signatures = [Sig('foo', nargs=2), Sig('bar', nargs='?')]
  781. failures = ['', '--foo', 'a', 'a b c d']
  782. successes = [
  783. ('a b', NS(foo=['a', 'b'], bar=None)),
  784. ('a b c', NS(foo=['a', 'b'], bar='c')),
  785. ]
  786. class TestPositionalsNargsZeroOrMore1(ParserTestCase):
  787. """Test a Positional with unlimited nargs followed by one with 1"""
  788. argument_signatures = [Sig('foo', nargs='*'), Sig('bar', nargs=1)]
  789. failures = ['', '--foo', ]
  790. successes = [
  791. ('a', NS(foo=[], bar=['a'])),
  792. ('a b', NS(foo=['a'], bar=['b'])),
  793. ('a b c', NS(foo=['a', 'b'], bar=['c'])),
  794. ]
  795. class TestPositionalsNargsOneOrMore1(ParserTestCase):
  796. """Test a Positional with one or more nargs followed by one with 1"""
  797. argument_signatures = [Sig('foo', nargs='+'), Sig('bar', nargs=1)]
  798. failures = ['', '--foo', 'a']
  799. successes = [
  800. ('a b', NS(foo=['a'], bar=['b'])),
  801. ('a b c', NS(foo=['a', 'b'], bar=['c'])),
  802. ]
  803. class TestPositionalsNargsOptional1(ParserTestCase):
  804. """Test a Positional with an Optional nargs followed by one with 1"""
  805. argument_signatures = [Sig('foo', nargs='?'), Sig('bar', nargs=1)]
  806. failures = ['', '--foo', 'a b c']
  807. successes = [
  808. ('a', NS(foo=None, bar=['a'])),
  809. ('a b', NS(foo='a', bar=['b'])),
  810. ]
  811. class TestPositionalsNargsNoneZeroOrMore1(ParserTestCase):
  812. """Test three Positionals: no nargs, unlimited nargs and 1 nargs"""
  813. argument_signatures = [
  814. Sig('foo'),
  815. Sig('bar', nargs='*'),
  816. Sig('baz', nargs=1),
  817. ]
  818. failures = ['', '--foo', 'a']
  819. successes = [
  820. ('a b', NS(foo='a', bar=[], baz=['b'])),
  821. ('a b c', NS(foo='a', bar=['b'], baz=['c'])),
  822. ]
  823. class TestPositionalsNargsNoneOneOrMore1(ParserTestCase):
  824. """Test three Positionals: no nargs, one or more nargs and 1 nargs"""
  825. argument_signatures = [
  826. Sig('foo'),
  827. Sig('bar', nargs='+'),
  828. Sig('baz', nargs=1),
  829. ]
  830. failures = ['', '--foo', 'a', 'b']
  831. successes = [
  832. ('a b c', NS(foo='a', bar=['b'], baz=['c'])),
  833. ('a b c d', NS(foo='a', bar=['b', 'c'], baz=['d'])),
  834. ]
  835. class TestPositionalsNargsNoneOptional1(ParserTestCase):
  836. """Test three Positionals: no nargs, optional narg and 1 nargs"""
  837. argument_signatures = [
  838. Sig('foo'),
  839. Sig('bar', nargs='?', default=0.625),
  840. Sig('baz', nargs=1),
  841. ]
  842. failures = ['', '--foo', 'a']
  843. successes = [
  844. ('a b', NS(foo='a', bar=0.625, baz=['b'])),
  845. ('a b c', NS(foo='a', bar='b', baz=['c'])),
  846. ]
  847. class TestPositionalsNargsOptionalOptional(ParserTestCase):
  848. """Test two optional nargs"""
  849. argument_signatures = [
  850. Sig('foo', nargs='?'),
  851. Sig('bar', nargs='?', default=42),
  852. ]
  853. failures = ['--foo', 'a b c']
  854. successes = [
  855. ('', NS(foo=None, bar=42)),
  856. ('a', NS(foo='a', bar=42)),
  857. ('a b', NS(foo='a', bar='b')),
  858. ]
  859. class TestPositionalsNargsOptionalZeroOrMore(ParserTestCase):
  860. """Test an Optional narg followed by unlimited nargs"""
  861. argument_signatures = [Sig('foo', nargs='?'), Sig('bar', nargs='*')]
  862. failures = ['--foo']
  863. successes = [
  864. ('', NS(foo=None, bar=[])),
  865. ('a', NS(foo='a', bar=[])),
  866. ('a b', NS(foo='a', bar=['b'])),
  867. ('a b c', NS(foo='a', bar=['b', 'c'])),
  868. ]
  869. class TestPositionalsNargsOptionalOneOrMore(ParserTestCase):
  870. """Test an Optional narg followed by one or more nargs"""
  871. argument_signatures = [Sig('foo', nargs='?'), Sig('bar', nargs='+')]
  872. failures = ['', '--foo']
  873. successes = [
  874. ('a', NS(foo=None, bar=['a'])),
  875. ('a b', NS(foo='a', bar=['b'])),
  876. ('a b c', NS(foo='a', bar=['b', 'c'])),
  877. ]
  878. class TestPositionalsChoicesString(ParserTestCase):
  879. """Test a set of single-character choices"""
  880. argument_signatures = [Sig('spam', choices=set('abcdefg'))]
  881. failures = ['', '--foo', 'h', '42', 'ef']
  882. successes = [
  883. ('a', NS(spam='a')),
  884. ('g', NS(spam='g')),
  885. ]
  886. class TestPositionalsChoicesInt(ParserTestCase):
  887. """Test a set of integer choices"""
  888. argument_signatures = [Sig('spam', type=int, choices=range(20))]
  889. failures = ['', '--foo', 'h', '42', 'ef']
  890. successes = [
  891. ('4', NS(spam=4)),
  892. ('15', NS(spam=15)),
  893. ]
  894. class TestPositionalsActionAppend(ParserTestCase):
  895. """Test the 'append' action"""
  896. argument_signatures = [
  897. Sig('spam', action='append'),
  898. Sig('spam', action='append', nargs=2),
  899. ]
  900. failures = ['', '--foo', 'a', 'a b', 'a b c d']
  901. successes = [
  902. ('a b c', NS(spam=['a', ['b', 'c']])),
  903. ]
  904. # ========================================
  905. # Combined optionals and positionals tests
  906. # ========================================
  907. class TestOptionalsNumericAndPositionals(ParserTestCase):
  908. """Tests negative number args when numeric options are present"""
  909. argument_signatures = [
  910. Sig('x', nargs='?'),
  911. Sig('-4', dest='y', action='store_true'),
  912. ]
  913. failures = ['-2', '-315']
  914. successes = [
  915. ('', NS(x=None, y=False)),
  916. ('a', NS(x='a', y=False)),
  917. ('-4', NS(x=None, y=True)),
  918. ('-4 a', NS(x='a', y=True)),
  919. ]
  920. class TestOptionalsAlmostNumericAndPositionals(ParserTestCase):
  921. """Tests negative number args when almost numeric options are present"""
  922. argument_signatures = [
  923. Sig('x', nargs='?'),
  924. Sig('-k4', dest='y', action='store_true'),
  925. ]
  926. failures = ['-k3']
  927. successes = [
  928. ('', NS(x=None, y=False)),
  929. ('-2', NS(x='-2', y=False)),
  930. ('a', NS(x='a', y=False)),
  931. ('-k4', NS(x=None, y=True)),
  932. ('-k4 a', NS(x='a', y=True)),
  933. ]
  934. class TestEmptyAndSpaceContainingArguments(ParserTestCase):
  935. argument_signatures = [
  936. Sig('x', nargs='?'),
  937. Sig('-y', '--yyy', dest='y'),
  938. ]
  939. failures = ['-y']
  940. successes = [
  941. ([''], NS(x='', y=None)),
  942. (['a badger'], NS(x='a badger', y=None)),
  943. (['-a badger'], NS(x='-a badger', y=None)),
  944. (['-y', ''], NS(x=None, y='')),
  945. (['-y', 'a badger'], NS(x=None, y='a badger')),
  946. (['-y', '-a badger'], NS(x=None, y='-a badger')),
  947. (['--yyy=a badger'], NS(x=None, y='a badger')),
  948. (['--yyy=-a badger'], NS(x=None, y='-a badger')),
  949. ]
  950. class TestPrefixCharacterOnlyArguments(ParserTestCase):
  951. parser_signature = Sig(prefix_chars='-+')
  952. argument_signatures = [
  953. Sig('-', dest='x', nargs='?', const='badger'),
  954. Sig('+', dest='y', type=int, default=42),
  955. Sig('-+-', dest='z', action='store_true'),
  956. ]
  957. failures = ['-y', '+ -']
  958. successes = [
  959. ('', NS(x=None, y=42, z=False)),
  960. ('-', NS(x='badger', y=42, z=False)),
  961. ('- X', NS(x='X', y=42, z=False)),
  962. ('+ -3', NS(x=None, y=-3, z=False)),
  963. ('-+-', NS(x=None, y=42, z=True)),
  964. ('- ===', NS(x='===', y=42, z=False)),
  965. ]
  966. class TestNargsZeroOrMore(ParserTestCase):
  967. """Tests specifying an args for an Optional that accepts zero or more"""
  968. argument_signatures = [Sig('-x', nargs='*'), Sig('y', nargs='*')]
  969. failures = []
  970. successes = [
  971. ('', NS(x=None, y=[])),
  972. ('-x', NS(x=[], y=[])),
  973. ('-x a', NS(x=['a'], y=[])),
  974. ('-x a -- b', NS(x=['a'], y=['b'])),
  975. ('a', NS(x=None, y=['a'])),
  976. ('a -x', NS(x=[], y=['a'])),
  977. ('a -x b', NS(x=['b'], y=['a'])),
  978. ]
  979. class TestNargsRemainder(ParserTestCase):
  980. """Tests specifying a positional with nargs=REMAINDER"""
  981. argument_signatures = [Sig('x'), Sig('y', nargs='...'), Sig('-z')]
  982. failures = ['', '-z', '-z Z']
  983. successes = [
  984. ('X', NS(x='X', y=[], z=None)),
  985. ('-z Z X', NS(x='X', y=[], z='Z')),
  986. ('X A B -z Z', NS(x='X', y=['A', 'B', '-z', 'Z'], z=None)),
  987. ('X Y --foo', NS(x='X', y=['Y', '--foo'], z=None)),
  988. ]
  989. class TestOptionLike(ParserTestCase):
  990. """Tests options that may or may not be arguments"""
  991. argument_signatures = [
  992. Sig('-x', type=float),
  993. Sig('-3', type=float, dest='y'),
  994. Sig('z', nargs='*'),
  995. ]
  996. failures = ['-x', '-y2.5', '-xa', '-x -a',
  997. '-x -3', '-x -3.5', '-3 -3.5',
  998. '-x -2.5', '-x -2.5 a', '-3 -.5',
  999. 'a x -1', '-x -1 a', '-3 -1 a']
  1000. successes = [
  1001. ('', NS(x=None, y=None, z=[])),
  1002. ('-x 2.5', NS(x=2.5, y=None, z=[])),
  1003. ('-x 2.5 a', NS(x=2.5, y=None, z=['a'])),
  1004. ('-3.5', NS(x=None, y=0.5, z=[])),
  1005. ('-3-.5', NS(x=None, y=-0.5, z=[])),
  1006. ('-3 .5', NS(x=None, y=0.5, z=[])),
  1007. ('a -3.5', NS(x=None, y=0.5, z=['a'])),
  1008. ('a', NS(x=None, y=None, z=['a'])),
  1009. ('a -x 1', NS(x=1.0, y=None, z=['a'])),
  1010. ('-x 1 a', NS(x=1.0, y=None, z=['a'])),
  1011. ('-3 1 a', NS(x=None, y=1.0, z=['a'])),
  1012. ]
  1013. class TestDefaultSuppress(ParserTestCase):
  1014. """Test actions with suppressed defaults"""
  1015. argument_signatures = [
  1016. Sig('foo', nargs='?', default=argparse.SUPPRESS),
  1017. Sig('bar', nargs='*', default=argparse.SUPPRESS),
  1018. Sig('--baz', action='store_true', default=argparse.SUPPRESS),
  1019. ]
  1020. failures = ['-x']
  1021. successes = [
  1022. ('', NS()),
  1023. ('a', NS(foo='a')),
  1024. ('a b', NS(foo='a', bar=['b'])),
  1025. ('--baz', NS(baz=True)),
  1026. ('a --baz', NS(foo='a', baz=True)),
  1027. ('--baz a b', NS(foo='a', bar=['b'], baz=True)),
  1028. ]
  1029. class TestParserDefaultSuppress(ParserTestCase):
  1030. """Test actions with a parser-level default of SUPPRESS"""
  1031. parser_signature = Sig(argument_default=argparse.SUPPRESS)
  1032. argument_signatures = [
  1033. Sig('foo', nargs='?'),
  1034. Sig('bar', nargs='*'),
  1035. Sig('--baz', action='store_true'),
  1036. ]
  1037. failures = ['-x']
  1038. successes = [
  1039. ('', NS()),
  1040. ('a', NS(foo='a')),
  1041. ('a b', NS(foo='a', bar=['b'])),
  1042. ('--baz', NS(baz=True)),
  1043. ('a --baz', NS(foo='a', baz=True)),
  1044. ('--baz a b', NS(foo='a', bar=['b'], baz=True)),
  1045. ]
  1046. class TestParserDefault42(ParserTestCase):
  1047. """Test actions with a parser-level default of 42"""
  1048. parser_signature = Sig(argument_default=42, version='1.0')
  1049. argument_signatures = [
  1050. Sig('foo', nargs='?'),
  1051. Sig('bar', nargs='*'),
  1052. Sig('--baz', action='store_true'),
  1053. ]
  1054. failures = ['-x']
  1055. successes = [
  1056. ('', NS(foo=42, bar=42, baz=42)),
  1057. ('a', NS(foo='a', bar=42, baz=42)),
  1058. ('a b', NS(foo='a', bar=['b'], baz=42)),
  1059. ('--baz', NS(foo=42, bar=42, baz=True)),
  1060. ('a --baz', NS(foo='a', bar=42, baz=True)),
  1061. ('--baz a b', NS(foo='a', bar=['b'], baz=True)),
  1062. ]
  1063. class TestArgumentsFromFile(TempDirMixin, ParserTestCase):
  1064. """Test reading arguments from a file"""
  1065. def setUp(self):
  1066. super(TestArgumentsFromFile, self).setUp()
  1067. file_texts = [
  1068. ('hello', 'hello world!\n'),
  1069. ('recursive', '-a\n'
  1070. 'A\n'
  1071. '@hello'),
  1072. ('invalid', '@no-such-path\n'),
  1073. ]
  1074. for path, text in file_texts:
  1075. file = open(path, 'w')
  1076. file.write(text)
  1077. file.close()
  1078. parser_signature = Sig(fromfile_prefix_chars='@')
  1079. argument_signatures = [
  1080. Sig('-a'),
  1081. Sig('x'),
  1082. Sig('y', nargs='+'),
  1083. ]
  1084. failures = ['', '-b', 'X', '@invalid', '@missing']
  1085. successes = [
  1086. ('X Y', NS(a=None, x='X', y=['Y'])),
  1087. ('X -a A Y Z', NS(a='A', x='X', y=['Y', 'Z'])),
  1088. ('@hello X', NS(a=None, x='hello world!', y=['X'])),
  1089. ('X @hello', NS(a=None, x='X', y=['hello world!'])),
  1090. ('-a B @recursive Y Z', NS(a='A', x='hello world!', y=['Y', 'Z'])),
  1091. ('X @recursive Z -a B', NS(a='B', x='X', y=['hello world!', 'Z'])),
  1092. ]
  1093. class TestArgumentsFromFileConverter(TempDirMixin, ParserTestCase):
  1094. """Test reading arguments from a file"""
  1095. def setUp(self):
  1096. super(TestArgumentsFromFileConverter, self).setUp()
  1097. file_texts = [
  1098. ('hello', 'hello world!\n'),
  1099. ]
  1100. for path, text in file_texts:
  1101. file = open(path, 'w')
  1102. file.write(text)
  1103. file.close()
  1104. class FromFileConverterArgumentParser(ErrorRaisingArgumentParser):
  1105. def convert_arg_line_to_args(self, arg_line):
  1106. for arg in arg_line.split():
  1107. if not arg.strip():
  1108. continue
  1109. yield arg
  1110. parser_class = FromFileConverterArgumentParser
  1111. parser_signature = Sig(fromfile_prefix_chars='@')
  1112. argument_signatures = [
  1113. Sig('y', nargs='+'),
  1114. ]
  1115. failures = []
  1116. successes = [
  1117. ('@hello X', NS(y=['hello', 'world!', 'X'])),
  1118. ]
  1119. # =====================
  1120. # Type conversion tests
  1121. # =====================
  1122. class TestFileTypeRepr(TestCase):
  1123. def test_r(self):
  1124. type = argparse.FileType('r')
  1125. self.assertEqual("FileType('r')", repr(type))
  1126. def test_wb_1(self):
  1127. type = argparse.FileType('wb', 1)
  1128. self.assertEqual("FileType('wb', 1)", repr(type))
  1129. class RFile(object):
  1130. seen = {}
  1131. def __init__(self, name):
  1132. self.name = name
  1133. __hash__ = None
  1134. def __eq__(self, other):
  1135. if other in self.seen:
  1136. text = self.seen[other]
  1137. else:
  1138. text = self.seen[other] = other.read()
  1139. other.close()
  1140. if not isinstance(text, str):
  1141. text = text.decode('ascii')
  1142. return self.name == other.name == text
  1143. class TestFileTypeR(TempDirMixin, ParserTestCase):
  1144. """Test the FileType option/argument type for reading files"""
  1145. def setUp(self):
  1146. super(TestFileTypeR, self).setUp()
  1147. for file_name in ['foo', 'bar']:
  1148. file = open(os.path.join(self.temp_dir, file_name), 'w')
  1149. file.write(file_name)
  1150. file.close()
  1151. self.create_readonly_file('readonly')
  1152. argument_signatures = [
  1153. Sig('-x', type=argparse.FileType()),
  1154. Sig('spam', type=argparse.FileType('r')),
  1155. ]
  1156. failures = ['-x', '-x bar', 'non-existent-file.txt']
  1157. successes = [
  1158. ('foo', NS(x=None, spam=RFile('foo'))),
  1159. ('-x foo bar', NS(x=RFile('foo'), spam=RFile('bar'))),
  1160. ('bar -x foo', NS(x=RFile('foo'), spam=RFile('bar'))),
  1161. ('-x - -', NS(x=sys.stdin, spam=sys.stdin)),
  1162. ('readonly', NS(x=None, spam=RFile('readonly'))),
  1163. ]
  1164. class TestFileTypeRB(TempDirMixin, ParserTestCase):
  1165. """Test the FileType option/argument type for reading files"""
  1166. def setUp(self):
  1167. super(TestFileTypeRB, self).setUp()
  1168. for file_name in ['foo', 'bar']:
  1169. file = open(os.path.join(self.temp_dir, file_name), 'w')
  1170. file.write(file_name)
  1171. file.close()
  1172. argument_signatures = [
  1173. Sig('-x', type=argparse.FileType('rb')),
  1174. Sig('spam', type=argparse.FileType('rb')),
  1175. ]
  1176. failures = ['-x', '-x bar']
  1177. successes = [
  1178. ('foo', NS(x=None, spam=RFile('foo'))),
  1179. ('-x foo bar', NS(x=RFile('foo'), spam=RFile('bar'))),
  1180. ('bar -x foo', NS(x=RFile('foo'), spam=RFile('bar'))),
  1181. ('-x - -', NS(x=sys.stdin, spam=sys.stdin)),
  1182. ]
  1183. class WFile(object):
  1184. seen = set()
  1185. def __init__(self, name):
  1186. self.name = name
  1187. __hash__ = None
  1188. def __eq__(self, other):
  1189. if other not in self.seen:
  1190. text = 'Check that file is writable.'
  1191. if 'b' in other.mode:
  1192. text = text.encode('ascii')
  1193. other.write(text)
  1194. other.close()
  1195. self.seen.add(other)
  1196. return self.name == other.name
  1197. class TestFileTypeW(TempDirMixin, ParserTestCase):
  1198. """Test the FileType option/argument type for writing files"""
  1199. def setUp(self):
  1200. super(TestFileTypeW, self).setUp()
  1201. self.create_readonly_file('readonly')
  1202. argument_signatures = [
  1203. Sig('-x', type=argparse.FileType('w')),
  1204. Sig('spam', type=argparse.FileType('w')),
  1205. ]
  1206. failures = ['-x', '-x bar']
  1207. failures = ['-x', '-x bar', 'readonly']
  1208. successes = [
  1209. ('foo', NS(x=None, spam=WFile('foo'))),
  1210. ('-x foo bar', NS(x=WFile('foo'), spam=WFile('bar'))),
  1211. ('bar -x foo', NS(x=WFile('foo'), spam=WFile('bar'))),
  1212. ('-x - -', NS(x=sys.stdout, spam=sys.stdout)),
  1213. ]
  1214. class TestFileTypeWB(TempDirMixin, ParserTestCase):
  1215. argument_signatures = [
  1216. Sig('-x', type=argparse.FileType('wb')),
  1217. Sig('spam', type=argparse.FileType('wb')),
  1218. ]
  1219. failures = ['-x', '-x bar']
  1220. successes = [
  1221. ('foo', NS(x=None, spam=WFile('foo'))),
  1222. ('-x foo bar', NS(x=WFile('foo'), spam=WFile('bar'))),
  1223. ('bar -x foo', NS(x=WFile('foo'), spam=WFile('bar'))),
  1224. ('-x - -', NS(x=sys.stdout, spam=sys.stdout)),
  1225. ]
  1226. class TestTypeCallable(ParserTestCase):
  1227. """Test some callables as option/argument types"""
  1228. argument_signatures = [
  1229. Sig('--eggs', type=complex),
  1230. Sig('spam', type=float),
  1231. ]
  1232. failures = ['a', '42j', '--eggs a', '--eggs 2i']
  1233. successes = [
  1234. ('--eggs=42 42', NS(eggs=42, spam=42.0)),
  1235. ('--eggs 2j -- -1.5', NS(eggs=2j, spam=-1.5)),
  1236. ('1024.675', NS(eggs=None, spam=1024.675)),
  1237. ]
  1238. class TestTypeUserDefined(ParserTestCase):
  1239. """Test a user-defined option/argument type"""
  1240. class MyType(TestCase):
  1241. def __init__(self, value):
  1242. self.value = value
  1243. __hash__ = None
  1244. def __eq__(self, other):
  1245. return (type(self), self.value) == (type(other), other.value)
  1246. argument_signatures = [
  1247. Sig('-x', type=MyType),
  1248. Sig('spam', type=MyType),
  1249. ]
  1250. failures = []
  1251. successes = [
  1252. ('a -x b', NS(x=MyType('b'), spam=MyType('a'))),
  1253. ('-xf g', NS(x=MyType('f'), spam=MyType('g'))),
  1254. ]
  1255. class TestTypeClassicClass(ParserTestCase):
  1256. """Test a classic class type"""
  1257. class C:
  1258. def __init__(self, value):
  1259. self.value = value
  1260. __hash__ = None
  1261. def __eq__(self, other):
  1262. return (type(self), self.value) == (type(other), other.value)
  1263. argument_signatures = [
  1264. Sig('-x', type=C),
  1265. Sig('spam', type=C),
  1266. ]
  1267. failures = []
  1268. successes = [
  1269. ('a -x b', NS(x=C('b'), spam=C('a'))),
  1270. ('-xf g', NS(x=C('f'), spam=C('g'))),
  1271. ]
  1272. class TestTypeRegistration(TestCase):
  1273. """Test a user-defined type by registering it"""
  1274. def test(self):
  1275. def get_my_type(string):
  1276. return 'my_type{%s}' % string
  1277. parser = argparse.ArgumentParser()
  1278. parser.register('type', 'my_type', get_my_type)
  1279. parser.add_argument('-x', type='my_type')
  1280. parser.add_argument('y', type='my_type')
  1281. self.assertEqual(parser.parse_args('1'.split()),
  1282. NS(x=None, y='my_type{1}'))
  1283. self.assertEqual(parser.parse_args('-x 1 42'.split()),
  1284. NS(x='my_type{1}', y='my_type{42}'))
  1285. # ============
  1286. # Action tests
  1287. # ============
  1288. class TestActionUserDefined(ParserTestCase):
  1289. """Test a user-defined option/argument action"""
  1290. class OptionalAction(argparse.Action):
  1291. def __call__(self, parser, namespace, value, option_string=None):
  1292. try:
  1293. # check destination and option string
  1294. assert self.dest == 'spam', 'dest: %s' % self.dest
  1295. assert option_string == '-s', 'flag: %s' % option_string
  1296. # when option is before argument, badger=2, and when
  1297. # option is after argument, badger=<whatever was set>
  1298. expected_ns = NS(spam=0.25)
  1299. if value in [0.125, 0.625]:
  1300. expected_ns.badger = 2
  1301. elif value in [2.0]:
  1302. expected_ns.badger = 84
  1303. else:
  1304. raise AssertionError('value: %s' % value)
  1305. assert expected_ns == namespace, ('expected %s, got %s' %
  1306. (expected_ns, namespace))
  1307. except AssertionError:
  1308. e = sys.exc_info()[1]
  1309. raise ArgumentParserError('opt_action failed: %s' % e)
  1310. setattr(namespace, 'spam', value)
  1311. class PositionalAction(argparse.Action):
  1312. def __call__(self, parser, namespace, value, option_string=None):
  1313. try:
  1314. assert option_string is None, ('option_string: %s' %
  1315. option_string)
  1316. # check destination
  1317. assert self.dest == 'badger', 'dest: %s' % self.dest
  1318. # when argument is before option, spam=0.25, and when
  1319. # option is after argument, spam=<whatever was set>
  1320. expected_ns = NS(badger=2)
  1321. if value in [42, 84]:
  1322. expected_ns.spam = 0.25
  1323. elif value in [1]:
  1324. expected_ns.spam = 0.625
  1325. elif value in [2]:
  1326. expected_ns.spam = 0.125
  1327. else:
  1328. raise AssertionError('value: %s' % value)
  1329. assert expected_ns == namespace, ('expected %s, got %s' %
  1330. (expected_ns, namespace))
  1331. except AssertionError:
  1332. e = sys.exc_info()[1]
  1333. raise ArgumentParserError('arg_action failed: %s' % e)
  1334. setattr(namespace, 'badger', value)
  1335. argument_signatures = [
  1336. Sig('-s', dest='spam', action=OptionalAction,
  1337. type=float, default=0.25),
  1338. Sig('badger', action=PositionalAction,
  1339. type=int, nargs='?', default=2),
  1340. ]
  1341. failures = []
  1342. successes = [
  1343. ('-s0.125', NS(spam=0.125, badger=2)),
  1344. ('42', NS(spam=0.25, badger=42)),
  1345. ('-s 0.625 1', NS(spam=0.625, badger=1)),
  1346. ('84 -s2', NS(spam=2.0, badger=84)),
  1347. ]
  1348. class TestActionRegistration(TestCase):
  1349. """Test a user-defined action supplied by registering it"""
  1350. class MyAction(argparse.Action):
  1351. def __call__(self, parser, namespace, values, option_string=None):
  1352. setattr(namespace, self.dest, 'foo[%s]' % values)
  1353. def test(self):
  1354. parser = argparse.ArgumentParser()
  1355. parser.register('action', 'my_action', self.MyAction)
  1356. parser.add_argument('badger', action='my_action')
  1357. self.assertEqual(parser.parse_args(['1']), NS(badger='foo[1]'))
  1358. self.assertEqual(parser.parse_args(['42']), NS(badger='foo[42]'))
  1359. # ================
  1360. # Subparsers tests
  1361. # ================
  1362. class TestAddSubparsers(TestCase):
  1363. """Test the add_subparsers method"""
  1364. def assertArgumentParserError(self, *args, **kwargs):
  1365. self.assertRaises(ArgumentParserError, *args, **kwargs)
  1366. def _get_parser(self, subparser_help=False, prefix_chars=None):
  1367. # create a parser with a subparsers argument
  1368. if prefix_chars:
  1369. parser = ErrorRaisingArgumentParser(
  1370. prog='PROG', description='main description', prefix_chars=prefix_chars)
  1371. parser.add_argument(
  1372. prefix_chars[0] * 2 + 'foo', action='store_true', help='foo help')
  1373. else:
  1374. parser = ErrorRaisingArgumentParser(
  1375. prog='PROG', description='main description')
  1376. parser.add_argument(
  1377. '--foo', action='store_true', help='foo help')
  1378. parser.add_argument(
  1379. 'bar', type=float, help='bar help')
  1380. # check that only one subparsers argument can be added
  1381. subparsers = parser.add_subparsers(help='command help')
  1382. self.assertArgumentParserError(parser.add_subparsers)
  1383. # add first sub-parser
  1384. parser1_kwargs = dict(description='1 description')
  1385. if subparser_help:
  1386. parser1_kwargs['help'] = '1 help'
  1387. parser1 = subparsers.add_parser('1', **parser1_kwargs)
  1388. parser1.add_argument('-w', type=int, help='w help')
  1389. parser1.add_argument('x', choices='abc', help='x help')
  1390. # add second sub-parser
  1391. parser2_kwargs = dict(description='2 description')
  1392. if subparser_help:
  1393. parser2_kwargs['help'] = '2 help'
  1394. parser2 = subparsers.add_parser('2', **parser2_kwargs)
  1395. parser2.add_argument('-y', choices='123', help='y help')
  1396. parser2.add_argument('z', type=complex, nargs='*', help='z help')
  1397. # return the main parser
  1398. return parser
  1399. def setUp(self):
  1400. super(TestAddSubparsers, self).setUp()
  1401. self.parser = self._get_parser()
  1402. self.command_help_parser = self._get_parser(subparser_help=True)
  1403. def test_parse_args_failures(self):
  1404. # check some failure cases:
  1405. for args_str in ['', 'a', 'a a', '0.5 a', '0.5 1',
  1406. '0.5 1 -y', '0.5 2 -w']:
  1407. args = args_str.split()
  1408. self.assertArgumentParserError(self.parser.parse_args, args)
  1409. def test_parse_args(self):
  1410. # check some non-failure cases:
  1411. self.assertEqual(
  1412. self.parser.parse_args('0.5 1 b -w 7'.split()),
  1413. NS(foo=False, bar=0.5, w=7, x='b'),
  1414. )
  1415. self.assertEqual(
  1416. self.parser.parse_args('0.25 --foo 2 -y 2 3j -- -1j'.split()),
  1417. NS(foo=True, bar=0.25, y='2', z=[3j, -1j]),
  1418. )
  1419. self.assertEqual(
  1420. self.parser.parse_args('--foo 0.125 1 c'.split()),
  1421. NS(foo=True, bar=0.125, w=None, x='c'),
  1422. )
  1423. def test_parse_known_args(self):
  1424. self.assertEqual(
  1425. self.parser.parse_known_args('0.5 1 b -w 7'.split()),
  1426. (NS(foo=False, bar=0.5, w=7, x='b'), []),
  1427. )
  1428. self.assertEqual(
  1429. self.parser.parse_known_args('0.5 -p 1 b -w 7'.split()),
  1430. (NS(foo=False, bar=0.5, w=7, x='b'), ['-p']),
  1431. )
  1432. self.assertEqual(
  1433. self.parser.parse_known_args('0.5 1 b -w 7 -p'.split()),
  1434. (NS(foo=False, bar=0.5, w=7, x='b'), ['-p']),
  1435. )
  1436. self.assertEqual(
  1437. self.parser.parse_known_args('0.5 1 b -q -rs -w 7'.split()),
  1438. (NS(foo=False, bar=0.5, w=7, x='b'), ['-q', '-rs']),
  1439. )
  1440. self.assertEqual(
  1441. self.parser.parse_known_args('0.5 -W 1 b -X Y -w 7 Z'.split()),
  1442. (NS(foo=False, bar=0.5, w=7, x='b'), ['-W', '-X', 'Y', 'Z']),
  1443. )
  1444. def test_dest(self):
  1445. parser = ErrorRaisingArgumentParser()
  1446. parser.add_argument('--foo', action='store_true')
  1447. subparsers = parser.add_subparsers(dest='bar')
  1448. parser1 = subparsers.add_parser('1')
  1449. parser1.add_argument('baz')
  1450. self.assertEqual(NS(foo=False, bar='1', baz='2'),
  1451. parser.parse_args('1 2'.split()))
  1452. def test_help(self):
  1453. self.assertEqual(self.parser.format_usage(),
  1454. 'usage: PROG [-h] [--foo] bar {1,2} ...\n')
  1455. self.assertEqual(self.parser.format_help(), textwrap.dedent('''\
  1456. usage: PROG [-h] [--foo] bar {1,2} ...
  1457. main description
  1458. positional arguments:
  1459. bar bar help
  1460. {1,2} command help
  1461. optional arguments:
  1462. -h, --help show this help message and exit
  1463. --foo foo help
  1464. '''))
  1465. def test_help_extra_prefix_chars(self):
  1466. # Make sure - is still used for help if it is a non-first prefix char
  1467. parser = self._get_parser(prefix_chars='+:-')
  1468. self.assertEqual(parser.format_usage(),
  1469. 'usage: PROG [-h] [++foo] bar {1,2} ...\n')
  1470. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  1471. usage: PROG [-h] [++foo] bar {1,2} ...
  1472. main description
  1473. positional arguments:
  1474. bar bar help
  1475. {1,2} command help
  1476. optional arguments:
  1477. -h, --help show this help message and exit
  1478. ++foo foo help
  1479. '''))
  1480. def test_help_alternate_prefix_chars(self):
  1481. parser = self._get_parser(prefix_chars='+:/')
  1482. self.assertEqual(parser.format_usage(),
  1483. 'usage: PROG [+h] [++foo] bar {1,2} ...\n')
  1484. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  1485. usage: PROG [+h] [++foo] bar {1,2} ...
  1486. main description
  1487. positional arguments:
  1488. bar bar help
  1489. {1,2} command help
  1490. optional arguments:
  1491. +h, ++help show this help message and exit
  1492. ++foo foo help
  1493. '''))
  1494. def test_parser_command_help(self):
  1495. self.assertEqual(self.command_help_parser.format_usage(),
  1496. 'usage: PROG [-h] [--foo] bar {1,2} ...\n')
  1497. self.assertEqual(self.command_help_parser.format_help(),
  1498. textwrap.dedent('''\
  1499. usage: PROG [-h] [--foo] bar {1,2} ...
  1500. main description
  1501. positional arguments:
  1502. bar bar help
  1503. {1,2} command help
  1504. 1 1 help
  1505. 2 2 help
  1506. optional arguments:
  1507. -h, --help show this help message and exit
  1508. --foo foo help
  1509. '''))
  1510. def test_subparser_title_help(self):
  1511. parser = ErrorRaisingArgumentParser(prog='PROG',
  1512. description='main description')
  1513. parser.add_argument('--foo', action='store_true', help='foo help')
  1514. parser.add_argument('bar', help='bar help')
  1515. subparsers = parser.add_subparsers(title='subcommands',
  1516. description='command help',
  1517. help='additional text')
  1518. parser1 = subparsers.add_parser('1')
  1519. parser2 = subparsers.add_parser('2')
  1520. self.assertEqual(parser.format_usage(),
  1521. 'usage: PROG [-h] [--foo] bar {1,2} ...\n')
  1522. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  1523. usage: PROG [-h] [--foo] bar {1,2} ...
  1524. main description
  1525. positional arguments:
  1526. bar bar help
  1527. optional arguments:
  1528. -h, --help show this help message and exit
  1529. --foo foo help
  1530. subcommands:
  1531. command help
  1532. {1,2} additional text
  1533. '''))
  1534. def _test_subparser_help(self, args_str, expected_help):
  1535. try:
  1536. self.parser.parse_args(args_str.split())
  1537. except ArgumentParserError:
  1538. err = sys.exc_info()[1]
  1539. if err.stdout != expected_help:
  1540. print(repr(expected_help))
  1541. print(repr(err.stdout))
  1542. self.assertEqual(err.stdout, expected_help)
  1543. def test_subparser1_help(self):
  1544. self._test_subparser_help('5.0 1 -h', textwrap.dedent('''\
  1545. usage: PROG bar 1 [-h] [-w W] {a,b,c}
  1546. 1 description
  1547. positional arguments:
  1548. {a,b,c} x help
  1549. optional arguments:
  1550. -h, --help show this help message and exit
  1551. -w W w help
  1552. '''))
  1553. def test_subparser2_help(self):
  1554. self._test_subparser_help('5.0 2 -h', textwrap.dedent('''\
  1555. usage: PROG bar 2 [-h] [-y {1,2,3}] [z [z ...]]
  1556. 2 description
  1557. positional arguments:
  1558. z z help
  1559. optional arguments:
  1560. -h, --help show this help message and exit
  1561. -y {1,2,3} y help
  1562. '''))
  1563. # ============
  1564. # Groups tests
  1565. # ============
  1566. class TestPositionalsGroups(TestCase):
  1567. """Tests that order of group positionals matches construction order"""
  1568. def test_nongroup_first(self):
  1569. parser = ErrorRaisingArgumentParser()
  1570. parser.add_argument('foo')
  1571. group = parser.add_argument_group('g')
  1572. group.add_argument('bar')
  1573. parser.add_argument('baz')
  1574. expected = NS(foo='1', bar='2', baz='3')
  1575. result = parser.parse_args('1 2 3'.split())
  1576. self.assertEqual(expected, result)
  1577. def test_group_first(self):
  1578. parser = ErrorRaisingArgumentParser()
  1579. group = parser.add_argument_group('xxx')
  1580. group.add_argument('foo')
  1581. parser.add_argument('bar')
  1582. parser.add_argument('baz')
  1583. expected = NS(foo='1', bar='2', baz='3')
  1584. result = parser.parse_args('1 2 3'.split())
  1585. self.assertEqual(expected, result)
  1586. def test_interleaved_groups(self):
  1587. parser = ErrorRaisingArgumentParser()
  1588. group = parser.add_argument_group('xxx')
  1589. parser.add_argument('foo')
  1590. group.add_argument('bar')
  1591. parser.add_argument('baz')
  1592. group = parser.add_argument_group('yyy')
  1593. group.add_argument('frell')
  1594. expected = NS(foo='1', bar='2', baz='3', frell='4')
  1595. result = parser.parse_args('1 2 3 4'.split())
  1596. self.assertEqual(expected, result)
  1597. # ===================
  1598. # Parent parser tests
  1599. # ===================
  1600. class TestParentParsers(TestCase):
  1601. """Tests that parsers can be created with parent parsers"""
  1602. def assertArgumentParserError(self, *args, **kwargs):
  1603. self.assertRaises(ArgumentParserError, *args, **kwargs)
  1604. def setUp(self):
  1605. super(TestParentParsers, self).setUp()
  1606. self.wxyz_parent = ErrorRaisingArgumentParser(add_help=False)
  1607. self.wxyz_parent.add_argument('--w')
  1608. x_group = self.wxyz_parent.add_argument_group('x')
  1609. x_group.add_argument('-y')
  1610. self.wxyz_parent.add_argument('z')
  1611. self.abcd_parent = ErrorRaisingArgumentParser(add_help=False)
  1612. self.abcd_parent.add_argument('a')
  1613. self.abcd_parent.add_argument('-b')
  1614. c_group = self.abcd_parent.add_argument_group('c')
  1615. c_group.add_argument('--d')
  1616. self.w_parent = ErrorRaisingArgumentParser(add_help=False)
  1617. self.w_parent.add_argument('--w')
  1618. self.z_parent = ErrorRaisingArgumentParser(add_help=False)
  1619. self.z_parent.add_argument('z')
  1620. # parents with mutually exclusive groups
  1621. self.ab_mutex_parent = ErrorRaisingArgumentParser(add_help=False)
  1622. group = self.ab_mutex_parent.add_mutually_exclusive_group()
  1623. group.add_argument('-a', action='store_true')
  1624. group.add_argument('-b', action='store_true')
  1625. self.main_program = os.path.basename(sys.argv[0])
  1626. def test_single_parent(self):
  1627. parser = ErrorRaisingArgumentParser(parents=[self.wxyz_parent])
  1628. self.assertEqual(parser.parse_args('-y 1 2 --w 3'.split()),
  1629. NS(w='3', y='1', z='2'))
  1630. def test_single_parent_mutex(self):
  1631. self._test_mutex_ab(self.ab_mutex_parent.parse_args)
  1632. parser = ErrorRaisingArgumentParser(parents=[self.ab_mutex_parent])
  1633. self._test_mutex_ab(parser.parse_args)
  1634. def test_single_granparent_mutex(self):
  1635. parents = [self.ab_mutex_parent]
  1636. parser = ErrorRaisingArgumentParser(add_help=False, parents=parents)
  1637. parser = ErrorRaisingArgumentParser(parents=[parser])
  1638. self._test_mutex_ab(parser.parse_args)
  1639. def _test_mutex_ab(self, parse_args):
  1640. self.assertEqual(parse_args([]), NS(a=False, b=False))
  1641. self.assertEqual(parse_args(['-a']), NS(a=True, b=False))
  1642. self.assertEqual(parse_args(['-b']), NS(a=False, b=True))
  1643. self.assertArgumentParserError(parse_args, ['-a', '-b'])
  1644. self.assertArgumentParserError(parse_args, ['-b', '-a'])
  1645. self.assertArgumentParserError(parse_args, ['-c'])
  1646. self.assertArgumentParserError(parse_args, ['-a', '-c'])
  1647. self.assertArgumentParserError(parse_args, ['-b', '-c'])
  1648. def test_multiple_parents(self):
  1649. parents = [self.abcd_parent, self.wxyz_parent]
  1650. parser = ErrorRaisingArgumentParser(parents=parents)
  1651. self.assertEqual(parser.parse_args('--d 1 --w 2 3 4'.split()),
  1652. NS(a='3', b=None, d='1', w='2', y=None, z='4'))
  1653. def test_multiple_parents_mutex(self):
  1654. parents = [self.ab_mutex_parent, self.wxyz_parent]
  1655. parser = ErrorRaisingArgumentParser(parents=parents)
  1656. self.assertEqual(parser.parse_args('-a --w 2 3'.split()),
  1657. NS(a=True, b=False, w='2', y=None, z='3'))
  1658. self.assertArgumentParserError(
  1659. parser.parse_args, '-a --w 2 3 -b'.split())
  1660. self.assertArgumentParserError(
  1661. parser.parse_args, '-a -b --w 2 3'.split())
  1662. def test_conflicting_parents(self):
  1663. self.assertRaises(
  1664. argparse.ArgumentError,
  1665. argparse.ArgumentParser,
  1666. parents=[self.w_parent, self.wxyz_parent])
  1667. def test_conflicting_parents_mutex(self):
  1668. self.assertRaises(
  1669. argparse.ArgumentError,
  1670. argparse.ArgumentParser,
  1671. parents=[self.abcd_parent, self.ab_mutex_parent])
  1672. def test_same_argument_name_parents(self):
  1673. parents = [self.wxyz_parent, self.z_parent]
  1674. parser = ErrorRaisingArgumentParser(parents=parents)
  1675. self.assertEqual(parser.parse_args('1 2'.split()),
  1676. NS(w=None, y=None, z='2'))
  1677. def test_subparser_parents(self):
  1678. parser = ErrorRaisingArgumentParser()
  1679. subparsers = parser.add_subparsers()
  1680. abcde_parser = subparsers.add_parser('bar', parents=[self.abcd_parent])
  1681. abcde_parser.add_argument('e')
  1682. self.assertEqual(parser.parse_args('bar -b 1 --d 2 3 4'.split()),
  1683. NS(a='3', b='1', d='2', e='4'))
  1684. def test_subparser_parents_mutex(self):
  1685. parser = ErrorRaisingArgumentParser()
  1686. subparsers = parser.add_subparsers()
  1687. parents = [self.ab_mutex_parent]
  1688. abc_parser = subparsers.add_parser('foo', parents=parents)
  1689. c_group = abc_parser.add_argument_group('c_group')
  1690. c_group.add_argument('c')
  1691. parents = [self.wxyz_parent, self.ab_mutex_parent]
  1692. wxyzabe_parser = subparsers.add_parser('bar', parents=parents)
  1693. wxyzabe_parser.add_argument('e')
  1694. self.assertEqual(parser.parse_args('foo -a 4'.split()),
  1695. NS(a=True, b=False, c='4'))
  1696. self.assertEqual(parser.parse_args('bar -b --w 2 3 4'.split()),
  1697. NS(a=False, b=True, w='2', y=None, z='3', e='4'))
  1698. self.assertArgumentParserError(
  1699. parser.parse_args, 'foo -a -b 4'.split())
  1700. self.assertArgumentParserError(
  1701. parser.parse_args, 'bar -b -a 4'.split())
  1702. def test_parent_help(self):
  1703. parents = [self.abcd_parent, self.wxyz_parent]
  1704. parser = ErrorRaisingArgumentParser(parents=parents)
  1705. parser_help = parser.format_help()
  1706. self.assertEqual(parser_help, textwrap.dedent('''\
  1707. usage: {} [-h] [-b B] [--d D] [--w W] [-y Y] a z
  1708. positional arguments:
  1709. a
  1710. z
  1711. optional arguments:
  1712. -h, --help show this help message and exit
  1713. -b B
  1714. --w W
  1715. c:
  1716. --d D
  1717. x:
  1718. -y Y
  1719. '''.format(self.main_program)))
  1720. def test_groups_parents(self):
  1721. parent = ErrorRaisingArgumentParser(add_help=False)
  1722. g = parent.add_argument_group(title='g', description='gd')
  1723. g.add_argument('-w')
  1724. g.add_argument('-x')
  1725. m = parent.add_mutually_exclusive_group()
  1726. m.add_argument('-y')
  1727. m.add_argument('-z')
  1728. parser = ErrorRaisingArgumentParser(parents=[parent])
  1729. self.assertRaises(ArgumentParserError, parser.parse_args,
  1730. ['-y', 'Y', '-z', 'Z'])
  1731. parser_help = parser.format_help()
  1732. self.assertEqual(parser_help, textwrap.dedent('''\
  1733. usage: {} [-h] [-w W] [-x X] [-y Y | -z Z]
  1734. optional arguments:
  1735. -h, --help show this help message and exit
  1736. -y Y
  1737. -z Z
  1738. g:
  1739. gd
  1740. -w W
  1741. -x X
  1742. '''.format(self.main_program)))
  1743. # ==============================
  1744. # Mutually exclusive group tests
  1745. # ==============================
  1746. class TestMutuallyExclusiveGroupErrors(TestCase):
  1747. def test_invalid_add_argument_group(self):
  1748. parser = ErrorRaisingArgumentParser()
  1749. raises = self.assertRaises
  1750. raises(TypeError, parser.add_mutually_exclusive_group, title='foo')
  1751. def test_invalid_add_argument(self):
  1752. parser = ErrorRaisingArgumentParser()
  1753. group = parser.add_mutually_exclusive_group()
  1754. add_argument = group.add_argument
  1755. raises = self.assertRaises
  1756. raises(ValueError, add_argument, '--foo', required=True)
  1757. raises(ValueError, add_argument, 'bar')
  1758. raises(ValueError, add_argument, 'bar', nargs='+')
  1759. raises(ValueError, add_argument, 'bar', nargs=1)
  1760. raises(ValueError, add_argument, 'bar', nargs=argparse.PARSER)
  1761. def test_help(self):
  1762. parser = ErrorRaisingArgumentParser(prog='PROG')
  1763. group1 = parser.add_mutually_exclusive_group()
  1764. group1.add_argument('--foo', action='store_true')
  1765. group1.add_argument('--bar', action='store_false')
  1766. group2 = parser.add_mutually_exclusive_group()
  1767. group2.add_argument('--soup', action='store_true')
  1768. group2.add_argument('--nuts', action='store_false')
  1769. expected = '''\
  1770. usage: PROG [-h] [--foo | --bar] [--soup | --nuts]
  1771. optional arguments:
  1772. -h, --help show this help message and exit
  1773. --foo
  1774. --bar
  1775. --soup
  1776. --nuts
  1777. '''
  1778. self.assertEqual(parser.format_help(), textwrap.dedent(expected))
  1779. class MEMixin(object):
  1780. def test_failures_when_not_required(self):
  1781. parse_args = self.get_parser(required=False).parse_args
  1782. error = ArgumentParserError
  1783. for args_string in self.failures:
  1784. self.assertRaises(error, parse_args, args_string.split())
  1785. def test_failures_when_required(self):
  1786. parse_args = self.get_parser(required=True).parse_args
  1787. error = ArgumentParserError
  1788. for args_string in self.failures + ['']:
  1789. self.assertRaises(error, parse_args, args_string.split())
  1790. def test_successes_when_not_required(self):
  1791. parse_args = self.get_parser(required=False).parse_args
  1792. successes = self.successes + self.successes_when_not_required
  1793. for args_string, expected_ns in successes:
  1794. actual_ns = parse_args(args_string.split())
  1795. self.assertEqual(actual_ns, expected_ns)
  1796. def test_successes_when_required(self):
  1797. parse_args = self.get_parser(required=True).parse_args
  1798. for args_string, expected_ns in self.successes:
  1799. actual_ns = parse_args(args_string.split())
  1800. self.assertEqual(actual_ns, expected_ns)
  1801. def test_usage_when_not_required(self):
  1802. format_usage = self.get_parser(required=False).format_usage
  1803. expected_usage = self.usage_when_not_required
  1804. self.assertEqual(format_usage(), textwrap.dedent(expected_usage))
  1805. def test_usage_when_required(self):
  1806. format_usage = self.get_parser(required=True).format_usage
  1807. expected_usage = self.usage_when_required
  1808. self.assertEqual(format_usage(), textwrap.dedent(expected_usage))
  1809. def test_help_when_not_required(self):
  1810. format_help = self.get_parser(required=False).format_help
  1811. help = self.usage_when_not_required + self.help
  1812. self.assertEqual(format_help(), textwrap.dedent(help))
  1813. def test_help_when_required(self):
  1814. format_help = self.get_parser(required=True).format_help
  1815. help = self.usage_when_required + self.help
  1816. self.assertEqual(format_help(), textwrap.dedent(help))
  1817. class TestMutuallyExclusiveSimple(MEMixin, TestCase):
  1818. def get_parser(self, required=None):
  1819. parser = ErrorRaisingArgumentParser(prog='PROG')
  1820. group = parser.add_mutually_exclusive_group(required=required)
  1821. group.add_argument('--bar', help='bar help')
  1822. group.add_argument('--baz', nargs='?', const='Z', help='baz help')
  1823. return parser
  1824. failures = ['--bar X --baz Y', '--bar X --baz']
  1825. successes = [
  1826. ('--bar X', NS(bar='X', baz=None)),
  1827. ('--bar X --bar Z', NS(bar='Z', baz=None)),
  1828. ('--baz Y', NS(bar=None, baz='Y')),
  1829. ('--baz', NS(bar=None, baz='Z')),
  1830. ]
  1831. successes_when_not_required = [
  1832. ('', NS(bar=None, baz=None)),
  1833. ]
  1834. usage_when_not_required = '''\
  1835. usage: PROG [-h] [--bar BAR | --baz [BAZ]]
  1836. '''
  1837. usage_when_required = '''\
  1838. usage: PROG [-h] (--bar BAR | --baz [BAZ])
  1839. '''
  1840. help = '''\
  1841. optional arguments:
  1842. -h, --help show this help message and exit
  1843. --bar BAR bar help
  1844. --baz [BAZ] baz help
  1845. '''
  1846. class TestMutuallyExclusiveLong(MEMixin, TestCase):
  1847. def get_parser(self, required=None):
  1848. parser = ErrorRaisingArgumentParser(prog='PROG')
  1849. parser.add_argument('--abcde', help='abcde help')
  1850. parser.add_argument('--fghij', help='fghij help')
  1851. group = parser.add_mutually_exclusive_group(required=required)
  1852. group.add_argument('--klmno', help='klmno help')
  1853. group.add_argument('--pqrst', help='pqrst help')
  1854. return parser
  1855. failures = ['--klmno X --pqrst Y']
  1856. successes = [
  1857. ('--klmno X', NS(abcde=None, fghij=None, klmno='X', pqrst=None)),
  1858. ('--abcde Y --klmno X',
  1859. NS(abcde='Y', fghij=None, klmno='X', pqrst=None)),
  1860. ('--pqrst X', NS(abcde=None, fghij=None, klmno=None, pqrst='X')),
  1861. ('--pqrst X --fghij Y',
  1862. NS(abcde=None, fghij='Y', klmno=None, pqrst='X')),
  1863. ]
  1864. successes_when_not_required = [
  1865. ('', NS(abcde=None, fghij=None, klmno=None, pqrst=None)),
  1866. ]
  1867. usage_when_not_required = '''\
  1868. usage: PROG [-h] [--abcde ABCDE] [--fghij FGHIJ]
  1869. [--klmno KLMNO | --pqrst PQRST]
  1870. '''
  1871. usage_when_required = '''\
  1872. usage: PROG [-h] [--abcde ABCDE] [--fghij FGHIJ]
  1873. (--klmno KLMNO | --pqrst PQRST)
  1874. '''
  1875. help = '''\
  1876. optional arguments:
  1877. -h, --help show this help message and exit
  1878. --abcde ABCDE abcde help
  1879. --fghij FGHIJ fghij help
  1880. --klmno KLMNO klmno help
  1881. --pqrst PQRST pqrst help
  1882. '''
  1883. class TestMutuallyExclusiveFirstSuppressed(MEMixin, TestCase):
  1884. def get_parser(self, required):
  1885. parser = ErrorRaisingArgumentParser(prog='PROG')
  1886. group = parser.add_mutually_exclusive_group(required=required)
  1887. group.add_argument('-x', help=argparse.SUPPRESS)
  1888. group.add_argument('-y', action='store_false', help='y help')
  1889. return parser
  1890. failures = ['-x X -y']
  1891. successes = [
  1892. ('-x X', NS(x='X', y=True)),
  1893. ('-x X -x Y', NS(x='Y', y=True)),
  1894. ('-y', NS(x=None, y=False)),
  1895. ]
  1896. successes_when_not_required = [
  1897. ('', NS(x=None, y=True)),
  1898. ]
  1899. usage_when_not_required = '''\
  1900. usage: PROG [-h] [-y]
  1901. '''
  1902. usage_when_required = '''\
  1903. usage: PROG [-h] -y
  1904. '''
  1905. help = '''\
  1906. optional arguments:
  1907. -h, --help show this help message and exit
  1908. -y y help
  1909. '''
  1910. class TestMutuallyExclusiveManySuppressed(MEMixin, TestCase):
  1911. def get_parser(self, required):
  1912. parser = ErrorRaisingArgumentParser(prog='PROG')
  1913. group = parser.add_mutually_exclusive_group(required=required)
  1914. add = group.add_argument
  1915. add('--spam', action='store_true', help=argparse.SUPPRESS)
  1916. add('--badger', action='store_false', help=argparse.SUPPRESS)
  1917. add('--bladder', help=argparse.SUPPRESS)
  1918. return parser
  1919. failures = [
  1920. '--spam --badger',
  1921. '--badger --bladder B',
  1922. '--bladder B --spam',
  1923. ]
  1924. successes = [
  1925. ('--spam', NS(spam=True, badger=True, bladder=None)),
  1926. ('--badger', NS(spam=False, badger=False, bladder=None)),
  1927. ('--bladder B', NS(spam=False, badger=True, bladder='B')),
  1928. ('--spam --spam', NS(spam=True, badger=True, bladder=None)),
  1929. ]
  1930. successes_when_not_required = [
  1931. ('', NS(spam=False, badger=True, bladder=None)),
  1932. ]
  1933. usage_when_required = usage_when_not_required = '''\
  1934. usage: PROG [-h]
  1935. '''
  1936. help = '''\
  1937. optional arguments:
  1938. -h, --help show this help message and exit
  1939. '''
  1940. class TestMutuallyExclusiveOptionalAndPositional(MEMixin, TestCase):
  1941. def get_parser(self, required):
  1942. parser = ErrorRaisingArgumentParser(prog='PROG')
  1943. group = parser.add_mutually_exclusive_group(required=required)
  1944. group.add_argument('--foo', action='store_true', help='FOO')
  1945. group.add_argument('--spam', help='SPAM')
  1946. group.add_argument('badger', nargs='*', default='X', help='BADGER')
  1947. return parser
  1948. failures = [
  1949. '--foo --spam S',
  1950. '--spam S X',
  1951. 'X --foo',
  1952. 'X Y Z --spam S',
  1953. '--foo X Y',
  1954. ]
  1955. successes = [
  1956. ('--foo', NS(foo=True, spam=None, badger='X')),
  1957. ('--spam S', NS(foo=False, spam='S', badger='X')),
  1958. ('X', NS(foo=False, spam=None, badger=['X'])),
  1959. ('X Y Z', NS(foo=False, spam=None, badger=['X', 'Y', 'Z'])),
  1960. ]
  1961. successes_when_not_required = [
  1962. ('', NS(foo=False, spam=None, badger='X')),
  1963. ]
  1964. usage_when_not_required = '''\
  1965. usage: PROG [-h] [--foo | --spam SPAM | badger [badger ...]]
  1966. '''
  1967. usage_when_required = '''\
  1968. usage: PROG [-h] (--foo | --spam SPAM | badger [badger ...])
  1969. '''
  1970. help = '''\
  1971. positional arguments:
  1972. badger BADGER
  1973. optional arguments:
  1974. -h, --help show this help message and exit
  1975. --foo FOO
  1976. --spam SPAM SPAM
  1977. '''
  1978. class TestMutuallyExclusiveOptionalsMixed(MEMixin, TestCase):
  1979. def get_parser(self, required):
  1980. parser = ErrorRaisingArgumentParser(prog='PROG')
  1981. parser.add_argument('-x', action='store_true', help='x help')
  1982. group = parser.add_mutually_exclusive_group(required=required)
  1983. group.add_argument('-a', action='store_true', help='a help')
  1984. group.add_argument('-b', action='store_true', help='b help')
  1985. parser.add_argument('-y', action='store_true', help='y help')
  1986. group.add_argument('-c', action='store_true', help='c help')
  1987. return parser
  1988. failures = ['-a -b', '-b -c', '-a -c', '-a -b -c']
  1989. successes = [
  1990. ('-a', NS(a=True, b=False, c=False, x=False, y=False)),
  1991. ('-b', NS(a=False, b=True, c=False, x=False, y=False)),
  1992. ('-c', NS(a=False, b=False, c=True, x=False, y=False)),
  1993. ('-a -x', NS(a=True, b=False, c=False, x=True, y=False)),
  1994. ('-y -b', NS(a=False, b=True, c=False, x=False, y=True)),
  1995. ('-x -y -c', NS(a=False, b=False, c=True, x=True, y=True)),
  1996. ]
  1997. successes_when_not_required = [
  1998. ('', NS(a=False, b=False, c=False, x=False, y=False)),
  1999. ('-x', NS(a=False, b=False, c=False, x=True, y=False)),
  2000. ('-y', NS(a=False, b=False, c=False, x=False, y=True)),
  2001. ]
  2002. usage_when_required = usage_when_not_required = '''\
  2003. usage: PROG [-h] [-x] [-a] [-b] [-y] [-c]
  2004. '''
  2005. help = '''\
  2006. optional arguments:
  2007. -h, --help show this help message and exit
  2008. -x x help
  2009. -a a help
  2010. -b b help
  2011. -y y help
  2012. -c c help
  2013. '''
  2014. class TestMutuallyExclusiveInGroup(MEMixin, TestCase):
  2015. def get_parser(self, required=None):
  2016. parser = ErrorRaisingArgumentParser(prog='PROG')
  2017. titled_group = parser.add_argument_group(
  2018. title='Titled group', description='Group description')
  2019. mutex_group = \
  2020. titled_group.add_mutually_exclusive_group(required=required)
  2021. mutex_group.add_argument('--bar', help='bar help')
  2022. mutex_group.add_argument('--baz', help='baz help')
  2023. return parser
  2024. failures = ['--bar X --baz Y', '--baz X --bar Y']
  2025. successes = [
  2026. ('--bar X', NS(bar='X', baz=None)),
  2027. ('--baz Y', NS(bar=None, baz='Y')),
  2028. ]
  2029. successes_when_not_required = [
  2030. ('', NS(bar=None, baz=None)),
  2031. ]
  2032. usage_when_not_required = '''\
  2033. usage: PROG [-h] [--bar BAR | --baz BAZ]
  2034. '''
  2035. usage_when_required = '''\
  2036. usage: PROG [-h] (--bar BAR | --baz BAZ)
  2037. '''
  2038. help = '''\
  2039. optional arguments:
  2040. -h, --help show this help message and exit
  2041. Titled group:
  2042. Group description
  2043. --bar BAR bar help
  2044. --baz BAZ baz help
  2045. '''
  2046. class TestMutuallyExclusiveOptionalsAndPositionalsMixed(MEMixin, TestCase):
  2047. def get_parser(self, required):
  2048. parser = ErrorRaisingArgumentParser(prog='PROG')
  2049. parser.add_argument('x', help='x help')
  2050. parser.add_argument('-y', action='store_true', help='y help')
  2051. group = parser.add_mutually_exclusive_group(required=required)
  2052. group.add_argument('a', nargs='?', help='a help')
  2053. group.add_argument('-b', action='store_true', help='b help')
  2054. group.add_argument('-c', action='store_true', help='c help')
  2055. return parser
  2056. failures = ['X A -b', '-b -c', '-c X A']
  2057. successes = [
  2058. ('X A', NS(a='A', b=False, c=False, x='X', y=False)),
  2059. ('X -b', NS(a=None, b=True, c=False, x='X', y=False)),
  2060. ('X -c', NS(a=None, b=False, c=True, x='X', y=False)),
  2061. ('X A -y', NS(a='A', b=False, c=False, x='X', y=True)),
  2062. ('X -y -b', NS(a=None, b=True, c=False, x='X', y=True)),
  2063. ]
  2064. successes_when_not_required = [
  2065. ('X', NS(a=None, b=False, c=False, x='X', y=False)),
  2066. ('X -y', NS(a=None, b=False, c=False, x='X', y=True)),
  2067. ]
  2068. usage_when_required = usage_when_not_required = '''\
  2069. usage: PROG [-h] [-y] [-b] [-c] x [a]
  2070. '''
  2071. help = '''\
  2072. positional arguments:
  2073. x x help
  2074. a a help
  2075. optional arguments:
  2076. -h, --help show this help message and exit
  2077. -y y help
  2078. -b b help
  2079. -c c help
  2080. '''
  2081. # =================================================
  2082. # Mutually exclusive group in parent parser tests
  2083. # =================================================
  2084. class MEPBase(object):
  2085. def get_parser(self, required=None):
  2086. parent = super(MEPBase, self).get_parser(required=required)
  2087. parser = ErrorRaisingArgumentParser(
  2088. prog=parent.prog, add_help=False, parents=[parent])
  2089. return parser
  2090. class TestMutuallyExclusiveGroupErrorsParent(
  2091. MEPBase, TestMutuallyExclusiveGroupErrors):
  2092. pass
  2093. class TestMutuallyExclusiveSimpleParent(
  2094. MEPBase, TestMutuallyExclusiveSimple):
  2095. pass
  2096. class TestMutuallyExclusiveLongParent(
  2097. MEPBase, TestMutuallyExclusiveLong):
  2098. pass
  2099. class TestMutuallyExclusiveFirstSuppressedParent(
  2100. MEPBase, TestMutuallyExclusiveFirstSuppressed):
  2101. pass
  2102. class TestMutuallyExclusiveManySuppressedParent(
  2103. MEPBase, TestMutuallyExclusiveManySuppressed):
  2104. pass
  2105. class TestMutuallyExclusiveOptionalAndPositionalParent(
  2106. MEPBase, TestMutuallyExclusiveOptionalAndPositional):
  2107. pass
  2108. class TestMutuallyExclusiveOptionalsMixedParent(
  2109. MEPBase, TestMutuallyExclusiveOptionalsMixed):
  2110. pass
  2111. class TestMutuallyExclusiveOptionalsAndPositionalsMixedParent(
  2112. MEPBase, TestMutuallyExclusiveOptionalsAndPositionalsMixed):
  2113. pass
  2114. # =================
  2115. # Set default tests
  2116. # =================
  2117. class TestSetDefaults(TestCase):
  2118. def test_set_defaults_no_args(self):
  2119. parser = ErrorRaisingArgumentParser()
  2120. parser.set_defaults(x='foo')
  2121. parser.set_defaults(y='bar', z=1)
  2122. self.assertEqual(NS(x='foo', y='bar', z=1),
  2123. parser.parse_args([]))
  2124. self.assertEqual(NS(x='foo', y='bar', z=1),
  2125. parser.parse_args([], NS()))
  2126. self.assertEqual(NS(x='baz', y='bar', z=1),
  2127. parser.parse_args([], NS(x='baz')))
  2128. self.assertEqual(NS(x='baz', y='bar', z=2),
  2129. parser.parse_args([], NS(x='baz', z=2)))
  2130. def test_set_defaults_with_args(self):
  2131. parser = ErrorRaisingArgumentParser()
  2132. parser.set_defaults(x='foo', y='bar')
  2133. parser.add_argument('-x', default='xfoox')
  2134. self.assertEqual(NS(x='xfoox', y='bar'),
  2135. parser.parse_args([]))
  2136. self.assertEqual(NS(x='xfoox', y='bar'),
  2137. parser.parse_args([], NS()))
  2138. self.assertEqual(NS(x='baz', y='bar'),
  2139. parser.parse_args([], NS(x='baz')))
  2140. self.assertEqual(NS(x='1', y='bar'),
  2141. parser.parse_args('-x 1'.split()))
  2142. self.assertEqual(NS(x='1', y='bar'),
  2143. parser.parse_args('-x 1'.split(), NS()))
  2144. self.assertEqual(NS(x='1', y='bar'),
  2145. parser.parse_args('-x 1'.split(), NS(x='baz')))
  2146. def test_set_defaults_subparsers(self):
  2147. parser = ErrorRaisingArgumentParser()
  2148. parser.set_defaults(x='foo')
  2149. subparsers = parser.add_subparsers()
  2150. parser_a = subparsers.add_parser('a')
  2151. parser_a.set_defaults(y='bar')
  2152. self.assertEqual(NS(x='foo', y='bar'),
  2153. parser.parse_args('a'.split()))
  2154. def test_set_defaults_parents(self):
  2155. parent = ErrorRaisingArgumentParser(add_help=False)
  2156. parent.set_defaults(x='foo')
  2157. parser = ErrorRaisingArgumentParser(parents=[parent])
  2158. self.assertEqual(NS(x='foo'), parser.parse_args([]))
  2159. def test_set_defaults_same_as_add_argument(self):
  2160. parser = ErrorRaisingArgumentParser()
  2161. parser.set_defaults(w='W', x='X', y='Y', z='Z')
  2162. parser.add_argument('-w')
  2163. parser.add_argument('-x', default='XX')
  2164. parser.add_argument('y', nargs='?')
  2165. parser.add_argument('z', nargs='?', default='ZZ')
  2166. # defaults set previously
  2167. self.assertEqual(NS(w='W', x='XX', y='Y', z='ZZ'),
  2168. parser.parse_args([]))
  2169. # reset defaults
  2170. parser.set_defaults(w='WW', x='X', y='YY', z='Z')
  2171. self.assertEqual(NS(w='WW', x='X', y='YY', z='Z'),
  2172. parser.parse_args([]))
  2173. def test_set_defaults_same_as_add_argument_group(self):
  2174. parser = ErrorRaisingArgumentParser()
  2175. parser.set_defaults(w='W', x='X', y='Y', z='Z')
  2176. group = parser.add_argument_group('foo')
  2177. group.add_argument('-w')
  2178. group.add_argument('-x', default='XX')
  2179. group.add_argument('y', nargs='?')
  2180. group.add_argument('z', nargs='?', default='ZZ')
  2181. # defaults set previously
  2182. self.assertEqual(NS(w='W', x='XX', y='Y', z='ZZ'),
  2183. parser.parse_args([]))
  2184. # reset defaults
  2185. parser.set_defaults(w='WW', x='X', y='YY', z='Z')
  2186. self.assertEqual(NS(w='WW', x='X', y='YY', z='Z'),
  2187. parser.parse_args([]))
  2188. # =================
  2189. # Get default tests
  2190. # =================
  2191. class TestGetDefault(TestCase):
  2192. def test_get_default(self):
  2193. parser = ErrorRaisingArgumentParser()
  2194. self.assertEqual(None, parser.get_default("foo"))
  2195. self.assertEqual(None, parser.get_default("bar"))
  2196. parser.add_argument("--foo")
  2197. self.assertEqual(None, parser.get_default("foo"))
  2198. self.assertEqual(None, parser.get_default("bar"))
  2199. parser.add_argument("--bar", type=int, default=42)
  2200. self.assertEqual(None, parser.get_default("foo"))
  2201. self.assertEqual(42, parser.get_default("bar"))
  2202. parser.set_defaults(foo="badger")
  2203. self.assertEqual("badger", parser.get_default("foo"))
  2204. self.assertEqual(42, parser.get_default("bar"))
  2205. # ==========================
  2206. # Namespace 'contains' tests
  2207. # ==========================
  2208. class TestNamespaceContainsSimple(TestCase):
  2209. def test_empty(self):
  2210. ns = argparse.Namespace()
  2211. self.assertEqual('' in ns, False)
  2212. self.assertEqual('' not in ns, True)
  2213. self.assertEqual('x' in ns, False)
  2214. def test_non_empty(self):
  2215. ns = argparse.Namespace(x=1, y=2)
  2216. self.assertEqual('x' in ns, True)
  2217. self.assertEqual('x' not in ns, False)
  2218. self.assertEqual('y' in ns, True)
  2219. self.assertEqual('' in ns, False)
  2220. self.assertEqual('xx' in ns, False)
  2221. self.assertEqual('z' in ns, False)
  2222. # =====================
  2223. # Help formatting tests
  2224. # =====================
  2225. class TestHelpFormattingMetaclass(type):
  2226. def __init__(cls, name, bases, bodydict):
  2227. if name == 'HelpTestCase':
  2228. return
  2229. class AddTests(object):
  2230. def __init__(self, test_class, func_suffix, std_name):
  2231. self.func_suffix = func_suffix
  2232. self.std_name = std_name
  2233. for test_func in [self.test_format,
  2234. self.test_print,
  2235. self.test_print_file]:
  2236. test_name = '%s_%s' % (test_func.__name__, func_suffix)
  2237. def test_wrapper(self, test_func=test_func):
  2238. test_func(self)
  2239. try:
  2240. test_wrapper.__name__ = test_name
  2241. except TypeError:
  2242. pass
  2243. setattr(test_class, test_name, test_wrapper)
  2244. def _get_parser(self, tester):
  2245. parser = argparse.ArgumentParser(
  2246. *tester.parser_signature.args,
  2247. **tester.parser_signature.kwargs)
  2248. for argument_sig in getattr(tester, 'argument_signatures', []):
  2249. parser.add_argument(*argument_sig.args,
  2250. **argument_sig.kwargs)
  2251. group_sigs = getattr(tester, 'argument_group_signatures', [])
  2252. for group_sig, argument_sigs in group_sigs:
  2253. group = parser.add_argument_group(*group_sig.args,
  2254. **group_sig.kwargs)
  2255. for argument_sig in argument_sigs:
  2256. group.add_argument(*argument_sig.args,
  2257. **argument_sig.kwargs)
  2258. subparsers_sigs = getattr(tester, 'subparsers_signatures', [])
  2259. if subparsers_sigs:
  2260. subparsers = parser.add_subparsers()
  2261. for subparser_sig in subparsers_sigs:
  2262. subparsers.add_parser(*subparser_sig.args,
  2263. **subparser_sig.kwargs)
  2264. return parser
  2265. def _test(self, tester, parser_text):
  2266. expected_text = getattr(tester, self.func_suffix)
  2267. expected_text = textwrap.dedent(expected_text)
  2268. if expected_text != parser_text:
  2269. print(repr(expected_text))
  2270. print(repr(parser_text))
  2271. for char1, char2 in zip(expected_text, parser_text):
  2272. if char1 != char2:
  2273. print('first diff: %r %r' % (char1, char2))
  2274. break
  2275. tester.assertEqual(expected_text, parser_text)
  2276. def test_format(self, tester):
  2277. parser = self._get_parser(tester)
  2278. format = getattr(parser, 'format_%s' % self.func_suffix)
  2279. self._test(tester, format())
  2280. def test_print(self, tester):
  2281. parser = self._get_parser(tester)
  2282. print_ = getattr(parser, 'print_%s' % self.func_suffix)
  2283. old_stream = getattr(sys, self.std_name)
  2284. setattr(sys, self.std_name, StdIOBuffer())
  2285. try:
  2286. print_()
  2287. parser_text = getattr(sys, self.std_name).getvalue()
  2288. finally:
  2289. setattr(sys, self.std_name, old_stream)
  2290. self._test(tester, parser_text)
  2291. def test_print_file(self, tester):
  2292. parser = self._get_parser(tester)
  2293. print_ = getattr(parser, 'print_%s' % self.func_suffix)
  2294. sfile = StdIOBuffer()
  2295. print_(sfile)
  2296. parser_text = sfile.getvalue()
  2297. self._test(tester, parser_text)
  2298. # add tests for {format,print}_{usage,help,version}
  2299. for func_suffix, std_name in [('usage', 'stdout'),
  2300. ('help', 'stdout'),
  2301. ('version', 'stderr')]:
  2302. AddTests(cls, func_suffix, std_name)
  2303. bases = TestCase,
  2304. HelpTestCase = TestHelpFormattingMetaclass('HelpTestCase', bases, {})
  2305. class TestHelpBiggerOptionals(HelpTestCase):
  2306. """Make sure that argument help aligns when options are longer"""
  2307. parser_signature = Sig(prog='PROG', description='DESCRIPTION',
  2308. epilog='EPILOG', version='0.1')
  2309. argument_signatures = [
  2310. Sig('-x', action='store_true', help='X HELP'),
  2311. Sig('--y', help='Y HELP'),
  2312. Sig('foo', help='FOO HELP'),
  2313. Sig('bar', help='BAR HELP'),
  2314. ]
  2315. argument_group_signatures = []
  2316. usage = '''\
  2317. usage: PROG [-h] [-v] [-x] [--y Y] foo bar
  2318. '''
  2319. help = usage + '''\
  2320. DESCRIPTION
  2321. positional arguments:
  2322. foo FOO HELP
  2323. bar BAR HELP
  2324. optional arguments:
  2325. -h, --help show this help message and exit
  2326. -v, --version show program's version number and exit
  2327. -x X HELP
  2328. --y Y Y HELP
  2329. EPILOG
  2330. '''
  2331. version = '''\
  2332. 0.1
  2333. '''
  2334. class TestHelpBiggerOptionalGroups(HelpTestCase):
  2335. """Make sure that argument help aligns when options are longer"""
  2336. parser_signature = Sig(prog='PROG', description='DESCRIPTION',
  2337. epilog='EPILOG', version='0.1')
  2338. argument_signatures = [
  2339. Sig('-x', action='store_true', help='X HELP'),
  2340. Sig('--y', help='Y HELP'),
  2341. Sig('foo', help='FOO HELP'),
  2342. Sig('bar', help='BAR HELP'),
  2343. ]
  2344. argument_group_signatures = [
  2345. (Sig('GROUP TITLE', description='GROUP DESCRIPTION'), [
  2346. Sig('baz', help='BAZ HELP'),
  2347. Sig('-z', nargs='+', help='Z HELP')]),
  2348. ]
  2349. usage = '''\
  2350. usage: PROG [-h] [-v] [-x] [--y Y] [-z Z [Z ...]] foo bar baz
  2351. '''
  2352. help = usage + '''\
  2353. DESCRIPTION
  2354. positional arguments:
  2355. foo FOO HELP
  2356. bar BAR HELP
  2357. optional arguments:
  2358. -h, --help show this help message and exit
  2359. -v, --version show program's version number and exit
  2360. -x X HELP
  2361. --y Y Y HELP
  2362. GROUP TITLE:
  2363. GROUP DESCRIPTION
  2364. baz BAZ HELP
  2365. -z Z [Z ...] Z HELP
  2366. EPILOG
  2367. '''
  2368. version = '''\
  2369. 0.1
  2370. '''
  2371. class TestHelpBiggerPositionals(HelpTestCase):
  2372. """Make sure that help aligns when arguments are longer"""
  2373. parser_signature = Sig(usage='USAGE', description='DESCRIPTION')
  2374. argument_signatures = [
  2375. Sig('-x', action='store_true', help='X HELP'),
  2376. Sig('--y', help='Y HELP'),
  2377. Sig('ekiekiekifekang', help='EKI HELP'),
  2378. Sig('bar', help='BAR HELP'),
  2379. ]
  2380. argument_group_signatures = []
  2381. usage = '''\
  2382. usage: USAGE
  2383. '''
  2384. help = usage + '''\
  2385. DESCRIPTION
  2386. positional arguments:
  2387. ekiekiekifekang EKI HELP
  2388. bar BAR HELP
  2389. optional arguments:
  2390. -h, --help show this help message and exit
  2391. -x X HELP
  2392. --y Y Y HELP
  2393. '''
  2394. version = ''
  2395. class TestHelpReformatting(HelpTestCase):
  2396. """Make sure that text after short names starts on the first line"""
  2397. parser_signature = Sig(
  2398. prog='PROG',
  2399. description=' oddly formatted\n'
  2400. 'description\n'
  2401. '\n'
  2402. 'that is so long that it should go onto multiple '
  2403. 'lines when wrapped')
  2404. argument_signatures = [
  2405. Sig('-x', metavar='XX', help='oddly\n'
  2406. ' formatted -x help'),
  2407. Sig('y', metavar='yyy', help='normal y help'),
  2408. ]
  2409. argument_group_signatures = [
  2410. (Sig('title', description='\n'
  2411. ' oddly formatted group\n'
  2412. '\n'
  2413. 'description'),
  2414. [Sig('-a', action='store_true',
  2415. help=' oddly \n'
  2416. 'formatted -a help \n'
  2417. ' again, so long that it should be wrapped over '
  2418. 'multiple lines')]),
  2419. ]
  2420. usage = '''\
  2421. usage: PROG [-h] [-x XX] [-a] yyy
  2422. '''
  2423. help = usage + '''\
  2424. oddly formatted description that is so long that it should go onto \
  2425. multiple
  2426. lines when wrapped
  2427. positional arguments:
  2428. yyy normal y help
  2429. optional arguments:
  2430. -h, --help show this help message and exit
  2431. -x XX oddly formatted -x help
  2432. title:
  2433. oddly formatted group description
  2434. -a oddly formatted -a help again, so long that it should \
  2435. be wrapped
  2436. over multiple lines
  2437. '''
  2438. version = ''
  2439. class TestHelpWrappingShortNames(HelpTestCase):
  2440. """Make sure that text after short names starts on the first line"""
  2441. parser_signature = Sig(prog='PROG', description= 'D\nD' * 30)
  2442. argument_signatures = [
  2443. Sig('-x', metavar='XX', help='XHH HX' * 20),
  2444. Sig('y', metavar='yyy', help='YH YH' * 20),
  2445. ]
  2446. argument_group_signatures = [
  2447. (Sig('ALPHAS'), [
  2448. Sig('-a', action='store_true', help='AHHH HHA' * 10)]),
  2449. ]
  2450. usage = '''\
  2451. usage: PROG [-h] [-x XX] [-a] yyy
  2452. '''
  2453. help = usage + '''\
  2454. D DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD \
  2455. DD DD DD
  2456. DD DD DD DD D
  2457. positional arguments:
  2458. yyy YH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH \
  2459. YHYH YHYH
  2460. YHYH YHYH YHYH YHYH YHYH YHYH YHYH YH
  2461. optional arguments:
  2462. -h, --help show this help message and exit
  2463. -x XX XHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH \
  2464. HXXHH HXXHH
  2465. HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HX
  2466. ALPHAS:
  2467. -a AHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH \
  2468. HHAAHHH
  2469. HHAAHHH HHAAHHH HHA
  2470. '''
  2471. version = ''
  2472. class TestHelpWrappingLongNames(HelpTestCase):
  2473. """Make sure that text after long names starts on the next line"""
  2474. parser_signature = Sig(usage='USAGE', description= 'D D' * 30,
  2475. version='V V'*30)
  2476. argument_signatures = [
  2477. Sig('-x', metavar='X' * 25, help='XH XH' * 20),
  2478. Sig('y', metavar='y' * 25, help='YH YH' * 20),
  2479. ]
  2480. argument_group_signatures = [
  2481. (Sig('ALPHAS'), [
  2482. Sig('-a', metavar='A' * 25, help='AH AH' * 20),
  2483. Sig('z', metavar='z' * 25, help='ZH ZH' * 20)]),
  2484. ]
  2485. usage = '''\
  2486. usage: USAGE
  2487. '''
  2488. help = usage + '''\
  2489. D DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD \
  2490. DD DD DD
  2491. DD DD DD DD D
  2492. positional arguments:
  2493. yyyyyyyyyyyyyyyyyyyyyyyyy
  2494. YH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH \
  2495. YHYH YHYH
  2496. YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YH
  2497. optional arguments:
  2498. -h, --help show this help message and exit
  2499. -v, --version show program's version number and exit
  2500. -x XXXXXXXXXXXXXXXXXXXXXXXXX
  2501. XH XHXH XHXH XHXH XHXH XHXH XHXH XHXH XHXH \
  2502. XHXH XHXH
  2503. XHXH XHXH XHXH XHXH XHXH XHXH XHXH XHXH XHXH XH
  2504. ALPHAS:
  2505. -a AAAAAAAAAAAAAAAAAAAAAAAAA
  2506. AH AHAH AHAH AHAH AHAH AHAH AHAH AHAH AHAH \
  2507. AHAH AHAH
  2508. AHAH AHAH AHAH AHAH AHAH AHAH AHAH AHAH AHAH AH
  2509. zzzzzzzzzzzzzzzzzzzzzzzzz
  2510. ZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH \
  2511. ZHZH ZHZH
  2512. ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZH
  2513. '''
  2514. version = '''\
  2515. V VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV \
  2516. VV VV VV
  2517. VV VV VV VV V
  2518. '''
  2519. class TestHelpUsage(HelpTestCase):
  2520. """Test basic usage messages"""
  2521. parser_signature = Sig(prog='PROG')
  2522. argument_signatures = [
  2523. Sig('-w', nargs='+', help='w'),
  2524. Sig('-x', nargs='*', help='x'),
  2525. Sig('a', help='a'),
  2526. Sig('b', help='b', nargs=2),
  2527. Sig('c', help='c', nargs='?'),
  2528. ]
  2529. argument_group_signatures = [
  2530. (Sig('group'), [
  2531. Sig('-y', nargs='?', help='y'),
  2532. Sig('-z', nargs=3, help='z'),
  2533. Sig('d', help='d', nargs='*'),
  2534. Sig('e', help='e', nargs='+'),
  2535. ])
  2536. ]
  2537. usage = '''\
  2538. usage: PROG [-h] [-w W [W ...]] [-x [X [X ...]]] [-y [Y]] [-z Z Z Z]
  2539. a b b [c] [d [d ...]] e [e ...]
  2540. '''
  2541. help = usage + '''\
  2542. positional arguments:
  2543. a a
  2544. b b
  2545. c c
  2546. optional arguments:
  2547. -h, --help show this help message and exit
  2548. -w W [W ...] w
  2549. -x [X [X ...]] x
  2550. group:
  2551. -y [Y] y
  2552. -z Z Z Z z
  2553. d d
  2554. e e
  2555. '''
  2556. version = ''
  2557. class TestHelpOnlyUserGroups(HelpTestCase):
  2558. """Test basic usage messages"""
  2559. parser_signature = Sig(prog='PROG', add_help=False)
  2560. argument_signatures = []
  2561. argument_group_signatures = [
  2562. (Sig('xxxx'), [
  2563. Sig('-x', help='x'),
  2564. Sig('a', help='a'),
  2565. ]),
  2566. (Sig('yyyy'), [
  2567. Sig('b', help='b'),
  2568. Sig('-y', help='y'),
  2569. ]),
  2570. ]
  2571. usage = '''\
  2572. usage: PROG [-x X] [-y Y] a b
  2573. '''
  2574. help = usage + '''\
  2575. xxxx:
  2576. -x X x
  2577. a a
  2578. yyyy:
  2579. b b
  2580. -y Y y
  2581. '''
  2582. version = ''
  2583. class TestHelpUsageLongProg(HelpTestCase):
  2584. """Test usage messages where the prog is long"""
  2585. parser_signature = Sig(prog='P' * 60)
  2586. argument_signatures = [
  2587. Sig('-w', metavar='W'),
  2588. Sig('-x', metavar='X'),
  2589. Sig('a'),
  2590. Sig('b'),
  2591. ]
  2592. argument_group_signatures = []
  2593. usage = '''\
  2594. usage: PPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPP
  2595. [-h] [-w W] [-x X] a b
  2596. '''
  2597. help = usage + '''\
  2598. positional arguments:
  2599. a
  2600. b
  2601. optional arguments:
  2602. -h, --help show this help message and exit
  2603. -w W
  2604. -x X
  2605. '''
  2606. version = ''
  2607. class TestHelpUsageLongProgOptionsWrap(HelpTestCase):
  2608. """Test usage messages where the prog is long and the optionals wrap"""
  2609. parser_signature = Sig(prog='P' * 60)
  2610. argument_signatures = [
  2611. Sig('-w', metavar='W' * 25),
  2612. Sig('-x', metavar='X' * 25),
  2613. Sig('-y', metavar='Y' * 25),
  2614. Sig('-z', metavar='Z' * 25),
  2615. Sig('a'),
  2616. Sig('b'),
  2617. ]
  2618. argument_group_signatures = []
  2619. usage = '''\
  2620. usage: PPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPP
  2621. [-h] [-w WWWWWWWWWWWWWWWWWWWWWWWWW] \
  2622. [-x XXXXXXXXXXXXXXXXXXXXXXXXX]
  2623. [-y YYYYYYYYYYYYYYYYYYYYYYYYY] [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
  2624. a b
  2625. '''
  2626. help = usage + '''\
  2627. positional arguments:
  2628. a
  2629. b
  2630. optional arguments:
  2631. -h, --help show this help message and exit
  2632. -w WWWWWWWWWWWWWWWWWWWWWWWWW
  2633. -x XXXXXXXXXXXXXXXXXXXXXXXXX
  2634. -y YYYYYYYYYYYYYYYYYYYYYYYYY
  2635. -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
  2636. '''
  2637. version = ''
  2638. class TestHelpUsageLongProgPositionalsWrap(HelpTestCase):
  2639. """Test usage messages where the prog is long and the positionals wrap"""
  2640. parser_signature = Sig(prog='P' * 60, add_help=False)
  2641. argument_signatures = [
  2642. Sig('a' * 25),
  2643. Sig('b' * 25),
  2644. Sig('c' * 25),
  2645. ]
  2646. argument_group_signatures = []
  2647. usage = '''\
  2648. usage: PPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPP
  2649. aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
  2650. ccccccccccccccccccccccccc
  2651. '''
  2652. help = usage + '''\
  2653. positional arguments:
  2654. aaaaaaaaaaaaaaaaaaaaaaaaa
  2655. bbbbbbbbbbbbbbbbbbbbbbbbb
  2656. ccccccccccccccccccccccccc
  2657. '''
  2658. version = ''
  2659. class TestHelpUsageOptionalsWrap(HelpTestCase):
  2660. """Test usage messages where the optionals wrap"""
  2661. parser_signature = Sig(prog='PROG')
  2662. argument_signatures = [
  2663. Sig('-w', metavar='W' * 25),
  2664. Sig('-x', metavar='X' * 25),
  2665. Sig('-y', metavar='Y' * 25),
  2666. Sig('-z', metavar='Z' * 25),
  2667. Sig('a'),
  2668. Sig('b'),
  2669. Sig('c'),
  2670. ]
  2671. argument_group_signatures = []
  2672. usage = '''\
  2673. usage: PROG [-h] [-w WWWWWWWWWWWWWWWWWWWWWWWWW] \
  2674. [-x XXXXXXXXXXXXXXXXXXXXXXXXX]
  2675. [-y YYYYYYYYYYYYYYYYYYYYYYYYY] \
  2676. [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
  2677. a b c
  2678. '''
  2679. help = usage + '''\
  2680. positional arguments:
  2681. a
  2682. b
  2683. c
  2684. optional arguments:
  2685. -h, --help show this help message and exit
  2686. -w WWWWWWWWWWWWWWWWWWWWWWWWW
  2687. -x XXXXXXXXXXXXXXXXXXXXXXXXX
  2688. -y YYYYYYYYYYYYYYYYYYYYYYYYY
  2689. -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
  2690. '''
  2691. version = ''
  2692. class TestHelpUsagePositionalsWrap(HelpTestCase):
  2693. """Test usage messages where the positionals wrap"""
  2694. parser_signature = Sig(prog='PROG')
  2695. argument_signatures = [
  2696. Sig('-x'),
  2697. Sig('-y'),
  2698. Sig('-z'),
  2699. Sig('a' * 25),
  2700. Sig('b' * 25),
  2701. Sig('c' * 25),
  2702. ]
  2703. argument_group_signatures = []
  2704. usage = '''\
  2705. usage: PROG [-h] [-x X] [-y Y] [-z Z]
  2706. aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
  2707. ccccccccccccccccccccccccc
  2708. '''
  2709. help = usage + '''\
  2710. positional arguments:
  2711. aaaaaaaaaaaaaaaaaaaaaaaaa
  2712. bbbbbbbbbbbbbbbbbbbbbbbbb
  2713. ccccccccccccccccccccccccc
  2714. optional arguments:
  2715. -h, --help show this help message and exit
  2716. -x X
  2717. -y Y
  2718. -z Z
  2719. '''
  2720. version = ''
  2721. class TestHelpUsageOptionalsPositionalsWrap(HelpTestCase):
  2722. """Test usage messages where the optionals and positionals wrap"""
  2723. parser_signature = Sig(prog='PROG')
  2724. argument_signatures = [
  2725. Sig('-x', metavar='X' * 25),
  2726. Sig('-y', metavar='Y' * 25),
  2727. Sig('-z', metavar='Z' * 25),
  2728. Sig('a' * 25),
  2729. Sig('b' * 25),
  2730. Sig('c' * 25),
  2731. ]
  2732. argument_group_signatures = []
  2733. usage = '''\
  2734. usage: PROG [-h] [-x XXXXXXXXXXXXXXXXXXXXXXXXX] \
  2735. [-y YYYYYYYYYYYYYYYYYYYYYYYYY]
  2736. [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
  2737. aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
  2738. ccccccccccccccccccccccccc
  2739. '''
  2740. help = usage + '''\
  2741. positional arguments:
  2742. aaaaaaaaaaaaaaaaaaaaaaaaa
  2743. bbbbbbbbbbbbbbbbbbbbbbbbb
  2744. ccccccccccccccccccccccccc
  2745. optional arguments:
  2746. -h, --help show this help message and exit
  2747. -x XXXXXXXXXXXXXXXXXXXXXXXXX
  2748. -y YYYYYYYYYYYYYYYYYYYYYYYYY
  2749. -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
  2750. '''
  2751. version = ''
  2752. class TestHelpUsageOptionalsOnlyWrap(HelpTestCase):
  2753. """Test usage messages where there are only optionals and they wrap"""
  2754. parser_signature = Sig(prog='PROG')
  2755. argument_signatures = [
  2756. Sig('-x', metavar='X' * 25),
  2757. Sig('-y', metavar='Y' * 25),
  2758. Sig('-z', metavar='Z' * 25),
  2759. ]
  2760. argument_group_signatures = []
  2761. usage = '''\
  2762. usage: PROG [-h] [-x XXXXXXXXXXXXXXXXXXXXXXXXX] \
  2763. [-y YYYYYYYYYYYYYYYYYYYYYYYYY]
  2764. [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
  2765. '''
  2766. help = usage + '''\
  2767. optional arguments:
  2768. -h, --help show this help message and exit
  2769. -x XXXXXXXXXXXXXXXXXXXXXXXXX
  2770. -y YYYYYYYYYYYYYYYYYYYYYYYYY
  2771. -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
  2772. '''
  2773. version = ''
  2774. class TestHelpUsagePositionalsOnlyWrap(HelpTestCase):
  2775. """Test usage messages where there are only positionals and they wrap"""
  2776. parser_signature = Sig(prog='PROG', add_help=False)
  2777. argument_signatures = [
  2778. Sig('a' * 25),
  2779. Sig('b' * 25),
  2780. Sig('c' * 25),
  2781. ]
  2782. argument_group_signatures = []
  2783. usage = '''\
  2784. usage: PROG aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
  2785. ccccccccccccccccccccccccc
  2786. '''
  2787. help = usage + '''\
  2788. positional arguments:
  2789. aaaaaaaaaaaaaaaaaaaaaaaaa
  2790. bbbbbbbbbbbbbbbbbbbbbbbbb
  2791. ccccccccccccccccccccccccc
  2792. '''
  2793. version = ''
  2794. class TestHelpVariableExpansion(HelpTestCase):
  2795. """Test that variables are expanded properly in help messages"""
  2796. parser_signature = Sig(prog='PROG')
  2797. argument_signatures = [
  2798. Sig('-x', type=int,
  2799. help='x %(prog)s %(default)s %(type)s %%'),
  2800. Sig('-y', action='store_const', default=42, const='XXX',
  2801. help='y %(prog)s %(default)s %(const)s'),
  2802. Sig('--foo', choices='abc',
  2803. help='foo %(prog)s %(default)s %(choices)s'),
  2804. Sig('--bar', default='baz', choices=[1, 2], metavar='BBB',
  2805. help='bar %(prog)s %(default)s %(dest)s'),
  2806. Sig('spam', help='spam %(prog)s %(default)s'),
  2807. Sig('badger', default=0.5, help='badger %(prog)s %(default)s'),
  2808. ]
  2809. argument_group_signatures = [
  2810. (Sig('group'), [
  2811. Sig('-a', help='a %(prog)s %(default)s'),
  2812. Sig('-b', default=-1, help='b %(prog)s %(default)s'),
  2813. ])
  2814. ]
  2815. usage = ('''\
  2816. usage: PROG [-h] [-x X] [-y] [--foo {a,b,c}] [--bar BBB] [-a A] [-b B]
  2817. spam badger
  2818. ''')
  2819. help = usage + '''\
  2820. positional arguments:
  2821. spam spam PROG None
  2822. badger badger PROG 0.5
  2823. optional arguments:
  2824. -h, --help show this help message and exit
  2825. -x X x PROG None int %
  2826. -y y PROG 42 XXX
  2827. --foo {a,b,c} foo PROG None a, b, c
  2828. --bar BBB bar PROG baz bar
  2829. group:
  2830. -a A a PROG None
  2831. -b B b PROG -1
  2832. '''
  2833. version = ''
  2834. class TestHelpVariableExpansionUsageSupplied(HelpTestCase):
  2835. """Test that variables are expanded properly when usage= is present"""
  2836. parser_signature = Sig(prog='PROG', usage='%(prog)s FOO')
  2837. argument_signatures = []
  2838. argument_group_signatures = []
  2839. usage = ('''\
  2840. usage: PROG FOO
  2841. ''')
  2842. help = usage + '''\
  2843. optional arguments:
  2844. -h, --help show this help message and exit
  2845. '''
  2846. version = ''
  2847. class TestHelpVariableExpansionNoArguments(HelpTestCase):
  2848. """Test that variables are expanded properly with no arguments"""
  2849. parser_signature = Sig(prog='PROG', add_help=False)
  2850. argument_signatures = []
  2851. argument_group_signatures = []
  2852. usage = ('''\
  2853. usage: PROG
  2854. ''')
  2855. help = usage
  2856. version = ''
  2857. class TestHelpSuppressUsage(HelpTestCase):
  2858. """Test that items can be suppressed in usage messages"""
  2859. parser_signature = Sig(prog='PROG', usage=argparse.SUPPRESS)
  2860. argument_signatures = [
  2861. Sig('--foo', help='foo help'),
  2862. Sig('spam', help='spam help'),
  2863. ]
  2864. argument_group_signatures = []
  2865. help = '''\
  2866. positional arguments:
  2867. spam spam help
  2868. optional arguments:
  2869. -h, --help show this help message and exit
  2870. --foo FOO foo help
  2871. '''
  2872. usage = ''
  2873. version = ''
  2874. class TestHelpSuppressOptional(HelpTestCase):
  2875. """Test that optional arguments can be suppressed in help messages"""
  2876. parser_signature = Sig(prog='PROG', add_help=False)
  2877. argument_signatures = [
  2878. Sig('--foo', help=argparse.SUPPRESS),
  2879. Sig('spam', help='spam help'),
  2880. ]
  2881. argument_group_signatures = []
  2882. usage = '''\
  2883. usage: PROG spam
  2884. '''
  2885. help = usage + '''\
  2886. positional arguments:
  2887. spam spam help
  2888. '''
  2889. version = ''
  2890. class TestHelpSuppressOptionalGroup(HelpTestCase):
  2891. """Test that optional groups can be suppressed in help messages"""
  2892. parser_signature = Sig(prog='PROG')
  2893. argument_signatures = [
  2894. Sig('--foo', help='foo help'),
  2895. Sig('spam', help='spam help'),
  2896. ]
  2897. argument_group_signatures = [
  2898. (Sig('group'), [Sig('--bar', help=argparse.SUPPRESS)]),
  2899. ]
  2900. usage = '''\
  2901. usage: PROG [-h] [--foo FOO] spam
  2902. '''
  2903. help = usage + '''\
  2904. positional arguments:
  2905. spam spam help
  2906. optional arguments:
  2907. -h, --help show this help message and exit
  2908. --foo FOO foo help
  2909. '''
  2910. version = ''
  2911. class TestHelpSuppressPositional(HelpTestCase):
  2912. """Test that positional arguments can be suppressed in help messages"""
  2913. parser_signature = Sig(prog='PROG')
  2914. argument_signatures = [
  2915. Sig('--foo', help='foo help'),
  2916. Sig('spam', help=argparse.SUPPRESS),
  2917. ]
  2918. argument_group_signatures = []
  2919. usage = '''\
  2920. usage: PROG [-h] [--foo FOO]
  2921. '''
  2922. help = usage + '''\
  2923. optional arguments:
  2924. -h, --help show this help message and exit
  2925. --foo FOO foo help
  2926. '''
  2927. version = ''
  2928. class TestHelpRequiredOptional(HelpTestCase):
  2929. """Test that required options don't look optional"""
  2930. parser_signature = Sig(prog='PROG')
  2931. argument_signatures = [
  2932. Sig('--foo', required=True, help='foo help'),
  2933. ]
  2934. argument_group_signatures = []
  2935. usage = '''\
  2936. usage: PROG [-h] --foo FOO
  2937. '''
  2938. help = usage + '''\
  2939. optional arguments:
  2940. -h, --help show this help message and exit
  2941. --foo FOO foo help
  2942. '''
  2943. version = ''
  2944. class TestHelpAlternatePrefixChars(HelpTestCase):
  2945. """Test that options display with different prefix characters"""
  2946. parser_signature = Sig(prog='PROG', prefix_chars='^;', add_help=False)
  2947. argument_signatures = [
  2948. Sig('^^foo', action='store_true', help='foo help'),
  2949. Sig(';b', ';;bar', help='bar help'),
  2950. ]
  2951. argument_group_signatures = []
  2952. usage = '''\
  2953. usage: PROG [^^foo] [;b BAR]
  2954. '''
  2955. help = usage + '''\
  2956. optional arguments:
  2957. ^^foo foo help
  2958. ;b BAR, ;;bar BAR bar help
  2959. '''
  2960. version = ''
  2961. class TestHelpNoHelpOptional(HelpTestCase):
  2962. """Test that the --help argument can be suppressed help messages"""
  2963. parser_signature = Sig(prog='PROG', add_help=False)
  2964. argument_signatures = [
  2965. Sig('--foo', help='foo help'),
  2966. Sig('spam', help='spam help'),
  2967. ]
  2968. argument_group_signatures = []
  2969. usage = '''\
  2970. usage: PROG [--foo FOO] spam
  2971. '''
  2972. help = usage + '''\
  2973. positional arguments:
  2974. spam spam help
  2975. optional arguments:
  2976. --foo FOO foo help
  2977. '''
  2978. version = ''
  2979. class TestHelpVersionOptional(HelpTestCase):
  2980. """Test that the --version argument can be suppressed help messages"""
  2981. parser_signature = Sig(prog='PROG', version='1.0')
  2982. argument_signatures = [
  2983. Sig('--foo', help='foo help'),
  2984. Sig('spam', help='spam help'),
  2985. ]
  2986. argument_group_signatures = []
  2987. usage = '''\
  2988. usage: PROG [-h] [-v] [--foo FOO] spam
  2989. '''
  2990. help = usage + '''\
  2991. positional arguments:
  2992. spam spam help
  2993. optional arguments:
  2994. -h, --help show this help message and exit
  2995. -v, --version show program's version number and exit
  2996. --foo FOO foo help
  2997. '''
  2998. version = '''\
  2999. 1.0
  3000. '''
  3001. class TestHelpNone(HelpTestCase):
  3002. """Test that no errors occur if no help is specified"""
  3003. parser_signature = Sig(prog='PROG')
  3004. argument_signatures = [
  3005. Sig('--foo'),
  3006. Sig('spam'),
  3007. ]
  3008. argument_group_signatures = []
  3009. usage = '''\
  3010. usage: PROG [-h] [--foo FOO] spam
  3011. '''
  3012. help = usage + '''\
  3013. positional arguments:
  3014. spam
  3015. optional arguments:
  3016. -h, --help show this help message and exit
  3017. --foo FOO
  3018. '''
  3019. version = ''
  3020. class TestHelpTupleMetavar(HelpTestCase):
  3021. """Test specifying metavar as a tuple"""
  3022. parser_signature = Sig(prog='PROG')
  3023. argument_signatures = [
  3024. Sig('-w', help='w', nargs='+', metavar=('W1', 'W2')),
  3025. Sig('-x', help='x', nargs='*', metavar=('X1', 'X2')),
  3026. Sig('-y', help='y', nargs=3, metavar=('Y1', 'Y2', 'Y3')),
  3027. Sig('-z', help='z', nargs='?', metavar=('Z1', )),
  3028. ]
  3029. argument_group_signatures = []
  3030. usage = '''\
  3031. usage: PROG [-h] [-w W1 [W2 ...]] [-x [X1 [X2 ...]]] [-y Y1 Y2 Y3] \
  3032. [-z [Z1]]
  3033. '''
  3034. help = usage + '''\
  3035. optional arguments:
  3036. -h, --help show this help message and exit
  3037. -w W1 [W2 ...] w
  3038. -x [X1 [X2 ...]] x
  3039. -y Y1 Y2 Y3 y
  3040. -z [Z1] z
  3041. '''
  3042. version = ''
  3043. class TestHelpRawText(HelpTestCase):
  3044. """Test the RawTextHelpFormatter"""
  3045. parser_signature = Sig(
  3046. prog='PROG', formatter_class=argparse.RawTextHelpFormatter,
  3047. description='Keep the formatting\n'
  3048. ' exactly as it is written\n'
  3049. '\n'
  3050. 'here\n')
  3051. argument_signatures = [
  3052. Sig('--foo', help=' foo help should also\n'
  3053. 'appear as given here'),
  3054. Sig('spam', help='spam help'),
  3055. ]
  3056. argument_group_signatures = [
  3057. (Sig('title', description=' This text\n'
  3058. ' should be indented\n'
  3059. ' exactly like it is here\n'),
  3060. [Sig('--bar', help='bar help')]),
  3061. ]
  3062. usage = '''\
  3063. usage: PROG [-h] [--foo FOO] [--bar BAR] spam
  3064. '''
  3065. help = usage + '''\
  3066. Keep the formatting
  3067. exactly as it is written
  3068. here
  3069. positional arguments:
  3070. spam spam help
  3071. optional arguments:
  3072. -h, --help show this help message and exit
  3073. --foo FOO foo help should also
  3074. appear as given here
  3075. title:
  3076. This text
  3077. should be indented
  3078. exactly like it is here
  3079. --bar BAR bar help
  3080. '''
  3081. version = ''
  3082. class TestHelpRawDescription(HelpTestCase):
  3083. """Test the RawTextHelpFormatter"""
  3084. parser_signature = Sig(
  3085. prog='PROG', formatter_class=argparse.RawDescriptionHelpFormatter,
  3086. description='Keep the formatting\n'
  3087. ' exactly as it is written\n'
  3088. '\n'
  3089. 'here\n')
  3090. argument_signatures = [
  3091. Sig('--foo', help=' foo help should not\n'
  3092. ' retain this odd formatting'),
  3093. Sig('spam', help='spam help'),
  3094. ]
  3095. argument_group_signatures = [
  3096. (Sig('title', description=' This text\n'
  3097. ' should be indented\n'
  3098. ' exactly like it is here\n'),
  3099. [Sig('--bar', help='bar help')]),
  3100. ]
  3101. usage = '''\
  3102. usage: PROG [-h] [--foo FOO] [--bar BAR] spam
  3103. '''
  3104. help = usage + '''\
  3105. Keep the formatting
  3106. exactly as it is written
  3107. here
  3108. positional arguments:
  3109. spam spam help
  3110. optional arguments:
  3111. -h, --help show this help message and exit
  3112. --foo FOO foo help should not retain this odd formatting
  3113. title:
  3114. This text
  3115. should be indented
  3116. exactly like it is here
  3117. --bar BAR bar help
  3118. '''
  3119. version = ''
  3120. class TestHelpArgumentDefaults(HelpTestCase):
  3121. """Test the ArgumentDefaultsHelpFormatter"""
  3122. parser_signature = Sig(
  3123. prog='PROG', formatter_class=argparse.ArgumentDefaultsHelpFormatter,
  3124. description='description')
  3125. argument_signatures = [
  3126. Sig('--foo', help='foo help - oh and by the way, %(default)s'),
  3127. Sig('--bar', action='store_true', help='bar help'),
  3128. Sig('spam', help='spam help'),
  3129. Sig('badger', nargs='?', default='wooden', help='badger help'),
  3130. ]
  3131. argument_group_signatures = [
  3132. (Sig('title', description='description'),
  3133. [Sig('--baz', type=int, default=42, help='baz help')]),
  3134. ]
  3135. usage = '''\
  3136. usage: PROG [-h] [--foo FOO] [--bar] [--baz BAZ] spam [badger]
  3137. '''
  3138. help = usage + '''\
  3139. description
  3140. positional arguments:
  3141. spam spam help
  3142. badger badger help (default: wooden)
  3143. optional arguments:
  3144. -h, --help show this help message and exit
  3145. --foo FOO foo help - oh and by the way, None
  3146. --bar bar help (default: False)
  3147. title:
  3148. description
  3149. --baz BAZ baz help (default: 42)
  3150. '''
  3151. version = ''
  3152. class TestHelpVersionAction(HelpTestCase):
  3153. """Test the default help for the version action"""
  3154. parser_signature = Sig(prog='PROG', description='description')
  3155. argument_signatures = [Sig('-V', '--version', action='version', version='3.6')]
  3156. argument_group_signatures = []
  3157. usage = '''\
  3158. usage: PROG [-h] [-V]
  3159. '''
  3160. help = usage + '''\
  3161. description
  3162. optional arguments:
  3163. -h, --help show this help message and exit
  3164. -V, --version show program's version number and exit
  3165. '''
  3166. version = ''
  3167. class TestHelpSubparsersOrdering(HelpTestCase):
  3168. """Test ordering of subcommands in help matches the code"""
  3169. parser_signature = Sig(prog='PROG',
  3170. description='display some subcommands',
  3171. version='0.1')
  3172. subparsers_signatures = [Sig(name=name)
  3173. for name in ('a', 'b', 'c', 'd', 'e')]
  3174. usage = '''\
  3175. usage: PROG [-h] [-v] {a,b,c,d,e} ...
  3176. '''
  3177. help = usage + '''\
  3178. display some subcommands
  3179. positional arguments:
  3180. {a,b,c,d,e}
  3181. optional arguments:
  3182. -h, --help show this help message and exit
  3183. -v, --version show program's version number and exit
  3184. '''
  3185. version = '''\
  3186. 0.1
  3187. '''
  3188. class TestHelpSubparsersWithHelpOrdering(HelpTestCase):
  3189. """Test ordering of subcommands in help matches the code"""
  3190. parser_signature = Sig(prog='PROG',
  3191. description='display some subcommands',
  3192. version='0.1')
  3193. subcommand_data = (('a', 'a subcommand help'),
  3194. ('b', 'b subcommand help'),
  3195. ('c', 'c subcommand help'),
  3196. ('d', 'd subcommand help'),
  3197. ('e', 'e subcommand help'),
  3198. )
  3199. subparsers_signatures = [Sig(name=name, help=help)
  3200. for name, help in subcommand_data]
  3201. usage = '''\
  3202. usage: PROG [-h] [-v] {a,b,c,d,e} ...
  3203. '''
  3204. help = usage + '''\
  3205. display some subcommands
  3206. positional arguments:
  3207. {a,b,c,d,e}
  3208. a a subcommand help
  3209. b b subcommand help
  3210. c c subcommand help
  3211. d d subcommand help
  3212. e e subcommand help
  3213. optional arguments:
  3214. -h, --help show this help message and exit
  3215. -v, --version show program's version number and exit
  3216. '''
  3217. version = '''\
  3218. 0.1
  3219. '''
  3220. # =====================================
  3221. # Optional/Positional constructor tests
  3222. # =====================================
  3223. class TestInvalidArgumentConstructors(TestCase):
  3224. """Test a bunch of invalid Argument constructors"""
  3225. def assertTypeError(self, *args, **kwargs):
  3226. parser = argparse.ArgumentParser()
  3227. self.assertRaises(TypeError, parser.add_argument,
  3228. *args, **kwargs)
  3229. def assertValueError(self, *args, **kwargs):
  3230. parser = argparse.ArgumentParser()
  3231. self.assertRaises(ValueError, parser.add_argument,
  3232. *args, **kwargs)
  3233. def test_invalid_keyword_arguments(self):
  3234. self.assertTypeError('-x', bar=None)
  3235. self.assertTypeError('-y', callback='foo')
  3236. self.assertTypeError('-y', callback_args=())
  3237. self.assertTypeError('-y', callback_kwargs={})
  3238. def test_missing_destination(self):
  3239. self.assertTypeError()
  3240. for action in ['append', 'store']:
  3241. self.assertTypeError(action=action)
  3242. def test_invalid_option_strings(self):
  3243. self.assertValueError('--')
  3244. self.assertValueError('---')
  3245. def test_invalid_type(self):
  3246. self.assertValueError('--foo', type='int')
  3247. self.assertValueError('--foo', type=(int, float))
  3248. def test_invalid_action(self):
  3249. self.assertValueError('-x', action='foo')
  3250. self.assertValueError('foo', action='baz')
  3251. self.assertValueError('--foo', action=('store', 'append'))
  3252. parser = argparse.ArgumentParser()
  3253. try:
  3254. parser.add_argument("--foo", action="store-true")
  3255. except ValueError:
  3256. e = sys.exc_info()[1]
  3257. expected = 'unknown action'
  3258. msg = 'expected %r, found %r' % (expected, e)
  3259. self.assertTrue(expected in str(e), msg)
  3260. def test_multiple_dest(self):
  3261. parser = argparse.ArgumentParser()
  3262. parser.add_argument(dest='foo')
  3263. try:
  3264. parser.add_argument('bar', dest='baz')
  3265. except ValueError:
  3266. e = sys.exc_info()[1]
  3267. expected = 'dest supplied twice for positional argument'
  3268. msg = 'expected %r, found %r' % (expected, e)
  3269. self.assertTrue(expected in str(e), msg)
  3270. def test_no_argument_actions(self):
  3271. for action in ['store_const', 'store_true', 'store_false',
  3272. 'append_const', 'count']:
  3273. for attrs in [dict(type=int), dict(nargs='+'),
  3274. dict(choices='ab')]:
  3275. self.assertTypeError('-x', action=action, **attrs)
  3276. def test_no_argument_no_const_actions(self):
  3277. # options with zero arguments
  3278. for action in ['store_true', 'store_false', 'count']:
  3279. # const is always disallowed
  3280. self.assertTypeError('-x', const='foo', action=action)
  3281. # nargs is always disallowed
  3282. self.assertTypeError('-x', nargs='*', action=action)
  3283. def test_more_than_one_argument_actions(self):
  3284. for action in ['store', 'append']:
  3285. # nargs=0 is disallowed
  3286. self.assertValueError('-x', nargs=0, action=action)
  3287. self.assertValueError('spam', nargs=0, action=action)
  3288. # const is disallowed with non-optional arguments
  3289. for nargs in [1, '*', '+']:
  3290. self.assertValueError('-x', const='foo',
  3291. nargs=nargs, action=action)
  3292. self.assertValueError('spam', const='foo',
  3293. nargs=nargs, action=action)
  3294. def test_required_const_actions(self):
  3295. for action in ['store_const', 'append_const']:
  3296. # nargs is always disallowed
  3297. self.assertTypeError('-x', nargs='+', action=action)
  3298. def test_parsers_action_missing_params(self):
  3299. self.assertTypeError('command', action='parsers')
  3300. self.assertTypeError('command', action='parsers', prog='PROG')
  3301. self.assertTypeError('command', action='parsers',
  3302. parser_class=argparse.ArgumentParser)
  3303. def test_required_positional(self):
  3304. self.assertTypeError('foo', required=True)
  3305. def test_user_defined_action(self):
  3306. class Success(Exception):
  3307. pass
  3308. class Action(object):
  3309. def __init__(self,
  3310. option_strings,
  3311. dest,
  3312. const,
  3313. default,
  3314. required=False):
  3315. if dest == 'spam':
  3316. if const is Success:
  3317. if default is Success:
  3318. raise Success()
  3319. def __call__(self, *args, **kwargs):
  3320. pass
  3321. parser = argparse.ArgumentParser()
  3322. self.assertRaises(Success, parser.add_argument, '--spam',
  3323. action=Action, default=Success, const=Success)
  3324. self.assertRaises(Success, parser.add_argument, 'spam',
  3325. action=Action, default=Success, const=Success)
  3326. # ================================
  3327. # Actions returned by add_argument
  3328. # ================================
  3329. class TestActionsReturned(TestCase):
  3330. def test_dest(self):
  3331. parser = argparse.ArgumentParser()
  3332. action = parser.add_argument('--foo')
  3333. self.assertEqual(action.dest, 'foo')
  3334. action = parser.add_argument('-b', '--bar')
  3335. self.assertEqual(action.dest, 'bar')
  3336. action = parser.add_argument('-x', '-y')
  3337. self.assertEqual(action.dest, 'x')
  3338. def test_misc(self):
  3339. parser = argparse.ArgumentParser()
  3340. action = parser.add_argument('--foo', nargs='?', const=42,
  3341. default=84, type=int, choices=[1, 2],
  3342. help='FOO', metavar='BAR', dest='baz')
  3343. self.assertEqual(action.nargs, '?')
  3344. self.assertEqual(action.const, 42)
  3345. self.assertEqual(action.default, 84)
  3346. self.assertEqual(action.type, int)
  3347. self.assertEqual(action.choices, [1, 2])
  3348. self.assertEqual(action.help, 'FOO')
  3349. self.assertEqual(action.metavar, 'BAR')
  3350. self.assertEqual(action.dest, 'baz')
  3351. # ================================
  3352. # Argument conflict handling tests
  3353. # ================================
  3354. class TestConflictHandling(TestCase):
  3355. def test_bad_type(self):
  3356. self.assertRaises(ValueError, argparse.ArgumentParser,
  3357. conflict_handler='foo')
  3358. def test_conflict_error(self):
  3359. parser = argparse.ArgumentParser()
  3360. parser.add_argument('-x')
  3361. self.assertRaises(argparse.ArgumentError,
  3362. parser.add_argument, '-x')
  3363. parser.add_argument('--spam')
  3364. self.assertRaises(argparse.ArgumentError,
  3365. parser.add_argument, '--spam')
  3366. def test_resolve_error(self):
  3367. get_parser = argparse.ArgumentParser
  3368. parser = get_parser(prog='PROG', conflict_handler='resolve')
  3369. parser.add_argument('-x', help='OLD X')
  3370. parser.add_argument('-x', help='NEW X')
  3371. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  3372. usage: PROG [-h] [-x X]
  3373. optional arguments:
  3374. -h, --help show this help message and exit
  3375. -x X NEW X
  3376. '''))
  3377. parser.add_argument('--spam', metavar='OLD_SPAM')
  3378. parser.add_argument('--spam', metavar='NEW_SPAM')
  3379. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  3380. usage: PROG [-h] [-x X] [--spam NEW_SPAM]
  3381. optional arguments:
  3382. -h, --help show this help message and exit
  3383. -x X NEW X
  3384. --spam NEW_SPAM
  3385. '''))
  3386. # =============================
  3387. # Help and Version option tests
  3388. # =============================
  3389. class TestOptionalsHelpVersionActions(TestCase):
  3390. """Test the help and version actions"""
  3391. def _get_error(self, func, *args, **kwargs):
  3392. try:
  3393. func(*args, **kwargs)
  3394. except ArgumentParserError:
  3395. return sys.exc_info()[1]
  3396. else:
  3397. self.assertRaises(ArgumentParserError, func, *args, **kwargs)
  3398. def assertPrintHelpExit(self, parser, args_str):
  3399. self.assertEqual(
  3400. parser.format_help(),
  3401. self._get_error(parser.parse_args, args_str.split()).stdout)
  3402. def assertPrintVersionExit(self, parser, args_str):
  3403. self.assertEqual(
  3404. parser.format_version(),
  3405. self._get_error(parser.parse_args, args_str.split()).stderr)
  3406. def assertArgumentParserError(self, parser, *args):
  3407. self.assertRaises(ArgumentParserError, parser.parse_args, args)
  3408. def test_version(self):
  3409. parser = ErrorRaisingArgumentParser(version='1.0')
  3410. self.assertPrintHelpExit(parser, '-h')
  3411. self.assertPrintHelpExit(parser, '--help')
  3412. self.assertPrintVersionExit(parser, '-v')
  3413. self.assertPrintVersionExit(parser, '--version')
  3414. def test_version_format(self):
  3415. parser = ErrorRaisingArgumentParser(prog='PPP', version='%(prog)s 3.5')
  3416. msg = self._get_error(parser.parse_args, ['-v']).stderr
  3417. self.assertEqual('PPP 3.5\n', msg)
  3418. def test_version_no_help(self):
  3419. parser = ErrorRaisingArgumentParser(add_help=False, version='1.0')
  3420. self.assertArgumentParserError(parser, '-h')
  3421. self.assertArgumentParserError(parser, '--help')
  3422. self.assertPrintVersionExit(parser, '-v')
  3423. self.assertPrintVersionExit(parser, '--version')
  3424. def test_version_action(self):
  3425. parser = ErrorRaisingArgumentParser(prog='XXX')
  3426. parser.add_argument('-V', action='version', version='%(prog)s 3.7')
  3427. msg = self._get_error(parser.parse_args, ['-V']).stderr
  3428. self.assertEqual('XXX 3.7\n', msg)
  3429. def test_no_help(self):
  3430. parser = ErrorRaisingArgumentParser(add_help=False)
  3431. self.assertArgumentParserError(parser, '-h')
  3432. self.assertArgumentParserError(parser, '--help')
  3433. self.assertArgumentParserError(parser, '-v')
  3434. self.assertArgumentParserError(parser, '--version')
  3435. def test_alternate_help_version(self):
  3436. parser = ErrorRaisingArgumentParser()
  3437. parser.add_argument('-x', action='help')
  3438. parser.add_argument('-y', action='version')
  3439. self.assertPrintHelpExit(parser, '-x')
  3440. self.assertPrintVersionExit(parser, '-y')
  3441. self.assertArgumentParserError(parser, '-v')
  3442. self.assertArgumentParserError(parser, '--version')
  3443. def test_help_version_extra_arguments(self):
  3444. parser = ErrorRaisingArgumentParser(version='1.0')
  3445. parser.add_argument('-x', action='store_true')
  3446. parser.add_argument('y')
  3447. # try all combinations of valid prefixes and suffixes
  3448. valid_prefixes = ['', '-x', 'foo', '-x bar', 'baz -x']
  3449. valid_suffixes = valid_prefixes + ['--bad-option', 'foo bar baz']
  3450. for prefix in valid_prefixes:
  3451. for suffix in valid_suffixes:
  3452. format = '%s %%s %s' % (prefix, suffix)
  3453. self.assertPrintHelpExit(parser, format % '-h')
  3454. self.assertPrintHelpExit(parser, format % '--help')
  3455. self.assertPrintVersionExit(parser, format % '-v')
  3456. self.assertPrintVersionExit(parser, format % '--version')
  3457. # ======================
  3458. # str() and repr() tests
  3459. # ======================
  3460. class TestStrings(TestCase):
  3461. """Test str() and repr() on Optionals and Positionals"""
  3462. def assertStringEqual(self, obj, result_string):
  3463. for func in [str, repr]:
  3464. self.assertEqual(func(obj), result_string)
  3465. def test_optional(self):
  3466. option = argparse.Action(
  3467. option_strings=['--foo', '-a', '-b'],
  3468. dest='b',
  3469. type='int',
  3470. nargs='+',
  3471. default=42,
  3472. choices=[1, 2, 3],
  3473. help='HELP',
  3474. metavar='METAVAR')
  3475. string = (
  3476. "Action(option_strings=['--foo', '-a', '-b'], dest='b', "
  3477. "nargs='+', const=None, default=42, type='int', "
  3478. "choices=[1, 2, 3], help='HELP', metavar='METAVAR')")
  3479. self.assertStringEqual(option, string)
  3480. def test_argument(self):
  3481. argument = argparse.Action(
  3482. option_strings=[],
  3483. dest='x',
  3484. type=float,
  3485. nargs='?',
  3486. default=2.5,
  3487. choices=[0.5, 1.5, 2.5],
  3488. help='H HH H',
  3489. metavar='MV MV MV')
  3490. string = (
  3491. "Action(option_strings=[], dest='x', nargs='?', "
  3492. "const=None, default=2.5, type=%r, choices=[0.5, 1.5, 2.5], "
  3493. "help='H HH H', metavar='MV MV MV')" % float)
  3494. self.assertStringEqual(argument, string)
  3495. def test_namespace(self):
  3496. ns = argparse.Namespace(foo=42, bar='spam')
  3497. string = "Namespace(bar='spam', foo=42)"
  3498. self.assertStringEqual(ns, string)
  3499. def test_parser(self):
  3500. parser = argparse.ArgumentParser(prog='PROG')
  3501. string = (
  3502. "ArgumentParser(prog='PROG', usage=None, description=None, "
  3503. "version=None, formatter_class=%r, conflict_handler='error', "
  3504. "add_help=True)" % argparse.HelpFormatter)
  3505. self.assertStringEqual(parser, string)
  3506. # ===============
  3507. # Namespace tests
  3508. # ===============
  3509. class TestNamespace(TestCase):
  3510. def test_constructor(self):
  3511. ns = argparse.Namespace()
  3512. self.assertRaises(AttributeError, getattr, ns, 'x')
  3513. ns = argparse.Namespace(a=42, b='spam')
  3514. self.assertEqual(ns.a, 42)
  3515. self.assertEqual(ns.b, 'spam')
  3516. def test_equality(self):
  3517. ns1 = argparse.Namespace(a=1, b=2)
  3518. ns2 = argparse.Namespace(b=2, a=1)
  3519. ns3 = argparse.Namespace(a=1)
  3520. ns4 = argparse.Namespace(b=2)
  3521. self.assertEqual(ns1, ns2)
  3522. self.assertNotEqual(ns1, ns3)
  3523. self.assertNotEqual(ns1, ns4)
  3524. self.assertNotEqual(ns2, ns3)
  3525. self.assertNotEqual(ns2, ns4)
  3526. self.assertTrue(ns1 != ns3)
  3527. self.assertTrue(ns1 != ns4)
  3528. self.assertTrue(ns2 != ns3)
  3529. self.assertTrue(ns2 != ns4)
  3530. # ===================
  3531. # File encoding tests
  3532. # ===================
  3533. class TestEncoding(TestCase):
  3534. def _test_module_encoding(self, path):
  3535. path, _ = os.path.splitext(path)
  3536. path += ".py"
  3537. with codecs.open(path, 'r', 'utf8') as f:
  3538. f.read()
  3539. def test_argparse_module_encoding(self):
  3540. self._test_module_encoding(argparse.__file__)
  3541. def test_test_argparse_module_encoding(self):
  3542. self._test_module_encoding(__file__)
  3543. # ===================
  3544. # ArgumentError tests
  3545. # ===================
  3546. class TestArgumentError(TestCase):
  3547. def test_argument_error(self):
  3548. msg = "my error here"
  3549. error = argparse.ArgumentError(None, msg)
  3550. self.assertEqual(str(error), msg)
  3551. # =======================
  3552. # ArgumentTypeError tests
  3553. # =======================
  3554. class TestArgumentTypeError(TestCase):
  3555. def test_argument_type_error(self):
  3556. def spam(string):
  3557. raise argparse.ArgumentTypeError('spam!')
  3558. parser = ErrorRaisingArgumentParser(prog='PROG', add_help=False)
  3559. parser.add_argument('x', type=spam)
  3560. try:
  3561. parser.parse_args(['XXX'])
  3562. except ArgumentParserError:
  3563. expected = 'usage: PROG x\nPROG: error: argument x: spam!\n'
  3564. msg = sys.exc_info()[1].stderr
  3565. self.assertEqual(expected, msg)
  3566. else:
  3567. self.fail()
  3568. # ======================
  3569. # parse_known_args tests
  3570. # ======================
  3571. class TestParseKnownArgs(TestCase):
  3572. def test_optionals(self):
  3573. parser = argparse.ArgumentParser()
  3574. parser.add_argument('--foo')
  3575. args, extras = parser.parse_known_args('--foo F --bar --baz'.split())
  3576. self.assertEqual(NS(foo='F'), args)
  3577. self.assertEqual(['--bar', '--baz'], extras)
  3578. def test_mixed(self):
  3579. parser = argparse.ArgumentParser()
  3580. parser.add_argument('-v', nargs='?', const=1, type=int)
  3581. parser.add_argument('--spam', action='store_false')
  3582. parser.add_argument('badger')
  3583. argv = ["B", "C", "--foo", "-v", "3", "4"]
  3584. args, extras = parser.parse_known_args(argv)
  3585. self.assertEqual(NS(v=3, spam=True, badger="B"), args)
  3586. self.assertEqual(["C", "--foo", "4"], extras)
  3587. # ==========================
  3588. # add_argument metavar tests
  3589. # ==========================
  3590. class TestAddArgumentMetavar(TestCase):
  3591. EXPECTED_MESSAGE = "length of metavar tuple does not match nargs"
  3592. def do_test_no_exception(self, nargs, metavar):
  3593. parser = argparse.ArgumentParser()
  3594. parser.add_argument("--foo", nargs=nargs, metavar=metavar)
  3595. def do_test_exception(self, nargs, metavar):
  3596. parser = argparse.ArgumentParser()
  3597. with self.assertRaises(ValueError) as cm:
  3598. parser.add_argument("--foo", nargs=nargs, metavar=metavar)
  3599. self.assertEqual(cm.exception.args[0], self.EXPECTED_MESSAGE)
  3600. # Unit tests for different values of metavar when nargs=None
  3601. def test_nargs_None_metavar_string(self):
  3602. self.do_test_no_exception(nargs=None, metavar="1")
  3603. def test_nargs_None_metavar_length0(self):
  3604. self.do_test_exception(nargs=None, metavar=tuple())
  3605. def test_nargs_None_metavar_length1(self):
  3606. self.do_test_no_exception(nargs=None, metavar=("1"))
  3607. def test_nargs_None_metavar_length2(self):
  3608. self.do_test_exception(nargs=None, metavar=("1", "2"))
  3609. def test_nargs_None_metavar_length3(self):
  3610. self.do_test_exception(nargs=None, metavar=("1", "2", "3"))
  3611. # Unit tests for different values of metavar when nargs=?
  3612. def test_nargs_optional_metavar_string(self):
  3613. self.do_test_no_exception(nargs="?", metavar="1")
  3614. def test_nargs_optional_metavar_length0(self):
  3615. self.do_test_exception(nargs="?", metavar=tuple())
  3616. def test_nargs_optional_metavar_length1(self):
  3617. self.do_test_no_exception(nargs="?", metavar=("1"))
  3618. def test_nargs_optional_metavar_length2(self):
  3619. self.do_test_exception(nargs="?", metavar=("1", "2"))
  3620. def test_nargs_optional_metavar_length3(self):
  3621. self.do_test_exception(nargs="?", metavar=("1", "2", "3"))
  3622. # Unit tests for different values of metavar when nargs=*
  3623. def test_nargs_zeroormore_metavar_string(self):
  3624. self.do_test_no_exception(nargs="*", metavar="1")
  3625. def test_nargs_zeroormore_metavar_length0(self):
  3626. self.do_test_exception(nargs="*", metavar=tuple())
  3627. def test_nargs_zeroormore_metavar_length1(self):
  3628. self.do_test_no_exception(nargs="*", metavar=("1"))
  3629. def test_nargs_zeroormore_metavar_length2(self):
  3630. self.do_test_no_exception(nargs="*", metavar=("1", "2"))
  3631. def test_nargs_zeroormore_metavar_length3(self):
  3632. self.do_test_exception(nargs="*", metavar=("1", "2", "3"))
  3633. # Unit tests for different values of metavar when nargs=+
  3634. def test_nargs_oneormore_metavar_string(self):
  3635. self.do_test_no_exception(nargs="+", metavar="1")
  3636. def test_nargs_oneormore_metavar_length0(self):
  3637. self.do_test_exception(nargs="+", metavar=tuple())
  3638. def test_nargs_oneormore_metavar_length1(self):
  3639. self.do_test_no_exception(nargs="+", metavar=("1"))
  3640. def test_nargs_oneormore_metavar_length2(self):
  3641. self.do_test_no_exception(nargs="+", metavar=("1", "2"))
  3642. def test_nargs_oneormore_metavar_length3(self):
  3643. self.do_test_exception(nargs="+", metavar=("1", "2", "3"))
  3644. # Unit tests for different values of metavar when nargs=...
  3645. def test_nargs_remainder_metavar_string(self):
  3646. self.do_test_no_exception(nargs="...", metavar="1")
  3647. def test_nargs_remainder_metavar_length0(self):
  3648. self.do_test_no_exception(nargs="...", metavar=tuple())
  3649. def test_nargs_remainder_metavar_length1(self):
  3650. self.do_test_no_exception(nargs="...", metavar=("1"))
  3651. def test_nargs_remainder_metavar_length2(self):
  3652. self.do_test_no_exception(nargs="...", metavar=("1", "2"))
  3653. def test_nargs_remainder_metavar_length3(self):
  3654. self.do_test_no_exception(nargs="...", metavar=("1", "2", "3"))
  3655. # Unit tests for different values of metavar when nargs=A...
  3656. def test_nargs_parser_metavar_string(self):
  3657. self.do_test_no_exception(nargs="A...", metavar="1")
  3658. def test_nargs_parser_metavar_length0(self):
  3659. self.do_test_exception(nargs="A...", metavar=tuple())
  3660. def test_nargs_parser_metavar_length1(self):
  3661. self.do_test_no_exception(nargs="A...", metavar=("1"))
  3662. def test_nargs_parser_metavar_length2(self):
  3663. self.do_test_exception(nargs="A...", metavar=("1", "2"))
  3664. def test_nargs_parser_metavar_length3(self):
  3665. self.do_test_exception(nargs="A...", metavar=("1", "2", "3"))
  3666. # Unit tests for different values of metavar when nargs=1
  3667. def test_nargs_1_metavar_string(self):
  3668. self.do_test_no_exception(nargs=1, metavar="1")
  3669. def test_nargs_1_metavar_length0(self):
  3670. self.do_test_exception(nargs=1, metavar=tuple())
  3671. def test_nargs_1_metavar_length1(self):
  3672. self.do_test_no_exception(nargs=1, metavar=("1"))
  3673. def test_nargs_1_metavar_length2(self):
  3674. self.do_test_exception(nargs=1, metavar=("1", "2"))
  3675. def test_nargs_1_metavar_length3(self):
  3676. self.do_test_exception(nargs=1, metavar=("1", "2", "3"))
  3677. # Unit tests for different values of metavar when nargs=2
  3678. def test_nargs_2_metavar_string(self):
  3679. self.do_test_no_exception(nargs=2, metavar="1")
  3680. def test_nargs_2_metavar_length0(self):
  3681. self.do_test_exception(nargs=2, metavar=tuple())
  3682. def test_nargs_2_metavar_length1(self):
  3683. self.do_test_no_exception(nargs=2, metavar=("1"))
  3684. def test_nargs_2_metavar_length2(self):
  3685. self.do_test_no_exception(nargs=2, metavar=("1", "2"))
  3686. def test_nargs_2_metavar_length3(self):
  3687. self.do_test_exception(nargs=2, metavar=("1", "2", "3"))
  3688. # Unit tests for different values of metavar when nargs=3
  3689. def test_nargs_3_metavar_string(self):
  3690. self.do_test_no_exception(nargs=3, metavar="1")
  3691. def test_nargs_3_metavar_length0(self):
  3692. self.do_test_exception(nargs=3, metavar=tuple())
  3693. def test_nargs_3_metavar_length1(self):
  3694. self.do_test_no_exception(nargs=3, metavar=("1"))
  3695. def test_nargs_3_metavar_length2(self):
  3696. self.do_test_exception(nargs=3, metavar=("1", "2"))
  3697. def test_nargs_3_metavar_length3(self):
  3698. self.do_test_no_exception(nargs=3, metavar=("1", "2", "3"))
  3699. # ============================
  3700. # from argparse import * tests
  3701. # ============================
  3702. class TestImportStar(TestCase):
  3703. def test(self):
  3704. for name in argparse.__all__:
  3705. self.assertTrue(hasattr(argparse, name))
  3706. def test_all_exports_everything_but_modules(self):
  3707. items = [
  3708. name
  3709. for name, value in vars(argparse).items()
  3710. if not name.startswith("_")
  3711. if not inspect.ismodule(value)
  3712. ]
  3713. self.assertEqual(sorted(items), sorted(argparse.__all__))
  3714. def test_main():
  3715. # silence warnings about version argument - these are expected
  3716. with test_support.check_warnings(
  3717. ('The "version" argument to ArgumentParser is deprecated.',
  3718. DeprecationWarning),
  3719. ('The (format|print)_version method is deprecated',
  3720. DeprecationWarning)):
  3721. test_support.run_unittest(__name__)
  3722. # Remove global references to avoid looking like we have refleaks.
  3723. RFile.seen = {}
  3724. WFile.seen = set()
  3725. if __name__ == '__main__':
  3726. test_main()