File tree Expand file tree Collapse file tree 1 file changed +5
-0
lines changed
core/generated-sources/src/test/kotlin/org/jetbrains/kotlinx/dataframe/statistics Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Original file line number Diff line number Diff line change @@ -104,6 +104,9 @@ class CumsumTests {
104104 " shorts" to columnOf(1 .toShort(), 2 .toShort(), null ),
105105 " bigInts" to columnOf(1 .toBigInteger(), 2 .toBigInteger(), null ),
106106 " mixed" to columnOf<Number ?>(1.0 , 2 , null ),
107+ " group" to columnOf(
108+ " ints" to columnOf(1 , 2 , 3 ),
109+ ),
107110 )
108111
109112 val res = df.cumSum()
@@ -116,6 +119,8 @@ class CumsumTests {
116119 res[" bigInts" ].values() shouldBe columnOf(1 .toBigInteger(), 2 .toBigInteger(), null ).values()
117120 // works for mixed columns of primitives, number-unifies them; in this case to Doubles
118121 res[" mixed" ].values() shouldBe columnOf(1.0 , 3.0 , Double .NaN ).values()
122+ // runs at any depth
123+ res[" group" ][" ints" ].values() shouldBe columnOf(1 , 3 , 6 ).values()
119124 }
120125
121126 @Test
You can’t perform that action at this time.
0 commit comments