MCPcopy Create free account
hub / github.com/dask/dask / test_get_meta_library_gpu

Function test_get_meta_library_gpu

dask/tests/test_utils.py:931–948  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

929
930
931def test_get_meta_library_gpu():
932 cp = pytest.importorskip("cupy")
933 cudf = pytest.importorskip("cudf")
934 da = pytest.importorskip("dask.array")
935 dd = pytest.importorskip("dask.dataframe")
936
937 assert get_meta_library(cudf.DataFrame()) == cudf
938 assert get_meta_library(cp.array([])) == cp
939
940 assert get_meta_library(cudf.DataFrame()) == get_meta_library(cudf.DataFrame)
941 assert get_meta_library(cp.ndarray([])) == get_meta_library(cp.ndarray)
942
943 assert get_meta_library(cudf.DataFrame()) == get_meta_library(
944 dd.from_dict({}, npartitions=1).to_backend("cudf")
945 )
946 assert get_meta_library(cp.ndarray([])) == get_meta_library(
947 da.from_array([]).to_backend("cupy")
948 )
949
950
951def test_is_empty_list_and_tuple():

Callers

nothing calls this directly

Calls 3

get_meta_libraryFunction · 0.90
to_backendMethod · 0.45
from_dictMethod · 0.45

Tested by

no test coverage detected