@@ -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