| import org.jetbrains.kotlinx.dataframe.* |
| import org.jetbrains.kotlinx.dataframe.annotations.* |
| import org.jetbrains.kotlinx.dataframe.api.* |
| import org.jetbrains.kotlinx.dataframe.io.* |
| |
| fun box(): String { |
| val df = dataFrameOf("a", "b", "c")(1, 2, 3, 4, 5, 6) |
| df.move { c and a }.toEnd().let { |
| it.checkCompileTimeSchemaEqualsRuntime() |
| } |
| df.moveToEnd { c and a }.let { |
| it.checkCompileTimeSchemaEqualsRuntime() |
| } |
| |
| val dfGrouped = df.group { a and b }.into("d") |
| |
| dfGrouped.move { d.b }.toEnd(insideGroup = true).let { |
| it.checkCompileTimeSchemaEqualsRuntime() |
| it.c |
| it.d |
| it.d.b |
| it.d.a |
| } |
| dfGrouped.moveToEnd(insideGroup = true) { d.b }.let { |
| it.checkCompileTimeSchemaEqualsRuntime() |
| it.c |
| it.d |
| it.d.b |
| it.d.a |
| } |
| |
| dfGrouped.move { d.b }.toEnd().let { |
| it.checkCompileTimeSchemaEqualsRuntime() |
| it.c |
| it.d |
| it.b |
| it.d.a |
| } |
| dfGrouped.moveToEnd { d.b }.let { |
| it.checkCompileTimeSchemaEqualsRuntime() |
| it.c |
| it.d |
| it.b |
| it.d.a |
| } |
| |
| return "OK" |
| } |