From 21b689b309d92ff0924c31eaa405c9e8dcce9591 Mon Sep 17 00:00:00 2001 From: Rick Ratzel Date: Wed, 28 Aug 2024 11:54:26 -0500 Subject: [PATCH] Updates test to ensure a query string is created using integer vertex IDs (as required). --- .../cugraph/tests/data_store/test_property_graph.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/python/cugraph/cugraph/tests/data_store/test_property_graph.py b/python/cugraph/cugraph/tests/data_store/test_property_graph.py index da5608e0193..50e11a64e60 100644 --- a/python/cugraph/cugraph/tests/data_store/test_property_graph.py +++ b/python/cugraph/cugraph/tests/data_store/test_property_graph.py @@ -2576,9 +2576,12 @@ def bench_extract_subgraph_for_rmat(gpubenchmark, rmat_PropertyGraph): scn = PropertyGraph.src_col_name dcn = PropertyGraph.dst_col_name + # Build a query string to extract a graph with only specific edges based on + # the integer vertex IDs. Other edge and/or vertex properties can be + # included in the query as well. verts = [] for i in range(0, 10000, 10): - verts.append(generated_df["src"].iloc[i]) + verts.append(int(generated_df["src"].iloc[i])) selected_edges = pG.select_edges(f"{scn}.isin({verts}) | {dcn}.isin({verts})") gpubenchmark( @@ -2618,9 +2621,12 @@ def bench_extract_subgraph_for_rmat_detect_duplicate_edges( scn = PropertyGraph.src_col_name dcn = PropertyGraph.dst_col_name + # Build a query string to extract a graph with only specific edges based on + # the integer vertex IDs. Other edge and/or vertex properties can be + # included in the query as well. verts = [] for i in range(0, 10000, 10): - verts.append(generated_df["src"].iloc[i]) + verts.append(int(generated_df["src"].iloc[i])) selected_edges = pG.select_edges(f"{scn}.isin({verts}) | {dcn}.isin({verts})")