Add test case and update code

autoflake.py

@@ -515,11 +515,13 @@ def get_imported_name(module):
     - `a as b` -> b
     """
     if '.' in module:
-        return str(module.split('.')[-1])
-    elif ' as ' in module:
-        return str(module.split(' as ')[-1])
+        name = module.split('.')[-1]
+    elif re.search(r'\bas\b', module):
+        name = re.split(r'\bas\b', module)[-1]
+    else:
+        name = module
     # str() to force python 2 to not use unicode
-    return str(module)
+    return str(name.strip())
 
 def get_indentation(line):
     """Return leading whitespace."""

test_autoflake.py

@@ -526,6 +526,22 @@ def test_filter_code_populate_all_from_import(self):
 """, ''.join(autoflake.filter_code("""
 from a.b import Foo
 from a.c import Bar
+""", populate_all=True)))
+
+    def test_filter_code_populate_all_as(self):
+        self.assertEqual("""
+import math as m
+__all__ = ['m']
+""", ''.join(autoflake.filter_code("""
+import math as m
+""", populate_all=True)))
+
+    def test_filter_code_populate_all_with_tab(self):
+        self.assertEqual("""
+import math\tas\tm
+__all__ = ['m']
+""", ''.join(autoflake.filter_code("""
+import math\tas\tm
 """, populate_all=True)))
 
     def test_fix_code(self):