def test_format_docstrings(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO('''
def f():
"""
triple-quoted docstring
with multiple lines
"""
x = """
this is a regular multi-line string, not a docstring
"""
return x
''')
main(host, ['-'])
self.assertMultiLineEqual(
host.stdout.getvalue(), '''
def f():
"""triple-quoted docstring
with multiple lines
"""
x = """
this is a regular multi-line string, not a docstring
"""
return x
''')
def test_format_docstrings(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO(
'''
def f():
"""
triple-quoted docstring
with multiple lines
"""
x = """
this is a regular multi-line string, not a docstring
"""
return x
'''
)
main(host, ["-"])
self.assertMultiLineEqual(
host.stdout.getvalue(),
'''
def f():
"""triple-quoted docstring
with multiple lines
"""
x = """
this is a regular multi-line string, not a docstring
"""
return x
''',
)
def test_format_docstrings_indentation(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO('''
def f():
"""This is a docstring
With extra indentation on this line.
"""
''')
main(host, ['-'])
self.assertMultiLineEqual(
host.stdout.getvalue(), '''
def f():
"""This is a docstring
With extra indentation on this line.
"""
''')
def test_format_docstrings_indentation(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO(
'''
def f():
"""This is a docstring
With extra indentation on this line.
"""
'''
)
main(host, ["-"])
self.assertMultiLineEqual(
host.stdout.getvalue(),
'''
def f():
"""This is a docstring
With extra indentation on this line.
"""
''',
)
def test_stdin_no_changes(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO(ACTUAL_INPUT)
main(host, ['--no-autopep8', '--leave-strings-alone', '-'])
self.assertMultiLineEqual(host.stdout.getvalue(), ACTUAL_INPUT)
def test_stdin_chromium(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO(ACTUAL_INPUT)
main(host, ['--chromium', '-'])
self.assertMultiLineEqual(host.stdout.getvalue(),
EXPECTED_CHROMIUM_OUTPUT)
def test_stdin_blink(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO(ACTUAL_INPUT)
main(host, ['-'])
self.assertMultiLineEqual(host.stdout.getvalue(),
EXPECTED_BLINK_OUTPUT)
def test_stdin_only_double_quoting(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO(ACTUAL_INPUT)
main(host, ['--no-autopep8', '--double-quote-strings', '-'])
self.assertMultiLineEqual(host.stdout.getvalue(),
EXPECTED_ONLY_DOUBLE_QUOTED_OUTPUT)
def test_stdin_only_double_quoting(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO(ACTUAL_INPUT)
main(host, ["--no-autopep8", "--double-quote-strings", "-"])
self.assertMultiLineEqual(host.stdout.getvalue(), EXPECTED_ONLY_DOUBLE_QUOTED_OUTPUT)
def test_stdin_no_changes(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO(ACTUAL_INPUT)
main(host, ["--no-autopep8", "--leave-strings-alone", "-"])
self.assertMultiLineEqual(host.stdout.getvalue(), ACTUAL_INPUT)
def test_stdin_chromium(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO(ACTUAL_INPUT)
main(host, ["--chromium", "-"])
self.assertMultiLineEqual(host.stdout.getvalue(), EXPECTED_CHROMIUM_OUTPUT)
def test_stdin_blink(self):
host = MockSystemHost()
host.stdin = StringIO.StringIO(ACTUAL_INPUT)
main(host, ["-"])
self.assertMultiLineEqual(host.stdout.getvalue(), EXPECTED_BLINK_OUTPUT)
