Skip to content

Commit 7a6bfdb

Browse files
committed
Update generated sources with recent changes
1 parent 89e6236 commit 7a6bfdb

File tree

2 files changed

+39
-0
lines changed
  • core/generated-sources/src
    • main/kotlin/org/jetbrains/kotlinx/dataframe/api
    • test/kotlin/org/jetbrains/kotlinx/dataframe/samples/api

2 files changed

+39
-0
lines changed

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/cast.kt

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,13 @@ public inline fun <reified T> AnyFrame.cast(verify: Boolean = true): DataFrame<T
3333
).cast()
3434
else cast()
3535

36+
public inline fun <reified T> AnyFrame.castTo(
37+
@Suppress("UNUSED_PARAMETER") df: DataFrame<T>,
38+
verify: Boolean = true
39+
): DataFrame<T> {
40+
return cast<T>(verify = verify)
41+
}
42+
3643
public fun <T> AnyRow.cast(): DataRow<T> = this as DataRow<T>
3744

3845
public inline fun <reified T> AnyRow.cast(verify: Boolean = true): DataRow<T> = df().cast<T>(verify)[0]

core/generated-sources/src/test/kotlin/org/jetbrains/kotlinx/dataframe/samples/api/Modify.kt

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,15 @@ import org.jetbrains.kotlinx.dataframe.api.at
1414
import org.jetbrains.kotlinx.dataframe.api.by
1515
import org.jetbrains.kotlinx.dataframe.api.byName
1616
import org.jetbrains.kotlinx.dataframe.api.cast
17+
import org.jetbrains.kotlinx.dataframe.api.castTo
1718
import org.jetbrains.kotlinx.dataframe.api.colsOf
1819
import org.jetbrains.kotlinx.dataframe.api.column
1920
import org.jetbrains.kotlinx.dataframe.api.columnGroup
2021
import org.jetbrains.kotlinx.dataframe.api.columnOf
2122
import org.jetbrains.kotlinx.dataframe.api.concat
2223
import org.jetbrains.kotlinx.dataframe.api.convert
2324
import org.jetbrains.kotlinx.dataframe.api.convertTo
25+
import org.jetbrains.kotlinx.dataframe.api.count
2426
import org.jetbrains.kotlinx.dataframe.api.dataFrameOf
2527
import org.jetbrains.kotlinx.dataframe.api.default
2628
import org.jetbrains.kotlinx.dataframe.api.dropNulls
@@ -100,6 +102,7 @@ import org.jetbrains.kotlinx.dataframe.explainer.PluginCallbackProxy
100102
import org.jetbrains.kotlinx.dataframe.explainer.TransformDataFrameExpressions
101103
import org.jetbrains.kotlinx.dataframe.impl.api.mapNotNullValues
102104
import org.jetbrains.kotlinx.dataframe.indices
105+
import org.jetbrains.kotlinx.dataframe.io.readJson
103106
import org.jetbrains.kotlinx.dataframe.io.readJsonStr
104107
import org.jetbrains.kotlinx.dataframe.io.renderToString
105108
import org.jetbrains.kotlinx.dataframe.testResource
@@ -1421,4 +1424,33 @@ class Modify : TestBase() {
14211424
| 1 kotlin /kotlin 180
14221425
|""".trimMargin()
14231426
}
1427+
1428+
@DataSchema
1429+
interface ImplicitSchema {
1430+
val perf: Double
1431+
}
1432+
1433+
@Test
1434+
@Ignore
1435+
@Suppress("UNUSED_VARIABLE")
1436+
fun castToGenerateSchema() {
1437+
// SampleStart
1438+
val sample = DataFrame.readJson("sample.json")
1439+
// SampleEnd
1440+
}
1441+
1442+
@Test
1443+
@Suppress("KotlinConstantConditions")
1444+
fun castTo() {
1445+
val sample = dataFrameOf("perf")(10.0, 20.0, 12.0).cast<ImplicitSchema>()
1446+
val files = listOf<String>() // not intended to run
1447+
// SampleStart
1448+
for (file in files) {
1449+
// df here is expected to have the same structure as sample
1450+
val df = DataFrame.readJson(file).castTo(sample)
1451+
val count = df.count { perf > 10.0 }
1452+
println("$file: $count")
1453+
}
1454+
// SampleEnd
1455+
}
14241456
}

0 commit comments

Comments
 (0)