diff --git a/datafusion/core/src/physical_plan/filter.rs b/datafusion/core/src/physical_plan/filter.rs index 6a3834b11222a..d4728d4ecf50f 100644 --- a/datafusion/core/src/physical_plan/filter.rs +++ b/datafusion/core/src/physical_plan/filter.rs @@ -282,12 +282,8 @@ mod tests { let partitions = 4; let input = test::scan_partitioned_csv(partitions)?; - let predicate: Arc = binary( - col("c2", &schema)?, - Operator::Gt, - lit(ScalarValue::from(1u32)), - &schema, - )?; + let predicate: Arc = + binary(col("c2", &schema)?, Operator::Gt, lit(1u32), &schema)?; let filter: Arc = Arc::new(FilterExec::try_new(predicate, input.clone())?); diff --git a/datafusion/physical-expr/src/expressions/case.rs b/datafusion/physical-expr/src/expressions/case.rs index ff3bfab634c3d..7e4b8867262b2 100644 --- a/datafusion/physical-expr/src/expressions/case.rs +++ b/datafusion/physical-expr/src/expressions/case.rs @@ -372,10 +372,10 @@ mod tests { let schema = batch.schema(); // CASE a WHEN 'foo' THEN 123 WHEN 'bar' THEN 456 END - let when1 = lit(ScalarValue::Utf8(Some("foo".to_string()))); - let then1 = lit(ScalarValue::Int32(Some(123))); - let when2 = lit(ScalarValue::Utf8(Some("bar".to_string()))); - let then2 = lit(ScalarValue::Int32(Some(456))); + let when1 = lit("foo"); + let then1 = lit(123i32); + let when2 = lit("bar"); + let then2 = lit(456i32); let expr = case( Some(col("a", &schema)?), @@ -402,11 +402,11 @@ mod tests { let schema = batch.schema(); // CASE a WHEN 'foo' THEN 123 WHEN 'bar' THEN 456 ELSE 999 END - let when1 = lit(ScalarValue::Utf8(Some("foo".to_string()))); - let then1 = lit(ScalarValue::Int32(Some(123))); - let when2 = lit(ScalarValue::Utf8(Some("bar".to_string()))); - let then2 = lit(ScalarValue::Int32(Some(456))); - let else_value = lit(ScalarValue::Int32(Some(999))); + let when1 = lit("foo"); + let then1 = lit(123i32); + let when2 = lit("bar"); + let then2 = lit(456i32); + let else_value = lit(999i32); let expr = case( Some(col("a", &schema)?),