Skip to content

Commit 3bff124

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

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

ibis/backends/tests/test_aggregation.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -687,12 +687,17 @@ def test_first_last_ordered(alltypes, method, filtered, include_null):
687687
)
688688
def test_first_last_ordered_in_mutate(alltypes, con, method, expected):
689689
# originally reported in https://github.com/ibis-project/ibis/issues/11656
690-
t = alltypes.select(
691-
a=ibis._.tinyint_col, val=ibis._.int_col, ob=ibis._.bigint_col
692-
).filter(ibis._.val.isin((4, 5)))
690+
t = (
691+
alltypes.select(a=ibis._.tinyint_col, val=ibis._.int_col, ob=ibis._.bigint_col)
692+
.filter(
693+
((ibis._.val == 4) & (ibis._.ob == 40))
694+
| ((ibis._.val == 5) & (ibis._.ob == 50))
695+
)
696+
.limit(10)
697+
)
693698
expr = t.mutate(new=method(t.val))
694699
actual = con.to_pyarrow(expr.new).to_pylist()
695-
assert actual == [expected] * len(actual)
700+
assert actual == [expected] * 10
696701

697702

698703
@pytest.mark.notimpl(

0 commit comments

Comments
 (0)