fixed PY-5198 Move statement: breaks code in case of one-line compound statements on module level

python/src/com/jetbrains/python/codeInsight/editorActions/moveUpDown/StatementMover.java

@@ -325,6 +325,9 @@ public class StatementMover extends LineMover {
       
       PsiElement whiteSpace = myStatementToAddLinebreak.getContainingFile().findElementAt(editor.getDocument().getLineStartOffset(line));
       String indent = StringUtil.repeatSymbol(' ', indentOptions.INDENT_SIZE);
+
+      PyStatementWithElse statementWithElse = PsiTreeUtil.getParentOfType(myStatementToAddLinebreak, PyStatementWithElse.class);
+      if (statementWithElse != null && statementWithElse.getParent() instanceof PyFile) indent = "\n";
       if (whiteSpace instanceof PsiWhiteSpace) indent += whiteSpace.getText();
       if (down) indent += StringUtil.repeatSymbol(' ', indentOptions.INDENT_SIZE);
       editor.getDocument().insertString(textRange.getStartOffset(), indent);

python/testData/mover/oneLineCompound.py

@@ -0,0 +1,3 @@
+c = 1
+if True: a = 1
+b =<caret> 2

python/testData/mover/oneLineCompound_afterDown.py

@@ -0,0 +1,3 @@
+c = 1
+if True: a = 1
+b =<caret> 2   

python/testData/mover/oneLineCompound_afterUp.py

@@ -0,0 +1,4 @@
+c = 1
+if True:
+    a = 1
+    b = 2

python/testSrc/com/jetbrains/python/PyStatementMoverTest.java

@@ -107,4 +107,8 @@ public class PyStatementMoverTest extends PyTestCase {
   public void testClass() { // PY-5196
     doTest();
   }
+
+  public void testOneLineCompound() { // PY-5198
+    doTest();
+  }
 }