Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
  • Loading branch information
madsbk committed Oct 11, 2024
1 parent 7b07c10 commit 6bc388c
Showing 1 changed file with 9 additions and 3 deletions.
12 changes: 9 additions & 3 deletions python/pylibcudf/pylibcudf/tests/test_contiguous_split.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
import pytest
from utils import assert_table_eq

mixed_pyarrow_tables = [
param_pyarrow_tables = [
pa.table([]),
pa.table({"a": [1, 2, 3], "b": [4, 5, 6], "c": [7, 8, 9]}),
pa.table({"a": [1, 2, 3]}),
Expand All @@ -21,10 +21,16 @@
],
}
),
pytest.param(
pa.array([{"a": [1, 2], "b": [3, 4]}]),
marks=pytest.mark.xfail(
reason="https://github.com/rapidsai/cudf/issues/17061"
),
),
]


@pytest.mark.parametrize("arrow_tbl", mixed_pyarrow_tables)
@pytest.mark.parametrize("arrow_tbl", param_pyarrow_tables)
def test_pack_and_unpack(arrow_tbl):
plc_tbl = plc.interop.from_arrow(arrow_tbl)
packed = plc.contiguous_split.pack(plc_tbl)
Expand All @@ -33,7 +39,7 @@ def test_pack_and_unpack(arrow_tbl):
assert_table_eq(arrow_tbl, res)


@pytest.mark.parametrize("arrow_tbl", mixed_pyarrow_tables)
@pytest.mark.parametrize("arrow_tbl", param_pyarrow_tables)
def test_pack_and_unpack_from_memoryviews(arrow_tbl):
plc_tbl = plc.interop.from_arrow(arrow_tbl)
packed = plc.contiguous_split.pack(plc_tbl)
Expand Down

0 comments on commit 6bc388c

Please sign in to comment.