Skip to content

Commit f57ce17

Browse files
committed
chore: make test data closer
1 parent ca22ef8 commit f57ce17

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

ibis/backends/tests/test_aggregation.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -689,10 +689,13 @@ def test_first_last_ordered_in_mutate(alltypes, con, method, expected):
689689
# originally reported in https://github.com/ibis-project/ibis/issues/11656
690690
t = alltypes.select(
691691
a=ibis._.tinyint_col, val=ibis._.int_col, ob=ibis._.bigint_col
692-
).filter(ibis._.val.isin((4, 5)))
693-
expr = t.mutate(new=method(t.val))
692+
).filter(
693+
((ibis._.val == 4) & (ibis._.ob == 40))
694+
| ((ibis._.val == 5) & (ibis._.ob == 50))
695+
)
696+
expr = t.mutate(new=method(t.val)).limit(10)
694697
actual = con.to_pyarrow(expr.new).to_pylist()
695-
assert actual == [expected] * len(actual)
698+
assert actual == [expected] * 10
696699

697700

698701
@pytest.mark.notimpl(

0 commit comments

Comments
 (0)