fix(tests): skip cudf_streaming tests gracefully when wheel unavailable

Author: vyasr

python/rapidsmpf/rapidsmpf/tests/streaming/test_allgather.py

@@ -9,6 +9,8 @@
 import numpy as np
 import pylibcudf as plc
 import pytest
+
+pytest.importorskip("cudf_streaming")
 from cudf_streaming.integrations.partition import unpack_and_concat
 from cudf_streaming.streaming.table_chunk import TableChunk
 

python/rapidsmpf/rapidsmpf/tests/streaming/test_define_actor.py

@@ -7,6 +7,8 @@
 
 import pylibcudf as plc
 import pytest
+
+pytest.importorskip("cudf_streaming")
 from cudf_streaming.streaming.table_chunk import TableChunk
 
 from rapidsmpf.streaming.chunks.arbitrary import ArbitraryChunk

python/rapidsmpf/rapidsmpf/tests/streaming/test_fanout.py

@@ -9,6 +9,8 @@
 
 import pylibcudf as plc
 import pytest
+
+pytest.importorskip("cudf_streaming")
 from cudf_streaming.streaming.table_chunk import TableChunk
 
 from rapidsmpf.streaming.core.actor import run_actor_network

python/rapidsmpf/rapidsmpf/tests/streaming/test_leaf_actor.py

@@ -7,6 +7,8 @@
 
 import pylibcudf as plc
 import pytest
+
+pytest.importorskip("cudf_streaming")
 from cudf_streaming.streaming.table_chunk import TableChunk
 
 from rapidsmpf.streaming.core.actor import run_actor_network

python/rapidsmpf/rapidsmpf/tests/streaming/test_shuffler.py

@@ -9,6 +9,8 @@
 import numpy as np
 import pylibcudf as plc
 import pytest
+
+pytest.importorskip("cudf_streaming")
 from cudf_streaming.integrations.partition import split_and_pack, unpack_and_concat
 from cudf_streaming.streaming.partition import (
     partition_and_pack,

python/rapidsmpf/rapidsmpf/tests/streaming/test_sparse_alltoall.py

@@ -9,6 +9,8 @@
 import numpy as np
 import pylibcudf as plc
 import pytest
+
+pytest.importorskip("cudf_streaming")
 from cudf_streaming.integrations.partition import unpack_and_concat
 
 from rapidsmpf.memory.packed_data import PackedData

python/rapidsmpf/rapidsmpf/tests/test_allgather.py

@@ -10,6 +10,8 @@
 import numpy as np
 import pylibcudf as plc
 import pytest
+
+pytest.importorskip("cudf_streaming")
 from cudf_streaming.integrations.partition import unpack_and_concat
 from pylibcudf.contiguous_split import pack
 

python/rapidsmpf/rapidsmpf/tests/test_examples.py

@@ -9,6 +9,8 @@
 import pylibcudf as plc
 import pytest
 
+pytest.importorskip("cudf_streaming")
+
 from rapidsmpf.memory.buffer_resource import BufferResource
 from rapidsmpf.testing import assert_eq
 

python/rapidsmpf/rapidsmpf/tests/test_partition.py

@@ -7,6 +7,8 @@
 import numpy as np
 import pylibcudf as plc
 import pytest
+
+pytest.importorskip("cudf_streaming")
 from cudf_streaming.integrations.partition import (
     partition_and_pack,
     spill_partitions,

python/rapidsmpf/rapidsmpf/tests/test_shuffler.py

@@ -8,6 +8,8 @@
 import numpy as np
 import pylibcudf as plc
 import pytest
+
+pytest.importorskip("cudf_streaming")
 from cudf_streaming.integrations.partition import (
     partition_and_pack,
     unpack_and_concat,