plugins/kotlin-dataframe/testData/box/aggregateDslIntoValueColumn.kt

// CHECK_TYPE_WITH_EXACT

import org.jetbrains.kotlinx.dataframe.*
import org.jetbrains.kotlinx.dataframe.annotations.*
import org.jetbrains.kotlinx.dataframe.api.*
import org.jetbrains.kotlinx.dataframe.io.*
import kotlin.reflect.typeOf

fun box(): String {
    val df = dataFrameOf(
        "key" to columnOf(0, 0),
        "numbers" to columnOf(1, null),
    )

    val result = df.groupBy { key }.aggregate {
        numbers into "aggregatedNumbers"
    }

    result[0].let {
        checkExactType<List<Int?>>(it.aggregatedNumbers)
    }
    result.compareSchemas(strict = true)
    return "OK"
}