def do_format_test(self, input_str, output_str, strip_len=6):
"""
Run the formatter on the input string and assert that the result matches
the output string
"""
input_str = strip_indent(input_str, strip_len)
output_str = strip_indent(output_str, strip_len)
if sys.version_info[0] < 3:
assert isinstance(input_str, unicode)
infile = io.StringIO(input_str)
outfile = io.StringIO()
__main__.process_file(self.config, infile, outfile)
delta_lines = list(
difflib.unified_diff(outfile.getvalue().split('\n'),
output_str.split('\n')))
delta = '\n'.join(delta_lines[2:])
if outfile.getvalue() != output_str:
message = ('Input text:\n-----------------\n{}\n'
'Output text:\n-----------------\n{}\n'
'Expected Output:\n-----------------\n{}\n'
'Diff:\n-----------------\n{}'.format(
input_str, outfile.getvalue(), output_str, delta))
if sys.version_info[0] < 3:
message = message.encode('utf-8')
raise AssertionError(message)
def do_format_test(self, input_str, output_str, strip_len=6):
"""
Run the formatter on the input string and assert that the result matches
the output string
"""
input_str = strip_indent(input_str, strip_len)
output_str = strip_indent(output_str, strip_len)
if sys.version_info[0] < 3:
assert isinstance(input_str, unicode)
infile = io.StringIO(input_str)
outfile = io.StringIO()
__main__.process_file(self.config, infile, outfile)
delta_lines = list(difflib.unified_diff(outfile.getvalue().split('\n'),
output_str.split('\n')))
delta = '\n'.join(delta_lines[2:])
if outfile.getvalue() != output_str:
message = ('Input text:\n-----------------\n{}\n'
'Output text:\n-----------------\n{}\n'
'Expected Output:\n-----------------\n{}\n'
'Diff:\n-----------------\n{}'
.format(input_str,
outfile.getvalue(),
output_str,
delta))
if sys.version_info[0] < 3:
message = message.encode('utf-8')
raise AssertionError(message)
def do_format_test(self, input_str, output_str):
"""
Run the formatter on the input string and assert that the result matches
the output string
"""
infile = StringIO.StringIO(input_str)
outfile = StringIO.StringIO()
__main__.process_file(self.config, infile, outfile)
if outfile.getvalue() != output_str:
message = ('Input text:\n-----------------\n{}\n'
'Output text:\n-----------------\n{}\n'
'Expected Output:\n-----------------\n{}'.format(
input_str, outfile.getvalue(), output_str))
raise AssertionError(message)
def assert_format(test, input_str, output_str=None, strip_len=0):
"""
Run the formatter on the input string and assert that the result matches
the output string
"""
if output_str is None:
output_str = input_str
input_str = strip_indent(input_str, strip_len)
output_str = strip_indent(output_str, strip_len)
if sys.version_info[0] < 3:
assert isinstance(input_str, unicode)
actual_str, _reflow_valid = __main__.process_file(test.config, input_str)
delta_lines = list(
difflib.unified_diff(output_str.split('\n'), actual_str.split('\n')))
delta = '\n'.join(delta_lines[2:])
if actual_str != output_str:
message = ('Input text:\n-----------------\n{}\n'
'Output text:\n-----------------\n{}\n'
'Expected Output:\n-----------------\n{}\n'
'Diff:\n-----------------\n{}'.format(
input_str, actual_str, output_str, delta))
if sys.version_info[0] < 3:
message = message.encode('utf-8')
raise AssertionError(message)
