Skip to content

Commit

Permalink
Merge pull request #9868 from NVIDIA/branch-23.12
Browse files Browse the repository at this point in the history
[auto-merge] branch-23.12 to branch-24.02 [skip ci] [bot]
  • Loading branch information
nvauto authored Nov 28, 2023
2 parents fd05f70 + 26c9e37 commit c65c5b8
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions integration_tests/src/main/python/collection_ops_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -278,6 +278,8 @@ def test_sequence_with_step(start_gen, stop_gen, step_gen):
# Get a step scalar from the 'step_gen' which follows the rules.
step_gen.start(random.Random(data_gen_seed))
step_lit = step_gen.gen()
if step_lit is None:
step_lit = "null"
assert_gpu_and_cpu_are_equal_collect(
lambda spark: three_col_df(spark, start_gen, stop_gen, step_gen).selectExpr(
"sequence(a, b, c)",
Expand Down Expand Up @@ -327,8 +329,8 @@ def test_sequence_illegal_boundaries(start_gen, stop_gen, step_gen):
# Exceed the max length of a sequence
# "Too long sequence: xxxxxxxxxx. Should be <= 2147483632"
sequence_too_long_length_gens = [
IntegerGen(min_val=2147483633, max_val=2147483633, special_cases=[]),
LongGen(min_val=2147483635, max_val=2147483635, special_cases=[None])
IntegerGen(min_val=2147483633, max_val=2147483633, special_cases=[], nullable=False),
LongGen(min_val=2147483635, max_val=2147483635, special_cases=[], nullable=False)
]

@pytest.mark.parametrize('stop_gen', sequence_too_long_length_gens, ids=idfn)
Expand Down

0 comments on commit c65c5b8

Please sign in to comment.