Skip to content

Commit

Permalink
Review Part 29
Browse files Browse the repository at this point in the history
  • Loading branch information
ozankabak committed Feb 14, 2025
1 parent 081e3f3 commit 6d0e9e3
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 21 deletions.
12 changes: 5 additions & 7 deletions datafusion/physical-expr/src/expressions/binary.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4424,13 +4424,11 @@ mod tests {
right: Arc<dyn PhysicalExpr>,
schema: &Schema,
) -> Result<BinaryExpr> {
Ok(
binary_op(Arc::clone(&left), op, Arc::clone(&right), schema)?
.as_any()
.downcast_ref::<BinaryExpr>()
.unwrap()
.clone(),
)
Ok(binary_op(left, op, right, schema)?
.as_any()
.downcast_ref::<BinaryExpr>()
.unwrap()
.clone())
}

/// Test for Uniform-Uniform, Unknown-Uniform, Uniform-Unknown and Unknown-Unknown evaluation.
Expand Down
20 changes: 6 additions & 14 deletions datafusion/physical-expr/src/utils/stats_v2_graph.rs
Original file line number Diff line number Diff line change
Expand Up @@ -405,7 +405,7 @@ pub fn compute_variance(
mod tests {
use std::sync::Arc;

use crate::expressions::{binary, try_cast, BinaryExpr, Column};
use crate::expressions::{binary, try_cast, Column};
use crate::intervals::cp_solver::PropagationResult;
use crate::utils::stats_v2_graph::{
compute_mean, compute_median, compute_variance, create_bernoulli_from_comparison,
Expand All @@ -432,16 +432,15 @@ mod tests {
op: Operator,
right: Arc<dyn PhysicalExpr>,
schema: &Schema,
) -> Result<BinaryExpr> {
) -> Result<Arc<dyn PhysicalExpr>> {
let left_type = left.data_type(schema)?;
let right_type = right.data_type(schema)?;
let binary_type_coercer = BinaryTypeCoercer::new(&left_type, &op, &right_type);
let (lhs, rhs) = binary_type_coercer.get_input_types()?;

let left_expr = try_cast(left, schema, lhs)?;
let right_expr = try_cast(right, schema, rhs)?;
let b = binary(left_expr, op, right_expr, schema);
Ok(b?.as_any().downcast_ref::<BinaryExpr>().unwrap().clone())
binary(left_expr, op, right_expr, schema)
}

// Expected test results were calculated in Wolfram Mathematica, by using
Expand Down Expand Up @@ -596,16 +595,9 @@ mod tests {
let c = Arc::new(Column::new("c", 2)) as _;
let d = Arc::new(Column::new("d", 3)) as _;

let left =
Arc::new(binary_expr(Arc::clone(&a), Plus, Arc::clone(&b), schema)?) as _;
let right =
Arc::new(binary_expr(Arc::clone(&c), Minus, Arc::clone(&d), schema)?) as _;
let expr = Arc::new(binary_expr(
Arc::clone(&left),
Eq,
Arc::clone(&right),
schema,
)?);
let left = binary_expr(a, Plus, b, schema)?;
let right = binary_expr(c, Minus, d, schema)?;
let expr = binary_expr(left, Eq, right, schema)?;

let mut graph = ExprStatisticGraph::try_new(expr, schema)?;
// 2, 5 and 6 are BinaryExpr
Expand Down

0 comments on commit 6d0e9e3

Please sign in to comment.