Skip to content

Commit f37241b

Browse files
Reduce downloaded test data in CI (#4951)
Attempting to improve testing times in CI. The tests run in CI do not require all of the test data that is downloaded. We should be able to skip calling `get_test_data.sh` since we're not running any of the benchmark tests. This PR also fixes a few missing annotations in the python tests discovered during test. There is still more work to be done, but that will be in a follow-on PR. Authors: - Chuck Hastings (https://github.com/ChuckHastings) Approvers: - Gil Forsyth (https://github.com/gforsyth) - Rick Ratzel (https://github.com/rlratzel) URL: #4951
1 parent 6618212 commit f37241b

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

ci/test_python.sh

-3
Original file line numberDiff line numberDiff line change
@@ -41,9 +41,6 @@ export LD_PRELOAD="${CONDA_PREFIX}/lib/libgomp.so.1"
4141

4242
# RAPIDS_DATASET_ROOT_DIR is used by test scripts
4343
export RAPIDS_DATASET_ROOT_DIR="$(realpath datasets)"
44-
pushd "${RAPIDS_DATASET_ROOT_DIR}"
45-
./get_test_data.sh --benchmark
46-
popd
4744

4845
EXITCODE=0
4946
trap "EXITCODE=1" ERR

python/cugraph/cugraph/tests/data_store/test_property_graph_mg.py

+2
Original file line numberDiff line numberDiff line change
@@ -1504,6 +1504,7 @@ def test_renumber_by_type_only_default_type(dask_client):
15041504

15051505

15061506
@pytest.mark.slow
1507+
@pytest.mark.mg
15071508
@pytest.mark.parametrize("N", [1, 3, 10, 30])
15081509
def bench_add_edges_cyber(benchmark, dask_client, N):
15091510
from cugraph.experimental import MGPropertyGraph
@@ -1527,6 +1528,7 @@ def func():
15271528

15281529

15291530
@pytest.mark.slow
1531+
@pytest.mark.mg
15301532
@pytest.mark.parametrize("n_rows", [1_000_000])
15311533
@pytest.mark.parametrize("n_feats", [128])
15321534
def bench_get_vector_features(benchmark, dask_client, n_rows, n_feats):

python/cugraph/cugraph/tests/structure/test_graph_mg.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2022-2024, NVIDIA CORPORATION.
1+
# Copyright (c) 2022-2025, NVIDIA CORPORATION.
22
# Licensed under the Apache License, Version 2.0 (the "License");
33
# you may not use this file except in compliance with the License.
44
# You may obtain a copy of the License at
@@ -371,6 +371,7 @@ def test_graph_creation_edge_properties(dask_client, graph_file, edge_props):
371371
G.from_dask_cudf_edgelist(df, source="0", destination="1", **prop_keys)
372372

373373

374+
@pytest.mark.mg
374375
@pytest.mark.parametrize("directed", [True, False])
375376
@pytest.mark.parametrize("renumber", [True, False])
376377
@pytest.mark.parametrize("graph_file", datasets)
@@ -422,6 +423,7 @@ def test_graph_creation_properties(dask_client, graph_file, directed, renumber):
422423
assert_frame_equal(sG_edgelist_view, mG_edgelist_view, check_dtype=False)
423424

424425

426+
@pytest.mark.mg
425427
@pytest.mark.parametrize("directed", [True, False])
426428
@pytest.mark.parametrize("renumber", [True, False])
427429
@pytest.mark.parametrize("graph_file", datasets)

0 commit comments

Comments
 (0)