From 9b8f90831426ff518b16787ab31107c38f97d819 Mon Sep 17 00:00:00 2001 From: Ruairidh MacLeod <5160559+rkm@users.noreply.github.com> Date: Fri, 5 Jun 2020 14:18:00 +0100 Subject: [PATCH] fix JobsArgument --help output --- src/flake8/main/options.py | 4 ++++ tests/unit/test_option_manager.py | 6 ++++++ 2 files changed, 10 insertions(+) diff --git a/src/flake8/main/options.py b/src/flake8/main/options.py index a620ed9..4504634 100644 --- a/src/flake8/main/options.py +++ b/src/flake8/main/options.py @@ -82,6 +82,10 @@ class JobsArgument: "{!r} must be 'auto' or an integer.".format(arg), ) + def __str__(self): + """Format our JobsArgument class.""" + return "auto" if self.is_auto else str(self.n_jobs) + def register_default_options(option_manager): """Register the default options on our OptionManager. diff --git a/tests/unit/test_option_manager.py b/tests/unit/test_option_manager.py index 3a63db9..09714b9 100644 --- a/tests/unit/test_option_manager.py +++ b/tests/unit/test_option_manager.py @@ -373,3 +373,9 @@ def test_parse_invalid_jobs_argument(optmanager, capsys): "'foo' must be 'auto' or an integer.\n" ) assert expected in output + + +def test_jobs_argument_str(): + """Test that JobsArgument has a correct __str__.""" + assert str(JobsArgument("auto")) == "auto" + assert str(JobsArgument("123")) == "123"