// WITH_FIR_TEST_COMPILER_PLUGIN
interface DataFrame<out T>

annotation class Refine

@Refine
fun <T, R> DataFrame<T>.add(columnName: String, expression: () -> R): DataFrame<Any?> = TODO()

fun test_1(df: DataFrame<*>?) {
    val df1 by lazy { df.add("column") { 1 } }

    <expr>df1.column</expr>
}