refactor tests

rapidsai · May 14, 2024 · c557adf · c557adf
1 parent b45a1a5
commit c557adf
Show file tree

Hide file tree

Showing 6 changed files with 2 additions and 8 deletions.
diff --git a/...g/cugraph_pyg/tests/test_cugraph_store.py → ...h_pyg/tests/data/test_dask_graph_store.py b/...g/cugraph_pyg/tests/test_cugraph_store.py → ...h_pyg/tests/data/test_dask_graph_store.py
diff --git a/...aph_pyg/tests/mg/test_mg_cugraph_store.py → ...yg/tests/data/test_dask_graph_store_mg.py b/...aph_pyg/tests/mg/test_mg_cugraph_store.py → ...yg/tests/data/test_dask_graph_store_mg.py
diff --git a/.../cugraph_pyg/tests/test_cugraph_loader.py → ...tests/loader/test_dask_neighbor_loader.py b/.../cugraph_pyg/tests/test_cugraph_loader.py → ...tests/loader/test_dask_neighbor_loader.py
diff --git a/...ph_pyg/tests/mg/test_mg_cugraph_loader.py → ...ts/loader/test_dask_neighbor_loader_mg.py b/...ph_pyg/tests/mg/test_mg_cugraph_loader.py → ...ts/loader/test_dask_neighbor_loader_mg.py
diff --git a/...cugraph_pyg/tests/test_cugraph_sampler.py → ...h_pyg/tests/sampler/test_sampler_utils.py b/...cugraph_pyg/tests/test_cugraph_sampler.py → ...h_pyg/tests/sampler/test_sampler_utils.py
@@ -17,7 +17,7 @@
 import pytest
 
 from cugraph_pyg.data import DaskGraphStore
-from cugraph_pyg.sampler.cugraph_sampler import (
+from cugraph_pyg.sampler.sampler_utils import (
     _sampler_output_from_sampling_results_heterogeneous,
 )
 
@@ -191,9 +191,3 @@ def test_neighbor_sample_mock_sampling_results(abc_graph):
     assert out.num_sampled_edges[("A", "ab", "B")].tolist() == [3, 0, 1, 0]
     assert out.num_sampled_edges[("B", "ba", "A")].tolist() == [0, 1, 0, 1]
     assert out.num_sampled_edges[("B", "bc", "C")].tolist() == [0, 2, 0, 2]
-
-
-@pytest.mark.skipif(isinstance(torch, MissingModule), reason="torch not available")
-@pytest.mark.skip("needs to be written")
-def test_neighbor_sample_renumbered():
-    pass
diff --git a/...h_pyg/tests/mg/test_mg_cugraph_sampler.py → ...yg/tests/sampler/test_sampler_utils_mg.py b/...h_pyg/tests/mg/test_mg_cugraph_sampler.py → ...yg/tests/sampler/test_sampler_utils_mg.py
@@ -17,7 +17,7 @@
 import pytest
 
 from cugraph_pyg.data import DaskGraphStore
-from cugraph_pyg.sampler.cugraph_sampler import (
+from cugraph_pyg.sampler.sampler_utils import (
     _sampler_output_from_sampling_results_heterogeneous,
 )