Add examples for rename

Kotlin · Nov 14, 2024 · 10baf11 · 10baf11
1 parent 3f054ab
commit 10baf11
Show file tree

Hide file tree

Showing 4 changed files with 1,103 additions and 0 deletions.
diff --git a/core/src/test/kotlin/org/jetbrains/kotlinx/dataframe/samples/api/Modify.kt b/core/src/test/kotlin/org/jetbrains/kotlinx/dataframe/samples/api/Modify.kt
@@ -1464,4 +1464,63 @@ class Modify : TestBase() {
         }
         // SampleEnd
     }
+
+    @Test
+    @TransformDataFrameExpressions
+    fun rename_properties() {
+        // SampleStart
+        df.rename { name }.into("fullName")
+        // SampleEnd
+    }
+
+    @Test
+    @TransformDataFrameExpressions
+    fun rename_accessors() {
+        // SampleStart
+        val name by columnGroup()
+        df.rename { name }.into("fullName")
+        // SampleEnd
+    }
+
+    @Test
+    @TransformDataFrameExpressions
+    fun rename_strings() {
+        // SampleStart
+        df.rename("name").into("fullName")
+        // SampleEnd
+    }
+
+    @Test
+    @TransformDataFrameExpressions
+    fun renameExpression_properties() {
+        // SampleStart
+        df.rename { age }.into {
+            val mean = it.data.mean()
+            "age [mean = $mean]"
+        }
+        // SampleEnd
+    }
+
+    @Test
+    @TransformDataFrameExpressions
+    fun renameExpression_accessors() {
+        // SampleStart
+        val age by column<Int>()
+        df.rename { age }.into {
+            val mean = it.data.mean()
+            "age [mean = $mean]"
+        }
+        // SampleEnd
+    }
+
+    @Test
+    @TransformDataFrameExpressions
+    fun renameExpression_strings() {
+        // SampleStart
+        df.rename("age").into {
+            val mean = it.data.cast<Int>().mean()
+            "age [mean = $mean]"
+        }
+        // SampleEnd
+    }
 }