Skip to content

Commit

Permalink
Force datagen_seed for test_ceil_scale_zero and test_decimal_round (#…
Browse files Browse the repository at this point in the history
…9848)

Signed-off-by: Haoyang Li <[email protected]>
  • Loading branch information
thirtiseven authored Nov 24, 2023
1 parent d3629fd commit 3d116a8
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions integration_tests/src/main/python/arithmetic_ops_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -606,6 +606,7 @@ def test_ceil(data_gen):
assert_gpu_and_cpu_are_equal_collect(
lambda spark : unary_op_df(spark, data_gen).selectExpr('ceil(a)'))

@datagen_overrides(seed=0, reason='https://github.com/NVIDIA/spark-rapids/issues/9846')
@pytest.mark.skipif(is_before_spark_330(), reason='scale parameter in Ceil function is not supported before Spark 3.3.0')
@pytest.mark.parametrize('data_gen', double_n_long_gens + _arith_decimal_gens_no_neg_scale, ids=idfn)
def test_ceil_scale_zero(data_gen):
Expand Down Expand Up @@ -693,6 +694,7 @@ def test_decimal_bround(data_gen):

@incompat
@approximate_float
@datagen_overrides(seed=0, reason="https://github.com/NVIDIA/spark-rapids/issues/9847")
@pytest.mark.parametrize('data_gen', _arith_data_gens_for_round, ids=idfn)
def test_decimal_round(data_gen):
assert_gpu_and_cpu_are_equal_collect(
Expand Down

0 comments on commit 3d116a8

Please sign in to comment.