diff --git a/datafusion/core/tests/sql/mod.rs b/datafusion/core/tests/sql/mod.rs index 522cdce1a40d3..7d175b65260f4 100644 --- a/datafusion/core/tests/sql/mod.rs +++ b/datafusion/core/tests/sql/mod.rs @@ -635,6 +635,9 @@ pub fn make_partition(sz: i32) -> RecordBatch { /// Specialised String representation fn col_str(column: &ArrayRef, row_index: usize) -> String { + // NullArray::is_null() does not work on NullArray. + // can remove check for DataType::Null when + // https://github.com/apache/arrow-rs/issues/4835 is fixed if column.data_type() == &DataType::Null || column.is_null(row_index) { return "NULL".to_string(); } diff --git a/datafusion/optimizer/src/analyzer/type_coercion.rs b/datafusion/optimizer/src/analyzer/type_coercion.rs index f976a3468df36..5e239f8e99348 100644 --- a/datafusion/optimizer/src/analyzer/type_coercion.rs +++ b/datafusion/optimizer/src/analyzer/type_coercion.rs @@ -881,6 +881,7 @@ mod test { #[test] fn scalar_function() -> Result<()> { + // test that automatic argument type coercion for scalar functions work let empty = empty(); let lit_expr = lit(10i64); let fun: BuiltinScalarFunction = BuiltinScalarFunction::Acos;