Skip to content

Commit

Permalink
Removed limit from the test (#5033)
Browse files Browse the repository at this point in the history
* removed limit

Signed-off-by: Raza Jafri <rjafri@nvidia.com>

* revert change to add executor flags

Signed-off-by: Raza Jafri <rjafri@nvidia.com>

* adding ignore_order to guarantee the order

Signed-off-by: Raza Jafri <rjafri@nvidia.com>

Co-authored-by: Raza Jafri <rjafri@nvidia.com>
  • Loading branch information
razajafri and razajafri authored Mar 24, 2022
1 parent ceb3558 commit 2c4c91f
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion integration_tests/src/main/python/cache_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,12 +117,13 @@ def partial_return_cache(spark):

@pytest.mark.parametrize('enable_vectorized_conf', enable_vectorized_confs, ids=idfn)
@allow_non_gpu('CollectLimitExec')
@ignore_order
def test_cache_reverse_order(enable_vectorized_conf):
col0 = StructGen([['child0', StructGen([['child1', byte_gen]])]])
col1 = StructGen([['child0', byte_gen]])
def partial_return():
def partial_return_cache(spark):
return two_col_df(spark, col0, col1).select(f.col("a"), f.col("b")).cache().limit(50).select(f.col("b"), f.col("a"))
return two_col_df(spark, col0, col1).select(f.col("a"), f.col("b")).cache().select(f.col("b"), f.col("a"))
return partial_return_cache
assert_gpu_and_cpu_are_equal_collect(partial_return(), conf=enable_vectorized_conf)

Expand Down

0 comments on commit 2c4c91f

Please sign in to comment.