33package org.jetbrains.kotlinx.dataframe.samples.api
44
55import org.jetbrains.kotlinx.dataframe.DataFrame
6- import org.jetbrains.kotlinx.dataframe.annotations.ColumnName
76import org.jetbrains.kotlinx.dataframe.annotations.DataSchema
87import org.jetbrains.kotlinx.dataframe.api.add
98import org.jetbrains.kotlinx.dataframe.api.cast
10- import org.jetbrains.kotlinx.dataframe.api.column
119import org.jetbrains.kotlinx.dataframe.api.dropNulls
1210import org.jetbrains.kotlinx.dataframe.api.filter
13- import org.jetbrains.kotlinx.dataframe.api.toListOf
1411import org.jetbrains.kotlinx.dataframe.explainer.TransformDataFrameExpressions
1512import org.jetbrains.kotlinx.dataframe.io.read
1613import org.junit.Ignore
@@ -34,91 +31,6 @@ class ApiLevels {
3431 // SampleEnd
3532 }
3633
37- @Test
38- @TransformDataFrameExpressions
39- fun accessors1 () {
40- // SampleStart
41- val survived by column<Boolean >() // accessor for Boolean column with name 'survived'
42- val home by column<String >()
43- val age by column<Int ?>()
44- val name by column<String >()
45- val lastName by column<String >()
46- // SampleEnd
47- }
48-
49- @Test
50- @TransformDataFrameExpressions
51- fun accessors2 () {
52- val survived by column<Boolean >()
53- val home by column<String >()
54- val age by column<Int ?>()
55- val name by column<String >()
56- val lastName by column<String >()
57- // SampleStart
58-
59- DataFrame .read(" titanic.csv" )
60- .add(lastName) { name().split(" ," ).last() }
61- .dropNulls { age }
62- .filter { survived() && home().endsWith(" NY" ) && age()!! in 10 .. 20 }
63- // SampleEnd
64- }
65-
66- @Test
67- @TransformDataFrameExpressions
68- fun accessors3 () {
69- // SampleStart
70- val survived by column<Boolean >()
71- val home by column<String >()
72- val age by column<Int ?>()
73- val name by column<String >()
74- val lastName by column<String >()
75-
76- DataFrame .read(" titanic.csv" )
77- .add(lastName) { name().split(" ," ).last() }
78- .dropNulls { age }
79- .filter { survived() && home().endsWith(" NY" ) && age()!! in 10 .. 20 }
80- // SampleEnd
81- }
82-
83- @Test
84- @TransformDataFrameExpressions
85- fun kproperties1 () {
86- // SampleStart
87- data class Passenger (
88- val survived : Boolean ,
89- val home : String ,
90- val age : Int ,
91- val lastName : String
92- )
93-
94- val passengers = DataFrame .read(" titanic.csv" )
95- .add(Passenger ::lastName) { " name" <String >().split(" ," ).last() }
96- .dropNulls(Passenger ::age)
97- .filter {
98- it[Passenger ::survived] &&
99- it[Passenger ::home].endsWith(" NY" ) &&
100- it[Passenger ::age] in 10 .. 20
101- }
102- .toListOf<Passenger >()
103- // SampleEnd
104- }
105-
106- @Test
107- @TransformDataFrameExpressions
108- fun kproperties2 () {
109- // SampleStart
110- data class Passenger (
111- @ColumnName(" survived" ) val isAlive : Boolean ,
112- @ColumnName(" home" ) val city : String ,
113- val name : String
114- )
115-
116- val passengers = DataFrame .read(" titanic.csv" )
117- .filter { it[Passenger ::city].endsWith(" NY" ) }
118- .toListOf<Passenger >()
119- // SampleEnd
120- }
121-
12234 @DataSchema
12335 interface TitanicPassenger {
12436 val survived: Boolean
0 commit comments