Skip to content

Commit dd535ad

Browse files
committed
add array lookup to test_multiindex_lookup_tuple_mixed_dtypes
1 parent 086ce45 commit dd535ad

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

tests/pandas/index_test.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -861,4 +861,11 @@ def test_multiindex_lookup_tuple_mixed_dtypes(self):
861861
assert mask.dtype == ak.bool_
862862

863863
# Expect exactly the first row to match
864-
assert mask.to_ndarray().tolist() == [True, False, False, False]
864+
assert mask.tolist() == [True, False, False, False]
865+
866+
assert midx.lookup(([ak.array([1, 2]), ak.array(["blue", "red"])])).tolist() == [
867+
False,
868+
True,
869+
True,
870+
False,
871+
]

0 commit comments

Comments
 (0)