more test for DataFrame API

3cham · May 3, 2024 · 55f7e7e · 55f7e7e
1 parent d04857e
commit 55f7e7e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -3,4 +3,4 @@
 
 Implementation of ["How query engines work?"](https://howqueryengineswork.com/) for learning queries engine & kotlin.
 
-Progress: ![](https://geps.dev/progress/32)
+Progress: ![](https://geps.dev/progress/36)
diff --git a/logical-plan/src/test/kotlin/DataFrameTest.kt b/logical-plan/src/test/kotlin/DataFrameTest.kt
@@ -54,6 +54,21 @@ class DataFrameTest {
         assertEquals(expected, actual)
     }
 
+    @Test
+    fun `aggregate query`() {
+
+        val df =
+            csv()
+                .aggregate(
+                    listOf(col("state")),
+                    listOf(Min(col("salary")), Max(col("salary")), Count(col("salary"))))
+
+        assertEquals(
+            "Aggregate: groupExpr=[#state], aggregateExpr=[MIN(#salary), MAX(#salary), COUNT(#salary)]\n" +
+                    "\tScan: employee.csv; projection=None\n",
+            format(df.logicalPlan()))
+    }
+
     private fun csv(): DataFrame {
         val csvSource = CsvDataSource(
             File("../testdata", "employee.csv").absolutePath,