fix the test

dlt-hub · May 10, 2024 · f204ceb · f204ceb
1 parent 6b66b7e
commit f204ceb
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/load/pipeline/test_pipelines.py b/tests/load/pipeline/test_pipelines.py
@@ -1014,9 +1014,10 @@ def table_3(make_data=False):
                     job_client.schema.tables[table_name]
                 ):
                     with client.with_staging_dataset():
-                        with client.execute_query(
-                            f"SELECT * FROM {client.make_qualified_table_name(table_name)}"
-                        ) as cur:
+                        tab_name = client.make_qualified_table_name(table_name)
+                        ind = tab_name.rfind('".')
+                        tab_name = tab_name[:ind] + "_staging" + tab_name[ind:]
+                        with client.execute_query(f"SELECT * FROM {tab_name}") as cur:
                             assert len(cur.fetchall()) == 0