Fix test.
diff --git a/yapftests/yapf_test.py b/yapftests/yapf_test.py
index 6425370..a545008 100644
--- a/yapftests/yapf_test.py
+++ b/yapftests/yapf_test.py
@@ -34,7 +34,8 @@
class FormatCodeTest(unittest.TestCase):
def _Check(self, unformatted_code, expected_formatted_code):
- formatted_code = yapf_api.FormatCode(unformatted_code)
+ formatted_code = yapf_api.FormatCode(unformatted_code,
+ style_config='chromium')
self.assertEqual(expected_formatted_code, formatted_code)
def testSimple(self):
@@ -45,9 +46,11 @@
def testNoEndingNewline(self):
unformatted_code = textwrap.dedent(u"""\
- if True: pass""")
+ if True:
+ pass""")
expected_formatted_code = textwrap.dedent(u"""\
- if True: pass
+ if True:
+ pass
""")
self._Check(unformatted_code, expected_formatted_code)