diff --git a/tests/unit_tests/test_tethys_cli/test_app_settings_command.py b/tests/unit_tests/test_tethys_cli/test_app_settings_command.py index 97e1c299f..d52bb524c 100644 --- a/tests/unit_tests/test_tethys_cli/test_app_settings_command.py +++ b/tests/unit_tests/test_tethys_cli/test_app_settings_command.py @@ -8,7 +8,9 @@ class TestCliAppSettingsCommand(unittest.TestCase): def setUp(self): - setup_django_patcher = mock.patch("tethys_cli.app_settings_commands.setup_django") + setup_django_patcher = mock.patch( + "tethys_cli.app_settings_commands.setup_django" + ) setup_django_patcher.start() self.addCleanup(setup_django_patcher.stop) diff --git a/tests/unit_tests/test_tethys_cli/test_site_commands.py b/tests/unit_tests/test_tethys_cli/test_site_commands.py index 286154f96..71ba0c541 100644 --- a/tests/unit_tests/test_tethys_cli/test_site_commands.py +++ b/tests/unit_tests/test_tethys_cli/test_site_commands.py @@ -86,7 +86,12 @@ def test_gen_site_content_with_yaml( @mock.patch("tethys_cli.site_commands.Path") @mock.patch("tethys_cli.site_commands.setup_django") def test_gen_site_content_with_yaml_invalid_category( - self, mock_setup_django, mock_path, mock_update_settings, mock_load_yaml, mock_warn + self, + mock_setup_django, + mock_path, + mock_update_settings, + mock_load_yaml, + mock_warn, ): mock_file_path = mock.MagicMock() mock_path.return_value = mock_file_path diff --git a/tethys_cli/cli_helpers.py b/tethys_cli/cli_helpers.py index d6fffc5cf..7e4431e5a 100644 --- a/tethys_cli/cli_helpers.py +++ b/tethys_cli/cli_helpers.py @@ -76,6 +76,7 @@ def wrapped(*args, **kwargs): result = func(*args, **kwargs) sys.stdout = stdout return result + return wrapped