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(),