Skip to content

Commit de9e1f0

Browse files
JolanrensenJolanrensen
committed
trying to get samples to compiler with compiler plugin
1 parent 1c67663 commit de9e1f0

File tree

7 files changed

+390
-467
lines changed

7 files changed

+390
-467
lines changed

‎samples/src/test/kotlin/org/jetbrains/kotlinx/dataframe/samples/api/Generate.kt‎

Lines changed: 109 additions & 183 deletions
Original file line numberDiff line numberDiff line change
@@ -1,183 +1,109 @@
1-
@file:Suppress("UNUSED_VARIABLE", "unused", "UNCHECKED_CAST", "ktlint", "ClassName")
2-
3-
package org.jetbrains.kotlinx.dataframe.samples.api
4-
5-
import org.jetbrains.kotlinx.dataframe.annotations.DataSchema
6-
import org.jetbrains.kotlinx.dataframe.api.add
7-
import org.jetbrains.kotlinx.dataframe.api.all
8-
import org.jetbrains.kotlinx.dataframe.api.cast
9-
import org.jetbrains.kotlinx.dataframe.api.dataFrameOf
10-
import org.jetbrains.kotlinx.dataframe.api.filter
11-
import org.jetbrains.kotlinx.dataframe.api.generateCode
12-
import org.jetbrains.kotlinx.dataframe.api.generateDataClasses
13-
import org.jetbrains.kotlinx.dataframe.api.generateInterfaces
14-
import org.jetbrains.kotlinx.dataframe.api.into
15-
import org.jetbrains.kotlinx.dataframe.api.rename
16-
import org.jetbrains.kotlinx.dataframe.api.sumOf
17-
import org.jetbrains.kotlinx.dataframe.api.toList
18-
import org.jetbrains.kotlinx.dataframe.samples.DataFrameSampleHelper
19-
import org.junit.Test
20-
21-
class Generate : DataFrameSampleHelper("generate_docs", "api") {
22-
val ordersAlice = dataFrameOf(
23-
"orderId" to listOf(101, 102),
24-
"amount" to listOf(50.0, 75.5),
25-
)
26-
27-
val ordersBob = dataFrameOf(
28-
"orderId" to listOf(103, 104, 105),
29-
"amount" to listOf(20.0, 30.0, 25.0),
30-
)
31-
32-
val df = dataFrameOf(
33-
"user" to listOf("Alice", "Bob"),
34-
"orders" to listOf(ordersAlice, ordersBob),
35-
)
36-
37-
@DataSchema(isOpen = false)
38-
interface _DataFrameType11 {
39-
val amount: kotlin.Double
40-
val orderId: kotlin.Int
41-
}
42-
43-
val org.jetbrains.kotlinx.dataframe.ColumnsContainer<_DataFrameType11>.amount: org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.Double>
44-
@JvmName(
45-
"_DataFrameType11_amount",
46-
)
47-
get() = this["amount"] as org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.Double>
48-
val org.jetbrains.kotlinx.dataframe.DataRow<_DataFrameType11>.amount: kotlin.Double
49-
@JvmName("_DataFrameType11_amount")
50-
get() = this["amount"] as kotlin.Double
51-
val org.jetbrains.kotlinx.dataframe.ColumnsContainer<_DataFrameType11>.orderId: org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.Int>
52-
@JvmName(
53-
"_DataFrameType11_orderId",
54-
)
55-
get() = this["orderId"] as org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.Int>
56-
val org.jetbrains.kotlinx.dataframe.DataRow<_DataFrameType11>.orderId: kotlin.Int
57-
@JvmName("_DataFrameType11_orderId")
58-
get() = this["orderId"] as kotlin.Int
59-
60-
@DataSchema
61-
interface _DataFrameType1 {
62-
val orders: List<_DataFrameType11>
63-
val user: kotlin.String
64-
}
65-
66-
val org.jetbrains.kotlinx.dataframe.ColumnsContainer<_DataFrameType1>.orders: org.jetbrains.kotlinx.dataframe.DataColumn<org.jetbrains.kotlinx.dataframe.DataFrame<_DataFrameType11>>
67-
@JvmName(
68-
"_DataFrameType1_orders",
69-
)
70-
get() = this["orders"] as org.jetbrains.kotlinx.dataframe.DataColumn<org.jetbrains.kotlinx.dataframe.DataFrame<_DataFrameType11>>
71-
val org.jetbrains.kotlinx.dataframe.DataRow<_DataFrameType1>.orders: org.jetbrains.kotlinx.dataframe.DataFrame<_DataFrameType11>
72-
@JvmName(
73-
"_DataFrameType1_orders",
74-
)
75-
get() = this["orders"] as org.jetbrains.kotlinx.dataframe.DataFrame<_DataFrameType11>
76-
val org.jetbrains.kotlinx.dataframe.ColumnsContainer<_DataFrameType1>.user: org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.String>
77-
@JvmName(
78-
"_DataFrameType1_user",
79-
)
80-
get() = this["user"] as org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.String>
81-
val org.jetbrains.kotlinx.dataframe.DataRow<_DataFrameType1>.user: kotlin.String
82-
@JvmName("_DataFrameType1_user")
83-
get() = this["user"] as kotlin.String
84-
85-
@DataSchema
86-
data class Customer1(val amount: Double, val orderId: Int)
87-
88-
@DataSchema
89-
data class Customer(val orders: List<Customer1>, val user: String)
90-
91-
val org.jetbrains.kotlinx.dataframe.ColumnsContainer<Customer1>.amount: org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.Double>
92-
@JvmName(
93-
"Customer1_amount",
94-
)
95-
get() = this["amount"] as org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.Double>
96-
val org.jetbrains.kotlinx.dataframe.DataRow<Customer1>.amount: kotlin.Double
97-
@JvmName("Customer1_amount")
98-
get() = this["amount"] as kotlin.Double
99-
val org.jetbrains.kotlinx.dataframe.ColumnsContainer<Customer1>.orderId: org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.Int>
100-
@JvmName(
101-
"Customer1_orderId",
102-
)
103-
get() = this["orderId"] as org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.Int>
104-
val org.jetbrains.kotlinx.dataframe.DataRow<Customer1>.orderId: kotlin.Int
105-
@JvmName("Customer1_orderId")
106-
get() = this["orderId"] as kotlin.Int
107-
108-
val org.jetbrains.kotlinx.dataframe.ColumnsContainer<Customer>.orders: org.jetbrains.kotlinx.dataframe.DataColumn<org.jetbrains.kotlinx.dataframe.DataFrame<Customer1>>
109-
@JvmName(
110-
"Customer_orders",
111-
)
112-
get() = this["orders"] as org.jetbrains.kotlinx.dataframe.DataColumn<org.jetbrains.kotlinx.dataframe.DataFrame<Customer1>>
113-
val org.jetbrains.kotlinx.dataframe.DataRow<Customer>.orders: org.jetbrains.kotlinx.dataframe.DataFrame<Customer1>
114-
@JvmName(
115-
"Customer_orders",
116-
)
117-
get() = this["orders"] as org.jetbrains.kotlinx.dataframe.DataFrame<Customer1>
118-
val org.jetbrains.kotlinx.dataframe.ColumnsContainer<Customer>.user: org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.String>
119-
@JvmName(
120-
"Customer_user",
121-
)
122-
get() = this["user"] as org.jetbrains.kotlinx.dataframe.DataColumn<kotlin.String>
123-
val org.jetbrains.kotlinx.dataframe.DataRow<Customer>.user: kotlin.String
124-
@JvmName("Customer_user")
125-
get() = this["user"] as kotlin.String
126-
127-
private val customers: List<Customer> = df.cast<Customer>().toList()
128-
129-
@Test
130-
fun notebook_test_generate_docs_1() {
131-
// SampleStart
132-
df
133-
// SampleEnd
134-
.saveDfHtmlSample()
135-
}
136-
137-
@Test
138-
fun notebook_test_generate_docs_2() {
139-
// SampleStart
140-
df.generateInterfaces()
141-
// SampleEnd
142-
}
143-
144-
@Test
145-
fun notebook_test_generate_docs_3() {
146-
// SampleStart
147-
df.cast<_DataFrameType1>().filter { orders.all { orderId >= 102 } }
148-
// SampleEnd
149-
// .saveDfHtmlSample()
150-
}
151-
152-
@Test
153-
fun notebook_test_generate_docs_4() {
154-
// SampleStart
155-
df.generateDataClasses("Customer")
156-
// SampleEnd
157-
}
158-
159-
@Test
160-
fun notebook_test_generate_docs_5() {
161-
// SampleStart
162-
val customers: List<Customer> = df.cast<Customer>().toList()
163-
// SampleEnd
164-
}
165-
166-
@Test
167-
fun notebook_test_generate_docs_6() {
168-
// SampleStart
169-
df.generateCode("Customer")
170-
// SampleEnd
171-
}
172-
173-
@Test
174-
fun notebook_test_generate_docs_7() {
175-
// SampleStart
176-
df.cast<Customer>()
177-
.add("ordersTotal") { orders.sumOf { it.amount } }
178-
.filter { user.startsWith("A") }
179-
.rename { user }.into("customer")
180-
// SampleEnd
181-
// .saveDfHtmlSample()
182-
}
183-
}
1+
//@file:Suppress("UNUSED_VARIABLE", "unused", "UNCHECKED_CAST", "ktlint", "ClassName")
2+
//
3+
//package org.jetbrains.kotlinx.dataframe.samples.api
4+
//
5+
//import org.jetbrains.kotlinx.dataframe.annotations.DataSchema
6+
//import org.jetbrains.kotlinx.dataframe.api.add
7+
//import org.jetbrains.kotlinx.dataframe.api.all
8+
//import org.jetbrains.kotlinx.dataframe.api.cast
9+
//import org.jetbrains.kotlinx.dataframe.api.dataFrameOf
10+
//import org.jetbrains.kotlinx.dataframe.api.filter
11+
//import org.jetbrains.kotlinx.dataframe.api.generateCode
12+
//import org.jetbrains.kotlinx.dataframe.api.generateDataClasses
13+
//import org.jetbrains.kotlinx.dataframe.api.generateInterfaces
14+
//import org.jetbrains.kotlinx.dataframe.api.into
15+
//import org.jetbrains.kotlinx.dataframe.api.rename
16+
//import org.jetbrains.kotlinx.dataframe.api.sumOf
17+
//import org.jetbrains.kotlinx.dataframe.api.toList
18+
//import org.jetbrains.kotlinx.dataframe.samples.DataFrameSampleHelper
19+
//import org.junit.Test
20+
//
21+
//class Generate : DataFrameSampleHelper("generate_docs", "api") {
22+
// private val ordersAlice = dataFrameOf(
23+
// "orderId" to listOf(101, 102),
24+
// "amount" to listOf(50.0, 75.5),
25+
// )
26+
//
27+
// private val ordersBob = dataFrameOf(
28+
// "orderId" to listOf(103, 104, 105),
29+
// "amount" to listOf(20.0, 30.0, 25.0),
30+
// )
31+
//
32+
// private val df = dataFrameOf(
33+
// "user" to listOf("Alice", "Bob"),
34+
// "orders" to listOf(ordersAlice, ordersBob),
35+
// )
36+
//
37+
// @DataSchema(isOpen = false)
38+
// interface _DataFrameType11 {
39+
// val amount: kotlin.Double
40+
// val orderId: kotlin.Int
41+
// }
42+
//
43+
// @DataSchema
44+
// interface _DataFrameType1 {
45+
// val orders: List<_DataFrameType11>
46+
// val user: kotlin.String
47+
// }
48+
//
49+
// @DataSchema
50+
// data class Customer1(val amount: Double, val orderId: Int)
51+
//
52+
// @DataSchema
53+
// data class Customer(val orders: List<Customer1>, val user: String)
54+
//
55+
// @Test
56+
// fun notebook_test_generate_docs_1() {
57+
// // SampleStart
58+
// df
59+
// // SampleEnd
60+
// .saveDfHtmlSample()
61+
// }
62+
//
63+
// @Test
64+
// fun notebook_test_generate_docs_2() {
65+
// // SampleStart
66+
// df.generateInterfaces()
67+
// // SampleEnd
68+
// }
69+
//
70+
// @Test
71+
// fun notebook_test_generate_docs_3() {
72+
// // SampleStart
73+
// df.filter { orders.all { orderId >= 102 } }
74+
// // SampleEnd
75+
// // .saveDfHtmlSample()
76+
// }
77+
//
78+
// @Test
79+
// fun notebook_test_generate_docs_4() {
80+
// // SampleStart
81+
// df.generateDataClasses("Customer")
82+
// // SampleEnd
83+
// }
84+
//
85+
// @Test
86+
// fun notebook_test_generate_docs_5() {
87+
// // SampleStart
88+
// val customers: List<Customer> = df.cast<Customer>().toList()
89+
// // SampleEnd
90+
// }
91+
//
92+
// @Test
93+
// fun notebook_test_generate_docs_6() {
94+
// // SampleStart
95+
// df.generateCode("Customer")
96+
// // SampleEnd
97+
// }
98+
//
99+
// @Test
100+
// fun notebook_test_generate_docs_7() {
101+
// // SampleStart
102+
// df.cast<Customer>()
103+
// .add("ordersTotal") { orders.sumOf { it.amount } }
104+
// .filter { user.startsWith("A") }
105+
// .rename { user }.into("customer")
106+
// // SampleEnd
107+
// // .saveDfHtmlSample()
108+
// }
109+
//}

0 commit comments

Comments
 (0)