from __future__ imports aren't unused

This commit is contained in:
Dmitry Jemerov
2010-07-01 21:53:46 +04:00
parent 78894d849b
commit 9f0f1b8f7d
4 changed files with 10 additions and 1 deletions
@@ -247,7 +247,10 @@ public class PyUnresolvedReferencesInspection extends PyInspection {
@Override
public void visitPyImportElement(PyImportElement node) {
super.visitPyImportElement(node);
myAllImports.add(node);
final PyFromImportStatement fromImport = PsiTreeUtil.getParentOfType(node, PyFromImportStatement.class);
if (fromImport == null || !fromImport.isFromFuture()) {
myAllImports.add(node);
}
}
@Override
@@ -0,0 +1 @@
from __future__ import generators
@@ -0,0 +1 @@
from __future__ import generators
@@ -28,6 +28,10 @@ public class PyOptimizeImportsTest extends PyLightFixtureTestCase {
doTest();
}
public void testFromFuture() throws Exception {
doTest();
}
private void doTest() throws Exception {
myFixture.configureByFile("optimizeImports/" + getTestName(true) + ".py");
OptimizeImportsAction.actionPerformedImpl(DataManager.getInstance().getDataContext(myFixture.getEditor().getContentComponent()));