Denis Zharkov
5a1c995b5c
Generate TABLESWITCH/LOOKUPSWITCH bytecode command in case of "when" by enum entries
26 lines
548 B
Kotlin
26 lines
548 B
Kotlin
import kotlin.test.assertEquals
|
|
|
|
enum class Season {
|
|
WINTER
|
|
SPRING
|
|
SUMMER
|
|
AUTUMN
|
|
}
|
|
|
|
fun bar(x : Season) : String {
|
|
when (x) {
|
|
Season.WINTER, Season.SPRING -> return "winter_spring"
|
|
Season.SUMMER, Season.SPRING -> return "summer"
|
|
else -> return "autumn"
|
|
}
|
|
}
|
|
|
|
fun box() : String {
|
|
assertEquals("winter_spring", bar(Season.WINTER))
|
|
assertEquals("winter_spring", bar(Season.SPRING))
|
|
assertEquals("summer", bar(Season.SUMMER))
|
|
assertEquals("autumn", bar(Season.AUTUMN))
|
|
|
|
return "OK"
|
|
}
|