fix assertion

Eventual-Inc · Feb 24, 2024 · b1b07d8 · b1b07d8
1 parent f839bf2
commit b1b07d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration/sql/test_databases.py b/tests/integration/sql/test_databases.py
@@ -12,4 +12,4 @@ def test_trino_create_dataframe_ok(db_url) -> None:
     df = daft.read_sql("SELECT * FROM tpch.sf1.nation", url)
     pd_df = pd.read_sql("SELECT * FROM tpch.sf1.nation", url)
 
-    assert df.equals(pd_df)
+    assert df.to_pandas().equals(pd_df)