fix incorrect path calculation in auto-import

python/src/com/jetbrains/python/codeInsight/imports/PythonReferenceImporter.java

@@ -112,7 +112,7 @@ public class PythonReferenceImporter implements ReferenceImporter {
         if (isTopLevel(symbol)) { // we only want top-level symbols
           PsiFileSystemItem srcfile = symbol instanceof PsiFileSystemItem ? ((PsiFileSystemItem)symbol).getParent() : symbol.getContainingFile();
           if (srcfile != null && srcfile != existing_import_file && srcfile != node.getContainingFile()) {
-            PyQualifiedName import_path = ResolveImportUtil.findCanonicalImportPath(symbol, node);
+            PyQualifiedName import_path = ResolveImportUtil.findCanonicalImportPath(srcfile, node);
             if (import_path != null && !seen_file_names.contains(import_path.toString())) {
               // a new, valid hit
               fix.addImport(symbol, srcfile, import_path, proposeAsName(node.getContainingFile(), ref_text, import_path));

python/testData/inspections/importFromModule/foo/bar.py

@@ -0,0 +1 @@
+<warning descr="Unresolved reference 'baz'">baz</warning>.quux()

python/testData/inspections/importFromModule/foo/bar_after.py

@@ -0,0 +1,3 @@
+from importFromModule.foo import baz
+
+baz.quux()

python/testData/inspections/importFromModule/foo/baz.py

@@ -0,0 +1 @@
+def quux(): pass

python/testSrc/com/jetbrains/python/PyQuickFixTest.java

@@ -31,6 +31,11 @@ public class PyQuickFixTest extends PyLightFixtureTestCase {
     doInspectionTest(new String[] { "AddImportDocComment.py", "ImportTarget.py" }, PyUnresolvedReferencesInspection.class, PyBundle.message("ACT.NAME.use.import"), true, true);
   }
 
+  public void testImportFromModule() {
+    doInspectionTest(new String[] { "importFromModule/foo/bar.py", "importFromModule/foo/baz.py", "importFromModule/foo/__init__.py" },
+                     PyUnresolvedReferencesInspection.class, PyBundle.message("ACT.NAME.use.import"), true, true);
+  }
+
   public void testQualifyByImport() {
     final PyCodeInsightSettings settings = PyCodeInsightSettings.getInstance();
     boolean oldPreferFrom = settings.PREFER_FROM_IMPORT;