Skip to content

Commit f31d25c

Browse files
committed
Use idiomatic dataframe API in calculate sum of all visits test.
1 parent cd200e8 commit f31d25c

File tree

1 file changed

+5
-1
lines changed
  • src/test/kotlin/org/jetbrains/kotlinx/dataframe/puzzles

1 file changed

+5
-1
lines changed

src/test/kotlin/org/jetbrains/kotlinx/dataframe/puzzles/BasicTests.kt

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,11 @@ class BasicTests {
112112
@Test
113113
fun `calculate sum of all visits`() {
114114
df[visits].sum() shouldBe 19
115-
df[ { "visits"<Int>() }].sum() shouldBe 19
115+
df.sum { visits } shouldBe 19
116+
117+
df["visits"].cast<Int>().sum() shouldBe 19
118+
df.sum { "visits"<Int>() } shouldBe 19
119+
df.sum("visits") shouldBe 19
116120
}
117121

118122
@Test

0 commit comments

Comments
 (0)