Function test_fill_null_all_columns

datafusion/core/tests/dataframe/mod.rs:6486–6527

()

#[tokio::test]
async fn test_fill_null_all_columns() -> Result<()> {
    let df = create_null_table().await?;

    // Use fill_null to replace nulls on all columns.
    // Only column "b" will be replaced since ScalarValue::Utf8(Some("default".to_string()))
    // can be cast to Utf8.
    let df_filled =
        df.fill_null(ScalarValue::Utf8(Some("default".to_string())), vec![])?;

    let results = df_filled.clone().collect().await?;

    assert_snapshot!(
        batches_to_sort_string(&results),
        @r"
    +---+---------+
    | a | b       |
    +---+---------+
    |   | default |
    | 1 | x       |
    | 3 | z       |
    +---+---------+
    "
    );

    // Fill column "a" null values with a value that cannot be cast to Int32.
    let df_filled = df_filled.fill_null(ScalarValue::Int32(Some(0)), vec![])?;

    let results = df_filled.collect().await?;
    assert_snapshot!(
        batches_to_sort_string(&results),
        @r"
    +---+---------+
    | a | b       |
    +---+---------+
    | 0 | default |
    | 1 | x       |
    | 3 | z       |
    +---+---------+
    "
    );
    Ok(())
}

#[tokio::test]
async fn test_insert_into_casting_support() -> Result<()> {