ploomber#892 Fix test

marshallwhiteorg · Sep 22, 2023 · 3058bd1 · 3058bd1
1 parent 0940c6d
commit 3058bd1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/tests/test_magic.py b/src/tests/test_magic.py
@@ -2042,18 +2042,20 @@ def test_comments_in_duckdb_select_summarize(ip_empty):
     ).result
     assert out.dict() == expected_summarize
 
+    expected_select = {"memid": (1, 2, 3, 5, 8)}
+
     out = ip_empty.run_cell(
         """%%sql
         /*x*/
         SELECT * FROM df
         """
     ).result
-    assert out.DataFrame().equals(df)
+    assert out.dict() == expected_select
 
     out = ip_empty.run_cell(
         """%%sql
         /*x*/
         FROM df SELECT *
         """
     ).result
-    assert out.DataFrame().equals(df)
+    assert out.dict() == expected_select