Remove unused aliased imports

Wilfred · Wilfred · commit e40d3faf5b74 · 2017-01-17T14:15:40.000Z
diff --git a/pyimport.el b/pyimport.el
@@ -247,6 +247,7 @@ on line number LINE, remove VAR (e.g. 'baz')."
             (pyimport--remove-on-line var))
         ;; If we only have "from foo import " left, remove the rest of the line.
         (when (or (looking-at (rx "from " (1+ (not (any space))) " import" (1+ space) line-end))
+                  (looking-at (rx "from " (1+ (not (any space))) " import " (1+ (not (any space))) " as" (1+ space) line-end))
                   (looking-at (rx "import " (1+ (not (any space))) " as" (1+ space) line-end)))
           (pyimport--delete-current-line)))))))
 
diff --git a/test/pyimport-test.el b/test/pyimport-test.el
@@ -32,6 +32,14 @@
     (should
      (equal (buffer-string) ""))))
 
+(ert-deftest remove-import-as ()
+  "Ensure we remove imports correctly when there are aliases."
+  (with-temp-buffer
+    (insert "from foo import bar as newname")
+    (pyimport--remove-import 1 "newname")
+    (should
+     (equal (buffer-string) ""))))
+
 (ert-deftest remove-on-line-first ()
   "We should remove the first occurrence, if present."
   (with-temp-buffer
