diff --git a/tests/integration/sql/test_databases.py b/tests/integration/sql/test_databases.py index e98f11d72e..28faa79321 100644 --- 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)