def test_get(data, index, expectation):
with expectation:
expect = Series(data).list.get(index)
if expectation == does_not_raise():
ds = dgd.from_cudf(Series(data), 5)
assert_eq(expect, ds.list.get(index).compute())
def test_take(data, list_indices, expectation):
with expectation:
expect = Series(data).list.take(list_indices)
if expectation == does_not_raise():
ds = dgd.from_cudf(Series(data), 5)
assert_eq(expect, ds.list.take(list_indices).compute())
@pytest.mark.parametrize(
"data, search_key",
[(data_test_2(), 1)],
)
def test_contains(data, search_key):
expect = Series(data).list.contains(search_key)
ds = dgd.from_cudf(Series(data), 5)
assert_eq(expect, ds.list.contains(search_key).compute())
@pytest.mark.parametrize(
"data, index, expectation",
[
(data_test_1(), 1, does_not_raise()),
(data_test_2(), 2, pytest.raises(IndexError)),
],
)
def test_get(data, index, expectation):
with expectation:
expect = Series(data).list.get(index)
if expectation == does_not_raise():
ds = dgd.from_cudf(Series(data), 5)
assert_eq(expect, ds.list.get(index).compute())
@pytest.mark.parametrize(
"data",
[data_test_1(), data_test_2(),