Add test with indented import

PyCQA · Oct 9, 2015 · 0b27929 · 0b27929
1 parent 353ed0d
commit 0b27929
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/test_autoflake.py b/test_autoflake.py
@@ -151,6 +151,21 @@ def test_filter_code(self):
 import os
 import re
 os.foo()
+""")))
+
+    def test_filter_code_with_conditional(self):
+        self.assertEqual(
+            """\
+import os
+if True:
+    pass
+os.foo()
+""",
+            ''.join(autoflake.filter_code("""\
+import os
+if True:
+    import re
+os.foo()
 """)))
 
     def test_filter_code_with_from(self):