Support subject variable in specialized code generators for 'when'

2018-06-09 17:00:25 +03:00
parent 3528405666
commit 34b76a3718
17 changed files with 427 additions and 20 deletions
@@ -0,0 +1,26 @@
+// !LANGUAGE: +VariableDeclarationInWhenSubject
+// WITH_RUNTIME
+// IGNORE_BACKEND: JS
+
+fun dense(x: Int): Int {
+    return when (val xx = x) {
+        -4 -> 9
+        -1 -> 10
+        0 -> x + 11
+        1 -> 12
+        4 -> 13
+        5 -> 14
+        6 -> 15
+        7 -> 16
+        8 -> 17
+        9 -> x + 9
+        else -> 19
+    }
+}
+
+fun box(): String {
+    var result = (-5..10).map(::dense).joinToString()
+
+    if (result != "19, 9, 19, 19, 10, 11, 12, 19, 19, 13, 14, 15, 16, 17, 18, 19") return "dense:" + result
+    return "OK"
+}
@@ -0,0 +1,19 @@
+// !LANGUAGE: +VariableDeclarationInWhenSubject
+// WITH_RUNTIME
+// IGNORE_BACKEND: JS
+
+fun sparse(x: Int): Int {
+    return when (val xx = (x % 4) * 100) {
+        100 -> 1
+        200 -> xx / 100
+        300 -> 3
+        else -> 4
+    }
+}
+
+fun box(): String {
+    var result = (0..3).map(::sparse).joinToString()
+
+    if (result != "4, 1, 2, 3") return "sparse:" + result
+    return "OK"
+}
@@ -0,0 +1,41 @@
+// !LANGUAGE: +VariableDeclarationInWhenSubject
+// WITH_RUNTIME
+// IGNORE_BACKEND: JS
+
+import kotlin.test.assertEquals
+
+enum class Season {
+    WINTER,
+    SPRING,
+    SUMMER,
+    AUTUMN
+}
+
+fun bar1(x : Season) : String {
+    return when (val xx = x) {
+        Season.WINTER, Season.SPRING -> "winter_spring $xx"
+        Season.SUMMER -> "summer"
+        else -> "autumn"
+    }
+}
+
+fun bar2(x : Season) : String {
+    return when (val xx = x) {
+        Season.WINTER, Season.SPRING -> "winter_spring $xx"
+        Season.SUMMER -> "summer"
+        Season.AUTUMN -> "autumn"
+    }
+}
+
+fun box() : String {
+    assertEquals("winter_spring WINTER", bar1(Season.WINTER))
+    assertEquals("winter_spring SPRING", bar1(Season.SPRING))
+    assertEquals("summer", bar1(Season.SUMMER))
+    assertEquals("autumn", bar1(Season.AUTUMN))
+
+    assertEquals("winter_spring WINTER", bar2(Season.WINTER))
+    assertEquals("winter_spring SPRING", bar2(Season.SPRING))
+    assertEquals("summer", bar2(Season.SUMMER))
+    assertEquals("autumn", bar2(Season.AUTUMN))
+    return "OK"
+}
@@ -0,0 +1,44 @@
+// !LANGUAGE: +VariableDeclarationInWhenSubject
+// WITH_RUNTIME
+// IGNORE_BACKEND: JS
+
+import kotlin.test.assertEquals
+
+enum class Season {
+    WINTER,
+    SPRING,
+    SUMMER,
+    AUTUMN
+}
+
+fun foo1(x : Season?) : String {
+    when(val xx = x) {
+        Season.AUTUMN, Season.SPRING -> return "autumn_or_spring $xx";
+        Season.SUMMER, null -> return "summer_or_null $xx"
+    }
+
+    return "other"
+}
+
+fun foo2(x : Season?) : String {
+    when(val xx = x) {
+        Season.AUTUMN, Season.SPRING -> return "autumn_or_spring $xx";
+        Season.SUMMER -> return "summer"
+    }
+
+    return "other"
+}
+
+fun box() : String {
+    assertEquals("autumn_or_spring AUTUMN", foo1(Season.AUTUMN))
+    assertEquals("autumn_or_spring SPRING", foo1(Season.SPRING))
+    assertEquals("summer_or_null SUMMER", foo1(Season.SUMMER))
+    assertEquals("summer_or_null null", foo1(null))
+
+    assertEquals("autumn_or_spring AUTUMN", foo2(Season.AUTUMN))
+    assertEquals("autumn_or_spring SPRING", foo2(Season.SPRING))
+    assertEquals("summer", foo2(Season.SUMMER))
+    assertEquals("other", foo2(null))
+
+    return "OK"
+}
@@ -0,0 +1,24 @@
+// !LANGUAGE: +VariableDeclarationInWhenSubject
+// WITH_RUNTIME
+// IGNORE_BACKEND: JS
+
+import kotlin.test.assertEquals
+
+fun foo(x : String) : String {
+    return when (val xx = x) {
+        "abc", "cde" -> "1 $xx"
+        "efg", "ghi" -> "2 $xx"
+        else -> "other $xx"
+    }
+}
+
+fun box() : String {
+    assertEquals("1 abc", foo("abc"))
+    assertEquals("1 cde", foo("cde"))
+    assertEquals("2 efg", foo("efg"))
+    assertEquals("2 ghi", foo("ghi"))
+
+    assertEquals("other xyz", foo("xyz"))
+
+    return "OK"
+}