Skip to content

Commit 45438f6

Browse files
committed
fix ut
1 parent cc106af commit 45438f6

File tree

4 files changed

+5
-2
lines changed

4 files changed

+5
-2
lines changed

backends-velox/src/test/scala/org/apache/gluten/functions/ArithmeticAnsiValidateSuite.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,8 @@ class ArithmeticAnsiValidateSuite extends FunctionsValidateSuite {
7676
runQueryAndCompare("SELECT int_field1 div 2 FROM datatab WHERE int_field1 IS NOT NULL") {
7777
checkGlutenOperatorMatch[ProjectExecTransformer]
7878
}
79-
intercept[ArithmeticException] {
80-
sql("SELECT 2147483647 div 0").collect()
79+
intercept[SparkException] {
80+
sql("SELECT 1 div 0 ").collect()
8181
}
8282
}
8383

shims/spark34/src/main/scala/org/apache/gluten/sql/shims/spark34/Spark34Shims.scala

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -562,6 +562,7 @@ class Spark34Shims extends SparkShims {
562562
case s: Subtract => s.evalMode == EvalMode.ANSI
563563
case d: Divide => d.evalMode == EvalMode.ANSI
564564
case m: Multiply => m.evalMode == EvalMode.ANSI
565+
case i: IntegralDivide => i.evalMode == EvalMode.ANSI
565566
case _ => false
566567
}
567568
}

shims/spark35/src/main/scala/org/apache/gluten/sql/shims/spark35/Spark35Shims.scala

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -601,6 +601,7 @@ class Spark35Shims extends SparkShims {
601601
case s: Subtract => s.evalMode == EvalMode.ANSI
602602
case d: Divide => d.evalMode == EvalMode.ANSI
603603
case m: Multiply => m.evalMode == EvalMode.ANSI
604+
case i: IntegralDivide => i.evalMode == EvalMode.ANSI
604605
case _ => false
605606
}
606607
}

shims/spark40/src/main/scala/org/apache/gluten/sql/shims/spark40/Spark40Shims.scala

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -608,6 +608,7 @@ class Spark40Shims extends SparkShims {
608608
case s: Subtract => s.evalMode == EvalMode.ANSI
609609
case d: Divide => d.evalMode == EvalMode.ANSI
610610
case m: Multiply => m.evalMode == EvalMode.ANSI
611+
case i: IntegralDivide => i.evalMode == EvalMode.ANSI
611612
case _ => false
612613
}
613614
}

0 commit comments

Comments
 (0)