Skip to content

Commit b64791f

Browse files
committed
fixed tests
1 parent ea4d035 commit b64791f

File tree

1 file changed

+4
-4
lines changed
  • core/src/test/kotlin/org/jetbrains/kotlinx/dataframe/statistics

1 file changed

+4
-4
lines changed

core/src/test/kotlin/org/jetbrains/kotlinx/dataframe/statistics/cumsum.kt

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,14 +37,14 @@ class CumsumTests {
3737
val res = df.cumSum(skipNA = false)
3838

3939
res[col].toList() shouldBe expectedNoSkip
40-
res[col2].toList() shouldBe listOf(1, 3, 6, 10, 15)
41-
res[col3].toList() shouldBe listOf(1, 3, 6, 10, null)
40+
res[col2].toList() shouldBe listOf(1.toShort(), 3, 6, 10, 15)
41+
res[col3].toList() shouldBe listOf(1.toByte(), 3, 6, 10, null)
4242
}
4343

4444
@Test
4545
fun `byte column`() {
46-
col.map { it?.toByte() }.cumSum().toList() shouldBe expected
47-
col.map { it?.toByte() }.cumSum(skipNA = false).toList() shouldBe expectedNoSkip
46+
col.map { it?.toByte() }.cumSum().toList() shouldBe expected.map { it?.toByte() }
47+
col.map { it?.toByte() }.cumSum(skipNA = false).toList() shouldBe expectedNoSkip.map { it?.toByte() }
4848
}
4949

5050
@Test

0 commit comments

Comments
 (0)