Vh!hdZddlZddlZGddZGddZGddZGd d Zy) z% Tests for Group Centrality Measures Nc<eZdZdZdZdZdZdZdZdZ dZ y ) TestGroupBetweennessCentralitycztjd}dg}tj||ddd}d}||k(sJy)D Group betweenness centrality for single node group NFweight normalized endpoints@nx path_graphgroup_betweenness_centralityselfGCbb_answers _/home/dcms/DCMS/lib/python3.12/site-packages/networkx/algorithms/centrality/tests/test_group.py"test_group_betweenness_single_nodezATestGroupBetweennessCentrality.test_group_betweenness_single_node sG MM!  C  + + q%5 H}}cztjd}dg}tj||ddd}d}||k(sJy)rrrNFTr g@rrs r%test_group_betweenness_with_endpointszDTestGroupBetweennessCentrality.test_group_betweenness_with_endpointssG MM!  C  + + q%4 H}}rc|tjd}ddg}tj||ddd}d}||k(sJy) ze Group betweenness centrality for group with more than 1 node and normalized rrNTFr g?rrs r!test_group_betweenness_normalizedz@TestGroupBetweennessCentrality.test_group_betweenness_normalized#sI MM!  F  + + q$% H}}rctjd}gdgdg}tj||dd}ddg}||k(sJy) 9 Group betweenness centrality value of 0 )rrrrrNFr r r r cycle_graphrrs r%test_two_group_betweenness_value_zerozDTestGroupBetweennessCentrality.test_two_group_betweenness_value_zero0sG NN1   "  + +Aq% P:H}}rcztjd}gd}tj||dd}d}||k(sJy)r!r#r$NFr%r&r'rs r!test_group_betweenness_value_zeroz@TestGroupBetweennessCentrality.test_group_betweenness_value_zero:s> NN1    + +Aq% PH}}rctjd}|jdddg}tj||dd}d}||k(sJy)zF Group betweenness centrality in a disconnected graph rrrNFr%r&)rr remove_edgerrs r)test_group_betweenness_disconnected_graphzHTestGroupBetweennessCentrality.test_group_betweenness_disconnected_graphDsN MM!  a C  + +Aq% PH}}rctjtj5tjtj dgddddy#1swYyxYw)J Node(s) in C not in graph, raises NodeNotFound exception r)r"N)pytestraisesr NodeNotFoundrrrs r(test_group_betweenness_node_not_in_graphzGTestGroupBetweennessCentrality.test_group_betweenness_node_not_in_graphOsF]]2?? + I  + +BMM!,rrrr s+    Irrc~eZdZejdZejdZdZdZdZ dZ dZ dZ d Z y ) TestProminentGroupnumpypandasctjd}d}tj||dd\}}ddg}}||k(r||k(sJy)z1 Prominent group for single node rrFr r @r;Nrrprominent_grouprrkrgrg_answers r test_prominent_group_single_nodez3TestProminentGroup.test_prominent_group_single_nodelsQ MM!  !!!Q5EJ1 1#(H}h..rctjd}d}tj||ddg\}}ddg}}||k(r||k(sJy)z4 Prominent group without some nodes rrFr;)r rr NrJrLs rtest_prominent_group_with_cz.TestProminentGroup.test_prominent_group_with_cvsS MM!  !!!Q5QC@1 1#(H}h..rctjd}d}tj||dd\}}dddg}}||k(r||k(sJy)zH Prominent group with normalized result, with endpoints r"r;TrH333333?rNrr(rKrLs r)test_prominent_group_normalized_endpointszz;TestProminentGroup.test_group_betweenness_directed_weighteds JJL 1a " 1a " 1a " 1a " 1a " 1a " 1a " !!!QxEJ1 1a&(H}h..rctjd}d}tj||ddd\}}dddg}}||k(r||k(sJy) zD Group betweenness centrality in a greedy algorithm r"r;T)r r greedyrTr#rNrUrLs r%test_prominent_group_greedy_algorithmz8TestProminentGroup.test_prominent_group_greedy_algorithmsV NN1  !!!Q44PTU1 1a&(H}h..rN)r?r@rAr3 importorskipnppdrPrRrVrXr[r>r_rBrrrDrDhsI   W %B   X &B/// /</"/rrDc$eZdZdZdZdZdZy)TestGroupClosenessCentralityctjd}tj|dg}tj|d}||k(sJy)zD Group closeness centrality for a single node group rrN)rrgroup_closeness_centralitycloseness_centralityrrcc_answers r test_group_closeness_single_nodez=TestGroupClosenessCentrality.test_group_closeness_single_nodesD MM!   ) )!aS 1**1a0H}}rctj}|jgdtj|ddg}d}||k(sJy)zE Group closeness centrality for a disconnected graph )rr;rr1rr;rN)rGraphadd_nodes_fromrfrhs r!test_group_closeness_disconnectedz>TestGroupClosenessCentrality.test_group_closeness_disconnectedsD HHJ &  ) )!aV 4H}}rcptjd}tj|ddg}d}||k(sJy)zV Group closeness centrality for a group with more than 1 node r1rr;N)rrrfrhs r"test_group_closeness_multiple_nodez?TestGroupClosenessCentrality.test_group_closeness_multiple_nodes9 MM!   ) )!aV 4H}}rctjtj5tjtj dgddddy#1swYyxYw)zJ Node(s) in S not in graph, raises NodeNotFound exception rr#r"r2N)r3r4rr5rfrr6s r&test_group_closeness_node_not_in_graphzCTestGroupClosenessCentrality.test_group_closeness_node_not_in_graphsF]]2?? + G  ) )"--*:I F G G Gr8N)r?r@rArkrorqrtrBrrrdrdsGrrdc*eZdZdZdZdZdZdZy)TestGroupDegreeCentralityctjd}tj|dg}tj|d}||k(sJy)zA Group degree centrality for a single node group r1rN)rrgroup_degree_centralitydegree_centralityrrdd_answers r(test_group_degree_centrality_single_nodezBTestGroupDegreeCentrality.test_group_degree_centrality_single_nodesG MM!   & &q1# .''*1-H}}rctj}|jgd|jgdtj|ddg}d}||k(sJy)zQ Group degree centrality for group with more than 1 node rr;rr1rr#r"r2)rr;)rr)rr#)rr")rr2)r;r)r;r1)r;rrr;N)rrmrnadd_edges_fromrxrzs r*test_group_degree_centrality_multiple_nodezDTestGroupDegreeCentrality.test_group_degree_centrality_multiple_nodesX HHJ 12  L   & &q1a& 1H}}rctj}|jgd|jgdtj|ddg}d}||k(sJy)z9 Group in-degree centrality in a DiGraph rrrr;rN)rr<rnrgroup_in_degree_centralityrzs rtest_group_in_degree_centralityz9TestGroupDegreeCentrality.test_group_in_degree_centralitysX JJL 12  L   ) )!aV 4H}}rctj}|jgd|jgdtj|ddg}d}||k(sJy)z: Group out-degree centrality in a DiGraph rrrr;N)rr<rnrgroup_out_degree_centralityrzs r test_group_out_degree_centralityz:TestGroupDegreeCentrality.test_group_out_degree_centralitysX JJL 12  L   * *1q!f 5H}}rctjtj5tjtj dgddddy#1swYyxYw)zA Node(s) in S not in graph, raises NetworkXError rrsN)r3r4r NetworkXErrorrxrr6s r.test_group_degree_centrality_node_not_in_graphzHTestGroupDegreeCentrality.test_group_degree_centrality_node_not_in_graphsH]]2++ , D  & &r}}Q'7 C D D Dr8N)r?r@rAr}rrrrrBrrrvrvs   Drrv)__doc__r3networkxrrrDrdrvrBrrrsE[[|M/M/`#G#GL7D7Dr