Skip to content

Commit ca22ef8

Browse files
committed
chore: use table instead of memtable
1 parent b0a59a1 commit ca22ef8

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

ibis/backends/tests/test_aggregation.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -685,12 +685,14 @@ def test_first_last_ordered(alltypes, method, filtered, include_null):
685685
),
686686
],
687687
)
688-
def test_first_last_ordered_in_mutate(con, method, expected):
688+
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 = ibis.memtable({"a": [1, 1, 2], "val": [4, 5, 6], "ob": [0, 2, 1]})
690+
t = alltypes.select(
691+
a=ibis._.tinyint_col, val=ibis._.int_col, ob=ibis._.bigint_col
692+
).filter(ibis._.val.isin((4, 5)))
691693
expr = t.mutate(new=method(t.val))
692694
actual = con.to_pyarrow(expr.new).to_pylist()
693-
assert actual == [expected] * 3
695+
assert actual == [expected] * len(actual)
694696

695697

696698
@pytest.mark.notimpl(

0 commit comments

Comments
 (0)