Skip to content

Commit 38c357f

Browse files
committed
Fix PropertyGraph MG tests
Updating these tests was missed in rapidsai#2449 Also, MG tests are finally working for me 🎉
1 parent f8e4aaa commit 38c357f

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

python/cugraph/cugraph/tests/mg/test_mg_property_graph.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -411,13 +411,14 @@ def test_property_names_attrs(dataset1_MGPropertyGraph):
411411
"""
412412
(pG, data) = dataset1_MGPropertyGraph
413413

414-
expected_vert_prop_names = ["merchant_id", "merchant_location",
415-
"merchant_size", "merchant_sales",
416-
"merchant_num_employees", "merchant_name",
417-
"user_id", "user_location", "vertical"]
418-
expected_edge_prop_names = ["user_id", "merchant_id", "volume", "time",
419-
"card_num", "card_type", "user_id_1",
420-
"user_id_2", "relationship_type", "stars"]
414+
# _VERTEX_ columns: "merchant_id", "user_id"
415+
expected_vert_prop_names = ["merchant_location", "merchant_size",
416+
"merchant_sales", "merchant_num_employees",
417+
"user_location", "merchant_name", "vertical"]
418+
# _SRC_ and _DST_ columns: "user_id", "user_id_1", "user_id_2"
419+
# Note that "merchant_id" is a property in for type "transactions"
420+
expected_edge_prop_names = ["merchant_id", "volume", "time", "card_num",
421+
"card_type", "relationship_type", "stars"]
421422

422423
# Extracting a subgraph with weights has/had a side-effect of adding a
423424
# weight column, so call extract_subgraph() to ensure the internal weight
@@ -576,6 +577,7 @@ def test_get_edge_data(dataset1_MGPropertyGraph):
576577
for d in ["transactions", "relationships", "referrals"]:
577578
for name in data[d][0]:
578579
expected_columns.add(name)
580+
expected_columns -= {'user_id', 'user_id_1', 'user_id_2'}
579581

580582
actual_columns = set(some_edge_data.columns)
581583

0 commit comments

Comments
 (0)