| 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("name", "age", "city", "weight")( | |
| "Alice", 15, "London", 54, | |
| "Bob", 45, "Dubai", 87, | |
| "Charlie", 20, "Moscow", null, | |
| "Charlie", 40, "Milan", null, | |
| "Bob", 30, "Tokyo", 68, | |
| "Alice", 20, null, 55, | |
| "Charlie", 30, "Moscow", 90, | |
| ) | |
| val df2 = df.move { name and city }.under("nameAndCity") | |
| val res = df2.convert { nameAndCity }.asFrame { it.remove { city } } | |
| res.compareSchemas(strict = true) | |
| return "OK" | |
| } |