Expand tests

PyCQA · Apr 11, 2017 · 71fa850 · 71fa850
1 parent 16bac0a
commit 71fa850
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test_autoflake.py b/test_autoflake.py
@@ -179,6 +179,18 @@ def test_filter_code_with_from(self):
 x = 1
 """)))
 
+    def test_filter_code_with_not_from(self):
+        self.assertEqual(
+            """\
+pass
+x = 1
+""",
+            ''.join(autoflake.filter_code("""\
+import frommer
+x = 1
+""",
+                                          remove_all_unused_imports=True)))
+
     def test_filter_code_should_avoid_inline_except(self):
         line = """\
 try: from zap import foo