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