File tree Expand file tree Collapse file tree 1 file changed +5
-0
lines changed
core/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 {
104
104
" shorts" to columnOf(1 .toShort(), 2 .toShort(), null ),
105
105
" bigInts" to columnOf(1 .toBigInteger(), 2 .toBigInteger(), null ),
106
106
" mixed" to columnOf<Number ?>(1.0 , 2 , null ),
107
+ " group" to columnOf(
108
+ " ints" to columnOf(1 , 2 , 3 ),
109
+ ),
107
110
)
108
111
109
112
val res = df.cumSum()
@@ -116,6 +119,8 @@ class CumsumTests {
116
119
res[" bigInts" ].values() shouldBe columnOf(1 .toBigInteger(), 2 .toBigInteger(), null ).values()
117
120
// works for mixed columns of primitives, number-unifies them; in this case to Doubles
118
121
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()
119
124
}
120
125
121
126
@Test
You can’t perform that action at this time.
0 commit comments