From 2c4c91f54c1d83845ba87340ffcd309f4e87a0c6 Mon Sep 17 00:00:00 2001 From: Raza Jafri Date: Thu, 24 Mar 2022 16:01:26 -0700 Subject: [PATCH] Removed limit from the test (#5033) * removed limit Signed-off-by: Raza Jafri * revert change to add executor flags Signed-off-by: Raza Jafri * adding ignore_order to guarantee the order Signed-off-by: Raza Jafri Co-authored-by: Raza Jafri --- integration_tests/src/main/python/cache_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/integration_tests/src/main/python/cache_test.py b/integration_tests/src/main/python/cache_test.py index 84503642e8d..4b3bce54fb9 100644 --- a/integration_tests/src/main/python/cache_test.py +++ b/integration_tests/src/main/python/cache_test.py @@ -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)