diff options
| -rw-r--r-- | src/flake8/formatting/base.py | 2 | ||||
| -rw-r--r-- | tests/unit/test_base_formatter.py | 8 |
2 files changed, 5 insertions, 5 deletions
diff --git a/src/flake8/formatting/base.py b/src/flake8/formatting/base.py index c4c67d5..1443e4c 100644 --- a/src/flake8/formatting/base.py +++ b/src/flake8/formatting/base.py @@ -172,7 +172,7 @@ class BaseFormatter(object): if self.output_fd is not None: self.output_fd.write(output + self.newline) if self.output_fd is None or self.options.tee: - print(output) + print(output, end=self.newline) def write(self, line, source): """Write the line either to the output file or stdout. diff --git a/tests/unit/test_base_formatter.py b/tests/unit/test_base_formatter.py index 87b5fee..e4dab2e 100644 --- a/tests/unit/test_base_formatter.py +++ b/tests/unit/test_base_formatter.py @@ -92,8 +92,8 @@ def test_write_uses_an_output_file(tee): if tee: assert print_func.called assert print_func.mock_calls == [ - mock.call(line), - mock.call(source), + mock.call(line, end='\n'), + mock.call(source, end='\n'), ] else: assert not print_func.called @@ -118,8 +118,8 @@ def test_write_uses_print(print_function): assert print_function.called is True assert print_function.call_count == 2 assert print_function.mock_calls == [ - mock.call(line), - mock.call(source), + mock.call(line, end='\n'), + mock.call(source, end='\n'), ] |