From 3058bd12968aa6192dfc91e00f0fad6dcfdaf2ef Mon Sep 17 00:00:00 2001 From: Marshall White Date: Fri, 22 Sep 2023 16:37:55 -0400 Subject: [PATCH] #892 Fix test --- src/tests/test_magic.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/tests/test_magic.py b/src/tests/test_magic.py index 9ff1ee750..133bfdf7f 100644 --- a/src/tests/test_magic.py +++ b/src/tests/test_magic.py @@ -2042,13 +2042,15 @@ 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 @@ -2056,4 +2058,4 @@ def test_comments_in_duckdb_select_summarize(ip_empty): FROM df SELECT * """ ).result - assert out.DataFrame().equals(df) + assert out.dict() == expected_select