Improve test suite (#425)

* Improve test suite * update
fugue-project · Feb 2, 2023 · 2aeb33a · 2aeb33a
1 parent 8c04e9f
commit 2aeb33a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fugue_test/builtin_suite.py b/fugue_test/builtin_suite.py
@@ -29,8 +29,8 @@
     PandasDataFrame,
     PartitionSpec,
     Processor,
+    QPDPandasEngine,
     Schema,
-    SqliteEngine,
     Transformer,
     cotransformer,
     output_cotransformer,
@@ -864,7 +864,7 @@ def test_aggregate(self):
             dag.run(self.engine)
 
         def test_select(self):
-            class MockEngine(SqliteEngine):
+            class MockEngine(QPDPandasEngine):
                 def __init__(self, execution_engine, p: int = 0):
                     super().__init__(execution_engine)
                     self.p = p
@@ -901,7 +901,7 @@ def select(self, dfs, statement):
                     "AS t1 INNER JOIN",
                     b,
                     "AS t2 ON t1.x=t2.x",
-                    sql_engine="sqlite",
+                    sql_engine="qpdpandas",
                 ).assert_eq(c)
 
                 # specify sql engine and params