fix test failure

snowflakedb · Oct 15, 2024 · f42030d · f42030d
1 parent dcf5100
commit f42030d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/test_cte.py b/tests/unit/test_cte.py
@@ -13,7 +13,7 @@
 def test_case1():
     nodes = [mock.create_autospec(SnowflakePlan) for _ in range(7)]
     for i, node in enumerate(nodes):
-        node.encoded_id = i
+        node.encoded_node_id_with_query = i
         node.source_plan = None
     nodes[0].children_plan_nodes = [nodes[1], nodes[3]]
     nodes[1].children_plan_nodes = [nodes[2], nodes[2]]
@@ -30,7 +30,7 @@ def test_case1():
 def test_case2():
     nodes = [mock.create_autospec(SnowflakePlan) for _ in range(7)]
     for i, node in enumerate(nodes):
-        node.encoded_id = i
+        node.encoded_node_id_with_query = i
         node.source_plan = None
     nodes[0].children_plan_nodes = [nodes[1], nodes[3]]
     nodes[1].children_plan_nodes = [nodes[2], nodes[2]]