refac

rlratzel · Jan 16, 2021 · c4b7fc3 · c4b7fc3
1 parent a36ad6b
commit c4b7fc3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/cugraph/tests/test_betweenness_centrality.py b/python/cugraph/tests/test_betweenness_centrality.py
@@ -120,7 +120,7 @@ def calc_betweenness_centrality(
     G = None
     Gnx = None
 
-    G, Gnx = Fixture_params_dts_sml
+    G, Gnx = Fixture_params_dts
 
     assert G is not None and Gnx is not None
     if multi_gpu_batch:

diff --git a/python/cugraph/tests/test_ecg.py b/python/cugraph/tests/test_ecg.py
@@ -69,7 +69,7 @@ def test_ecg_clustering(graph_file, min_weight, ensemble_size):
     assert cu_score > (0.95 * golden_score)
 
 
-@pytest.mark.parametrize("graph_file", utils.DATASETS)
+@pytest.mark.parametrize("graph_file", DATASETS)
 @pytest.mark.parametrize("min_weight", MIN_WEIGHTS)
 @pytest.mark.parametrize("ensemble_size", ENSEMBLE_SIZES)
 def test_ecg_clustering_nx(graph_file, min_weight, ensemble_size):