From 3b7a76ffbe7735143992ce9120650ea430c4805f Mon Sep 17 00:00:00 2001 From: Phillip Cloud <417981+cpcloud@users.noreply.github.com> Date: Tue, 16 Jul 2024 09:16:07 -0400 Subject: [PATCH] test(flink): show more readable assertion error for flaky test --- ibis/backends/flink/tests/test_memtable.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ibis/backends/flink/tests/test_memtable.py b/ibis/backends/flink/tests/test_memtable.py index cfdf6e87a600..e07804da518b 100644 --- a/ibis/backends/flink/tests/test_memtable.py +++ b/ibis/backends/flink/tests/test_memtable.py @@ -8,7 +8,7 @@ @pytest.mark.parametrize( - "data,schema,expected", + ("data", "schema", "expected"), [ pytest.param( {"value": [{"a": 1}, {"a": 2}]}, @@ -36,7 +36,8 @@ def test_create_memtable(con, data, schema, expected): # `TableEnvironment.execute_sql()` and `TableEnvironment.sql_query()`; this doesn't # seem to be an issue if we don't execute memtable directly result = con.raw_sql(con.compile(t)).collect() - assert all(element in result for element in expected) + for element in expected: + assert element in result @pytest.mark.notyet(