#1245

mabel-dev · Nov 7, 2023 · 04021cd · 04021cd
1 parent 1b4c5f5
commit 04021cd
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/storage/test_cache_in_memory.py b/tests/storage/test_cache_in_memory.py
@@ -61,7 +61,7 @@ def test_cache_in_subqueries():
     # read the data once, this should populate the cache
     conn = opteryx.connect()
     cur = conn.cursor()
-    cur.execute("SELECT * FROM (SELECT * FROM testdata.flat.tweets WITH(NO_PARTITION));")
+    cur.execute("SELECT * FROM (SELECT * FROM testdata.flat.tweets WITH(NO_PARTITION)) AS SQ;")
     cur.arrow()
 
     stats = cur.stats
@@ -72,7 +72,7 @@ def test_cache_in_subqueries():
     # read the data a second time, this should hit the cache
     conn = opteryx.connect()
     cur = conn.cursor()
-    cur.execute("SELECT * FROM (SELECT * FROM testdata.flat.tweets WITH(NO_PARTITION));")
+    cur.execute("SELECT * FROM (SELECT * FROM testdata.flat.tweets WITH(NO_PARTITION)) AS SQ;")
     cur.arrow()
 
     stats = cur.stats
@@ -83,7 +83,9 @@ def test_cache_in_subqueries():
     # read the data with the no cache directive
     conn = opteryx.connect()
     cur = conn.cursor()
-    cur.execute("SELECT * FROM (SELECT * FROM testdata.flat.tweets WITH(NO_CACHE, NO_PARTITION));")
+    cur.execute(
+        "SELECT * FROM (SELECT * FROM testdata.flat.tweets WITH(NO_CACHE, NO_PARTITION)) AS SQ;"
+    )
     cur.arrow()
 
     stats = cur.stats