Skip to content

Commit

Permalink
test(flink): show more readable assertion error for flaky test
Browse files Browse the repository at this point in the history
  • Loading branch information
cpcloud authored and jcrist committed Jul 16, 2024
1 parent 97af53c commit 3b7a76f
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions ibis/backends/flink/tests/test_memtable.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@


@pytest.mark.parametrize(
"data,schema,expected",
("data", "schema", "expected"),
[
pytest.param(
{"value": [{"a": 1}, {"a": 2}]},
Expand Down Expand Up @@ -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(
Expand Down

0 comments on commit 3b7a76f

Please sign in to comment.