diff --git a/integration_tests/src/main/python/ast_test.py b/integration_tests/src/main/python/ast_test.py index 2c06c51a876..94976ea2208 100644 --- a/integration_tests/src/main/python/ast_test.py +++ b/integration_tests/src/main/python/ast_test.py @@ -380,3 +380,9 @@ def test_or(data_gen): f.col('a') | f.lit(True), f.lit(False) | f.col('b'), f.col('a') | f.col('b'))) + +def test_multi_tier_ast(): + assert_gpu_ast( + is_supported=True, + func=lambda spark: spark.range(10).withColumn("x", f.col("id")).repartition(1)\ + .selectExpr("(id < x) == (id < (id + x))"))