Skip to content

Commit c55726c

Browse files
committed
[Compiler plugin] Add test for readExcel(StringColumns)
1 parent f267630 commit c55726c

File tree

3 files changed

+19
-1
lines changed

3 files changed

+19
-1
lines changed

plugins/kotlin-dataframe/src/org/jetbrains/kotlinx/dataframe/plugin/loadInterpreter.kt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@ import org.jetbrains.kotlinx.dataframe.plugin.impl.api.Move0
9090
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.PairConstructor
9191
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.PairToConstructor
9292
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ReadExcel
93+
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.StringColumnsConstructor
9394
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrame
9495
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameColumn
9596
import org.jetbrains.kotlinx.dataframe.plugin.impl.api.ToDataFrameDefault
@@ -232,7 +233,7 @@ internal inline fun <reified T> String.load(): T {
232233
"DataFrameOf0" -> DataFrameOf0()
233234
"DataFrameBuilderInvoke0" -> DataFrameBuilderInvoke0()
234235
"ToDataFrameColumn" -> ToDataFrameColumn()
235-
"StringColumns" -> ToDataFrameColumn()
236+
"StringColumns" -> StringColumnsConstructor()
236237
"ReadExcel" -> ReadExcel()
237238
"FillNulls0" -> FillNulls0()
238239
"UpdateWith0" -> UpdateWith0()
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
import org.jetbrains.kotlinx.dataframe.*
2+
import org.jetbrains.kotlinx.dataframe.annotations.*
3+
import org.jetbrains.kotlinx.dataframe.api.*
4+
import org.jetbrains.kotlinx.dataframe.io.*
5+
6+
fun box(): String {
7+
val df = @Import DataFrame.readExcel("testResources/sample.xls", stringColumns = StringColumns("A"))
8+
val d1: String = df.col1[0]
9+
val d2: Double = df.col2[0]
10+
return "OK"
11+
}

plugins/kotlin-dataframe/tests-gen/org/jetbrains/kotlin/fir/dataframe/DataFrameBlackBoxCodegenTestGenerated.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -346,6 +346,12 @@ public void testReadExcel() {
346346
runTest("testData/box/readExcel.kt");
347347
}
348348

349+
@Test
350+
@TestMetadata("readExcel_stringColumns.kt")
351+
public void testReadExcel_stringColumns() {
352+
runTest("testData/box/readExcel_stringColumns.kt");
353+
}
354+
349355
@Test
350356
@TestMetadata("readJson.kt")
351357
public void testReadJson() {

0 commit comments

Comments
 (0)