From c0253807afa18235dc26c0279be0e017e6623878 Mon Sep 17 00:00:00 2001
From: Yun Zou <yun.zou@snowflake.com>
Date: Fri, 30 Aug 2024 18:16:39 -0700
Subject: [PATCH] enable test

---
 tests/integ/test_cte.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/tests/integ/test_cte.py b/tests/integ/test_cte.py
index 87a91deab0..b148ad6a68 100644
--- a/tests/integ/test_cte.py
+++ b/tests/integ/test_cte.py
@@ -423,10 +423,6 @@ def test_table(session):
     assert count_number_of_ctes(df_result.queries["queries"][-1]) == 1
 
 
-@pytest.mark.skipif(
-    "config.getoption('disable_sql_simplifier', default=False)",
-    reason="TODO SNOW-1556590: Re-enable test_sql in test_cte.py when sql simplifier is disabled once new CTE implementation is completed",
-)
 @pytest.mark.parametrize(
     "query",
     [
@@ -435,6 +431,11 @@ def test_table(session):
     ],
 )
 def test_sql(session, query):
+    if not session._query_compilation_stage_enabled:
+        pytest.skip(
+            "CTE query generation without the new query generation doesn't work correctly"
+        )
+
     df = session.sql(query).filter(lit(True))
     df_result = df.union_all(df).select("*")
     check_result(session, df_result, expect_cte_optimized=True)