Module test_collection_cuda

distributed.protocol.tests.test_collection_cuda

Functions

▷ def test_serialize_cupy(collection, y, y_serializer) @pytest.mark.parametrize("collection", [tuple, dict]) @pytest.mark.parametrize("y,y_serializer", [(50, "cuda"), (None, "pickle")])
@pytest.mark.parametrize("collection", [tuple, dict])
@pytest.mark.parametrize("y,y_serializer", [(50, "cuda"), (None, "pickle")])
def test_serialize_cupy(
collection,
y,
y_serializer,
)
source link
▷ def test_serialize_pandas_pandas(collection, df2, df2_serializer) @pytest.mark.parametrize("collection", [tuple, dict]) @pytest.mark.parametrize( "df2,df2_serializer", [(pd.DataFrame({"C": [3, 4, 5], "D": [2.5, 3.5, 4.5]}), "cuda"), (None, "pickle")], )
@pytest.mark.parametrize("collection", [tuple, dict])
@pytest.mark.parametrize( "df2,df2_serializer", [(pd.DataFrame({"C": [3, 4, 5], "D": [2.5, 3.5, 4.5]}), "cuda"), (None, "pickle")], )
def test_serialize_pandas_pandas(
collection,
df2,
df2_serializer,
)
source link