File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed
main/kotlin/org/jetbrains/kotlinx/dataframe/api
test/kotlin/org/jetbrains/kotlinx/dataframe/codeGen Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -3,6 +3,8 @@ package org.jetbrains.kotlinx.dataframe.api
3
3
import org.jetbrains.dataframe.impl.codeGen.CodeGenerator
4
4
import org.jetbrains.kotlinx.dataframe.DataFrame
5
5
import org.jetbrains.kotlinx.dataframe.codeGen.MarkerVisibility
6
+ import org.jetbrains.kotlinx.dataframe.codeGen.NameNormalizer
7
+ import org.jetbrains.kotlinx.dataframe.impl.codeGen.from
6
8
7
9
public inline fun <reified T > DataFrame<T>.generateCode (
8
10
fields : Boolean = true,
@@ -36,7 +38,7 @@ public inline fun <reified T> DataFrame<T>.generateInterfaces(): CodeString = ge
36
38
37
39
public inline fun <reified T > DataFrame<T>.generateDataClasses (
38
40
markerName : String? = null,
39
- extensionProperties : Boolean = true ,
41
+ extensionProperties : Boolean = false ,
40
42
visibility : MarkerVisibility = MarkerVisibility .IMPLICIT_PUBLIC ,
41
43
useFqNames : Boolean = false
42
44
): CodeString {
@@ -49,8 +51,8 @@ public inline fun <reified T> DataFrame<T>.generateDataClasses(
49
51
extensionProperties = extensionProperties,
50
52
isOpen = false ,
51
53
visibility = visibility,
52
- asDataClass = true
53
- ).code.declarations
54
+ asDataClass = true ,
55
+ fieldNameNormalizer = nameNormalizer
54
56
).code.declarations.toCodeString()
55
57
}
56
58
Original file line number Diff line number Diff line change @@ -311,7 +311,7 @@ class CodeGenerationTests : BaseTest() {
311
311
312
312
@Test
313
313
fun `check method generateDataClasses` () {
314
- val code = typed.groupBy { name }.toDataFrame().generateDataClasses(extensionProperties = false )
314
+ val code = typed.groupBy { name }.toDataFrame().generateDataClasses()
315
315
316
316
code shouldBe """
317
317
@DataSchema
You can’t perform that action at this time.
0 commit comments