Skip to content

Commit b28756f

Browse files
committed
test: columns Vec<String>
1 parent b9bbbd3 commit b28756f

File tree

1 file changed

+5
-4
lines changed
  • datafusion/core/tests/dataframe

1 file changed

+5
-4
lines changed

datafusion/core/tests/dataframe/mod.rs

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5376,10 +5376,10 @@ async fn test_fill_null() -> Result<()> {
53765376

53775377
// Use fill_null to replace nulls on each column.
53785378
let df_filled = df
5379-
.fill_null(ScalarValue::Int32(Some(0)), Some(vec!["a".to_string()]))?
5379+
.fill_null(ScalarValue::Int32(Some(0)), vec!["a".to_string()])?
53805380
.fill_null(
53815381
ScalarValue::Utf8(Some("default".to_string())),
5382-
Some(vec!["b".to_string()]),
5382+
vec!["b".to_string()],
53835383
)?;
53845384

53855385
let results = df_filled.collect().await?;
@@ -5403,7 +5403,8 @@ async fn test_fill_null_all_columns() -> Result<()> {
54035403
// Use fill_null to replace nulls on all columns.
54045404
// Only column "b" will be replaced since ScalarValue::Utf8(Some("default".to_string()))
54055405
// can be cast to Utf8.
5406-
let df_filled = df.fill_null(ScalarValue::Utf8(Some("default".to_string())), None)?;
5406+
let df_filled =
5407+
df.fill_null(ScalarValue::Utf8(Some("default".to_string())), vec![])?;
54075408

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

@@ -5420,7 +5421,7 @@ async fn test_fill_null_all_columns() -> Result<()> {
54205421
assert_batches_sorted_eq!(expected, &results);
54215422

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

54255426
let results = df_filled.collect().await?;
54265427
let expected = [

0 commit comments

Comments
 (0)