[python] Improve test coverage of ImportToImportFromIntention

GitOrigin-RevId: 59b2ebdfe4a8c9bbaf264f0db8b45a5af2b4533a

python/testData/intentions/ImportOfMultipleModulesToFromImportOfOnesAttributes/importOfMultipleModulesToFromImportOfOnesAttributes.py

@@ -0,0 +1,4 @@
+import <caret>mod, another
+
+mod.func()
+mod.Class()

python/testData/intentions/ImportOfMultipleModulesToFromImportOfOnesAttributes/importOfMultipleModulesToFromImportOfOnesAttributes_after.py

@@ -0,0 +1,5 @@
+import another
+from mod import func, Class
+
+func()
+Class()

python/testData/intentions/ImportOfMultipleModulesToFromImportOfOnesAttributes/mod.py

@@ -0,0 +1,5 @@
+class Class:
+    pass
+    
+def func():
+    pass

python/testData/intentions/ImportToFromImportKeepingOriginalImport/importToFromImportKeepingOriginalImport.py

@@ -0,0 +1,5 @@
+import <caret>mod
+
+print(mod)
+print(mod.func()
+print(mod.Class())

python/testData/intentions/ImportToFromImportKeepingOriginalImport/importToFromImportKeepingOriginalImport_after.py

@@ -0,0 +1,6 @@
+import mod
+from mod import func, Class
+
+print(mod)
+print(func()
+print(Class())

python/testData/intentions/ImportToFromImportKeepingOriginalImport/mod.py

@@ -0,0 +1,5 @@
+class Class:
+    pass
+    
+def func():
+    pass

python/testData/intentions/RelativeFromImportOfModuleItselfToRelativeFromImportOfItsAttributes/mod.py

@@ -0,0 +1,5 @@
+class Class:
+    pass
+    
+def func():
+    pass

python/testData/intentions/RelativeFromImportOfModuleItselfToRelativeFromImportOfItsAttributes/relativeFromImportOfModuleItselfToRelativeFromImportOfItsAttributes.py

@@ -0,0 +1,4 @@
+from . import <caret>mod
+
+mod.Class()
+mod.func()

python/testData/intentions/RelativeFromImportOfModuleItselfToRelativeFromImportOfItsAttributes/relativeFromImportOfModuleItselfToRelativeFromImportOfItsAttributes_after.py

@@ -0,0 +1,4 @@
+from .mod import func, Class
+
+Class()
+func()

python/testData/intentions/RelativeFromImportOfMultipleModulesToRelativeFromImportOfOnesAttributes/mod.py

@@ -0,0 +1,5 @@
+class Class:
+    pass
+    
+def func():
+    pass

python/testData/intentions/RelativeFromImportOfMultipleModulesToRelativeFromImportOfOnesAttributes/relativeFromImportOfMultipleModulesToRelativeFromImportOfOnesAttributes.py

@@ -0,0 +1,4 @@
+from . import <caret>mod, another
+
+mod.Class()
+mod.func()

python/testData/intentions/RelativeFromImportOfMultipleModulesToRelativeFromImportOfOnesAttributes/relativeFromImportOfMultipleModulesToRelativeFromImportOfOnesAttributes_after.py

@@ -0,0 +1,5 @@
+from . import another
+from .mod import func, Class
+
+Class()
+func()

python/testSrc/com/jetbrains/python/intentions/PyIntentionTest.java

@@ -473,6 +473,22 @@ public class PyIntentionTest extends PyTestCase {
     doMultiFileTest("Remove 'pkg.mod' qualifier");
   }
 
+  public void testRelativeFromImportOfModuleItselfToRelativeFromImportOfItsAttributes() {
+    doMultiFileTest("Convert to 'from .mod import ...'");
+  }
+
+  public void testRelativeFromImportOfMultipleModulesToRelativeFromImportOfOnesAttributes() {
+    doMultiFileTest("Convert to 'from .mod import ...'");
+  }
+
+  public void testImportOfMultipleModulesToFromImportOfOnesAttributes() {
+    doMultiFileTest("Convert to 'from mod import ...'");
+  }
+
+  public void testImportToFromImportKeepingOriginalImport() {
+    doMultiFileTest("Convert to 'from mod import ...'");
+  }
+
   public void testTypeInDocstring() {
     getIndentOptions().INDENT_SIZE = 2;
     doDocParamTypeTest(DocStringFormat.REST);