Merge branch 'formatter-print-observe-newline' into 'master'

Respect a formatter's newline setting when printing

See merge request pycqa/flake8!222
This commit is contained in:
Ian Stapleton Cordasco 2018-01-01 00:57:09 +00:00
commit 6381d86f00
2 changed files with 5 additions and 5 deletions

View file

@ -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.

View file

@ -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'),
]