diff --git a/tests/unit/test_base_formatter.py b/tests/unit/test_base_formatter.py index 463dff7ab..52a860c86 100644 --- a/tests/unit/test_base_formatter.py +++ b/tests/unit/test_base_formatter.py @@ -141,16 +141,17 @@ def test_write_without_stdout_buffer(): line = "Something to write" source = "source" - buffer_side_effect = AttributeError("'_io.StringIO' object has no \ - attribute 'buffer'") + buffer_side_effect = AttributeError( + "'_io.StringIO' object has no \ + attribute 'buffer'" + ) buffer_write_mock = mock.Mock(side_effect=buffer_side_effect) write_mock = mock.Mock() mock.patch("flake8.formatting.base.sys.stdout.write", write_mock) with mock.patch( - "flake8.formatting.base.sys.stdout.buffer.write", - buffer_write_mock + "flake8.formatting.base.sys.stdout.buffer.write", buffer_write_mock ): formatter = base.BaseFormatter(options()) formatter.write(line, source)