allow wrapping in imports (PY-9163)

python/src/com/jetbrains/python/formatter/PyBlock.java

@@ -146,7 +146,8 @@ public class PyBlock implements ASTBlock {
         childIndent = Indent.getNormalIndent();
       }
     }
-    else if (childType == PyElementTypes.IMPORT_ELEMENT && hasLineBreaksBefore(child, 1)) {
+    else if (childType == PyElementTypes.IMPORT_ELEMENT) {
+      wrap = Wrap.createWrap(WrapType.NORMAL, true);
       childIndent = Indent.getNormalIndent();
     }
     if (ourListElementTypes.contains(parentType)) {

python/testData/formatter/wrapImports.py

@@ -0,0 +1 @@
+from common.resources.health import URLHealthCheck, JSONURLHealthCheck, HealthHandler, HealthResult

python/testData/formatter/wrapImports_after.py

@@ -0,0 +1,2 @@
+from common.resources.health import URLHealthCheck, JSONURLHealthCheck, \
+    HealthHandler, HealthResult

python/testSrc/com/jetbrains/python/PyFormatterTest.java

@@ -345,6 +345,11 @@ public class PyFormatterTest extends PyTestCase {
     doTest();
   }
 
+  public void testWrapImports() {  // PY-9163
+    settings().RIGHT_MARGIN = 80;
+    doTest();
+  }
+
   private void doTest() {
     doTest(false);
   }