@@ -26,6 +26,8 @@ class JupyterCodegenTests : JupyterReplTestCase() {
26
26
""" .trimIndent()
27
27
)
28
28
res1 shouldBe Unit
29
+
30
+ @Language(" kts" )
29
31
val res2 = execRaw(" df" ) as AnyFrame
30
32
31
33
res2[" value" ].type shouldBe typeOf<List <Any ?>>()
@@ -126,6 +128,7 @@ class JupyterCodegenTests : JupyterReplTestCase() {
126
128
)
127
129
res1 shouldBe Unit
128
130
131
+ @Language(" kts" )
129
132
val res2 = execRaw(" df.`1`" )
130
133
res2.shouldBeInstanceOf<ValueColumn <* >>()
131
134
}
@@ -141,6 +144,7 @@ class JupyterCodegenTests : JupyterReplTestCase() {
141
144
)
142
145
res1.shouldBeInstanceOf<MimeTypedResult >()
143
146
147
+ @Language(" kts" )
144
148
val res2 = exec(
145
149
""" listOf(df.`{a}`[0], df.`(b)`[0], df.`{c}`[0])"""
146
150
)
@@ -157,6 +161,8 @@ class JupyterCodegenTests : JupyterReplTestCase() {
157
161
""" .trimIndent()
158
162
)
159
163
res1.shouldBeInstanceOf<MimeTypedResult >()
164
+
165
+ @Language(" kts" )
160
166
val res2 = exec(
161
167
" listOf(df.`\$ id`[0])"
162
168
)
@@ -174,6 +180,8 @@ class JupyterCodegenTests : JupyterReplTestCase() {
174
180
)
175
181
res1.shouldBeInstanceOf<MimeTypedResult >()
176
182
println (res1.entries.joinToString())
183
+
184
+ @Language(" kts" )
177
185
val res2 = exec(
178
186
" listOf(df.`Day's`[0])"
179
187
)
@@ -193,6 +201,8 @@ class JupyterCodegenTests : JupyterReplTestCase() {
193
201
)
194
202
res1.shouldBeInstanceOf<MimeTypedResult >()
195
203
println (res1.entries.joinToString())
204
+
205
+ @Language(" kts" )
196
206
val res2 = exec(
197
207
" listOf(df.`Test `[0])"
198
208
)
@@ -212,6 +222,8 @@ class JupyterCodegenTests : JupyterReplTestCase() {
212
222
)
213
223
res1.shouldBeInstanceOf<MimeTypedResult >()
214
224
println (res1.entries.joinToString())
225
+
226
+ @Language(" kts" )
215
227
val res2 = exec(
216
228
" listOf(df.`Test `[0])"
217
229
)
@@ -220,6 +232,7 @@ class JupyterCodegenTests : JupyterReplTestCase() {
220
232
221
233
@Test
222
234
fun `generic interface` () {
235
+ @Language(" kts" )
223
236
val res1 = exec(
224
237
"""
225
238
@DataSchema
@@ -229,6 +242,8 @@ class JupyterCodegenTests : JupyterReplTestCase() {
229
242
""" .trimIndent()
230
243
)
231
244
res1.shouldBeInstanceOf<Unit >()
245
+
246
+ @Language(" kts" )
232
247
val res2 = exec(
233
248
"""
234
249
val <T> ColumnsContainer<Generic<T>>.test1: DataColumn<T> get() = field
@@ -240,6 +255,7 @@ class JupyterCodegenTests : JupyterReplTestCase() {
240
255
241
256
@Test
242
257
fun `generic interface with upper bound` () {
258
+ @Language(" kts" )
243
259
val res1 = exec(
244
260
"""
245
261
@DataSchema
@@ -249,6 +265,8 @@ class JupyterCodegenTests : JupyterReplTestCase() {
249
265
""" .trimIndent()
250
266
)
251
267
res1.shouldBeInstanceOf<Unit >()
268
+
269
+ @Language(" kts" )
252
270
val res2 = exec(
253
271
"""
254
272
val <T : String> ColumnsContainer<Generic<T>>.test1: DataColumn<T> get() = field
@@ -260,6 +278,7 @@ class JupyterCodegenTests : JupyterReplTestCase() {
260
278
261
279
@Test
262
280
fun `generic interface with variance and user type in type parameters` () {
281
+ @Language(" kts" )
263
282
val res1 = exec(
264
283
"""
265
284
interface UpperBound
@@ -271,6 +290,8 @@ class JupyterCodegenTests : JupyterReplTestCase() {
271
290
""" .trimIndent()
272
291
)
273
292
res1.shouldBeInstanceOf<Unit >()
293
+
294
+ @Language(" kts" )
274
295
val res2 = exec(
275
296
"""
276
297
val <T : UpperBound> ColumnsContainer<Generic<T>>.test1: DataColumn<T> get() = field
@@ -282,7 +303,8 @@ class JupyterCodegenTests : JupyterReplTestCase() {
282
303
283
304
@Test
284
305
fun `generate a new marker when dataframe marker is not a data schema so that columns are accessible with extensions` () {
285
- exec(
306
+ @Language(" kts" )
307
+ val a = exec(
286
308
"""
287
309
enum class State {
288
310
Idle, Productive, Maintenance
@@ -305,7 +327,8 @@ class JupyterCodegenTests : JupyterReplTestCase() {
305
327
""" .trimIndent()
306
328
)
307
329
shouldNotThrowAny {
308
- exec(
330
+ @Language(" kts" )
331
+ val b = exec(
309
332
"""
310
333
events.toolId
311
334
events.state
0 commit comments