Fixed #33657 -- Allowed customizing formatter class of argument parsers.

This commit is contained in:
Abhinav Yadav 2022-06-20 21:04:52 +05:30 committed by GitHub
parent d7f5bfd241
commit 2887b9f67c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 2 deletions

View file

@ -1,4 +1,5 @@
import os
from argparse import ArgumentDefaultsHelpFormatter
from io import StringIO
from unittest import mock
@ -408,8 +409,14 @@ class CommandTests(SimpleTestCase):
def test_create_parser_kwargs(self):
"""BaseCommand.create_parser() passes kwargs to CommandParser."""
epilog = "some epilog text"
parser = BaseCommand().create_parser("prog_name", "subcommand", epilog=epilog)
parser = BaseCommand().create_parser(
"prog_name",
"subcommand",
epilog=epilog,
formatter_class=ArgumentDefaultsHelpFormatter,
)
self.assertEqual(parser.epilog, epilog)
self.assertEqual(parser.formatter_class, ArgumentDefaultsHelpFormatter)
def test_outputwrapper_flush(self):
out = StringIO()