update test (#6071)

Signed-off-by: remzi <13716567376yh@gmail.com>
NVIDIA · Aug 1, 2022 · e2d18fd · e2d18fd
1 parent f2344dc
commit e2d18fd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/integration_tests/src/main/python/arithmetic_ops_test.py b/integration_tests/src/main/python/arithmetic_ops_test.py
@@ -921,7 +921,7 @@ def test_add_overflow_with_ansi_enabled(data, tp, expr):
         assert_gpu_and_cpu_error(
             lambda spark: _get_overflow_df(spark, data, tp, expr).collect(),
             conf=ansi_enabled_conf,
-            error_message='overflow')
+            error_message='java.lang.ArithmeticException' if is_before_spark_330() else 'SparkArithmeticException')
     elif isinstance(tp, DecimalType):
         assert_gpu_and_cpu_error(
             lambda spark: _get_overflow_df(spark, data, tp, expr).collect(),
@@ -950,7 +950,7 @@ def test_subtraction_overflow_with_ansi_enabled(data, tp, expr):
         assert_gpu_and_cpu_error(
             lambda spark: _get_overflow_df(spark, data, tp, expr).collect(),
             conf=ansi_enabled_conf,
-            error_message='overflow')
+            error_message='java.lang.ArithmeticException' if is_before_spark_330() else 'SparkArithmeticException')
     elif isinstance(tp, DecimalType):
         assert_gpu_and_cpu_error(
             lambda spark: _get_overflow_df(spark, data, tp, expr).collect(),