Cleanup getLineCount

idea/testData/refactoring/extractFunction/controlFlow/evaluateExpression/evalWhenExpr.kt.after

@@ -8,10 +8,8 @@ fun foo(a: Int): Int {
     return i(a, b)
 }
 
-private fun i(a: Int, b: Int): Int {
-    return when (a + b) {
-        0 -> b
-        1 -> -b
-        else -> a - b
-    }
+private fun i(a: Int, b: Int) = when (a + b) {
+    0 -> b
+    1 -> -b
+    else -> a - b
 }

idea/testData/refactoring/extractFunction/parameters/candidateTypes/multipleTypes2.kt.after

@@ -18,11 +18,9 @@ fun foo(o: Any) {
     val x = i(o)
 }
 
-private fun i(o: Any): Int {
-    return when (o) {
-        is A -> {
-            if (o is T) o.a + o.t else o.a
-        }
-        else -> o.hashCode()
+private fun i(o: Any) = when (o) {
+    is A -> {
+        if (o is T) o.a + o.t else o.a
     }
+    else -> o.hashCode()
 }