Skip to content

Commit

Permalink
added test cases and verified that algorithm works for undirected gra…
Browse files Browse the repository at this point in the history
…phs (#2349)

closes #2239

Authors:
  - Don Acosta (https://github.com/acostadon)

Approvers:
  - Rick Ratzel (https://github.com/rlratzel)

URL: #2349
  • Loading branch information
acostadon authored Jul 7, 2022
1 parent 526fbfc commit 5dd0267
Showing 1 changed file with 10 additions and 5 deletions.
15 changes: 10 additions & 5 deletions python/cugraph/cugraph/tests/mg/test_mg_degree.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,17 +32,22 @@ def setup_function():

IS_DIRECTED = [True, False]

DATA_PATH = [(RAPIDS_DATASET_ROOT_DIR_PATH /
"karate-asymmetric.csv").as_posix(),
(RAPIDS_DATASET_ROOT_DIR_PATH /
"polbooks.csv").as_posix(),
(RAPIDS_DATASET_ROOT_DIR_PATH /
"email-Eu-core.csv").as_posix()]


@pytest.mark.skipif(
is_single_gpu(), reason="skipping MG testing on Single GPU system"
)
@pytest.mark.parametrize("directed", IS_DIRECTED)
def test_dask_mg_degree(dask_client, directed):

input_data_path = (RAPIDS_DATASET_ROOT_DIR_PATH /
"karate-asymmetric.csv").as_posix()
print(f"dataset={input_data_path}")
@pytest.mark.parametrize("data_file", DATA_PATH)
def test_dask_mg_degree(dask_client, directed, data_file):

input_data_path = data_file
chunksize = cugraph.dask.get_chunksize(input_data_path)

ddf = dask_cudf.read_csv(
Expand Down

0 comments on commit 5dd0267

Please sign in to comment.