Add reformatting test for a list comprehension

google · Mar 26, 2015 · 3756f78 · 3756f78
1 parent 3256077
commit 3756f78
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/yapftests/reformatter_test.py b/yapftests/reformatter_test.py
@@ -343,6 +343,19 @@ def x(self):
     uwlines = _ParseAndUnwrap(unformatted_code)
     self.assertEqual(expected_formatted_code, reformatter.Reformat(uwlines))
 
+  def testListComprehension(self):
+    unformatted_code = textwrap.dedent("""\
+        def given(y):
+            [k for k in ()
+              if k in y]
+        """)
+    expected_formatted_code = textwrap.dedent("""\
+        def given(y):
+          [k for k in () if k in y]
+        """)
+    uwlines = _ParseAndUnwrap(unformatted_code)
+    self.assertEqual(expected_formatted_code, reformatter.Reformat(uwlines))
+
   def testOpeningAndClosingBrackets(self):
     unformatted_code = textwrap.dedent("""\
         foo( ( 1, 2, 3, ) )