Vh7ddlZejdejdddlZddlmZGddZGddZy) Nnumpyscipy)node_classificationc<eZdZdZdZdZdZdZdZdZ dZ y ) TestHarmonicFunctionctjd}d}d|jd|<d|jd|<tj||}|ddk(sJ|ddk(sJ|d dk(sJ|ddk(sJy NlabelArB label_namenx path_graphnodesrharmonic_functionselfGr predicteds b/home/dcms/DCMS/lib/python3.12/site-packages/networkx/algorithms/tests/test_node_classification.pytest_path_graphz$TestHarmonicFunction.test_path_graph s MM!  !$ :!$ :'99! S |s"""|s"""|s"""|s"""ctjtj5tjd}t j |dddy#1swYyxYwNr )pytestraisesr NetworkXErrorrrrrrs rtest_no_labelsz#TestHarmonicFunction.test_no_labelssE ]]2++ , 5 a A  1 1! 4 5 5 5 +AA!ctjtj5tj}t j |dddy#1swYyxYwN)r!r"rr#Graphrrr$s r test_no_nodesz"TestHarmonicFunction.test_no_nodessB ]]2++ , 5 A  1 1! 4 5 5 5 *AA c tjtj5tj}|j d|j dt j|dddy#1swYyxYwNrr)r!r"rr#r)add_noderrr$s r test_no_edgesz"TestHarmonicFunction.test_no_edges sX ]]2++ , 5 A JJqM JJqM  1 1! 4  5 5 5 A A99Bc~tjtj5tj}|j dd|j dd|j ddd}d|j d|<d|j d|<tj|dddy#1swYyxYwNrrrrr r r r!r"rNetworkXNotImplementedDiGraphadd_edgerrrrrrs r test_digraphz!TestHarmonicFunction.test_digraph' ]]244 5 5 A JJq!  JJq!  JJq!  J%(AGGAJz "%(AGGAJz "  1 1! 4 5 5 5 BB33B<ctjd}d}d|jd|<tj||}|ddk(sJ|ddk(sJ|ddk(sJ|ddk(sJy Nr r r rrrrrrrs rtest_one_labeled_nodez*TestHarmonicFunction.test_one_labeled_node2s MM!  !$ :'99! S |s"""|s"""|s"""|s"""rctj}d}tj||}t t |D]}|||j ||k(rJy)Nclubr)rkarate_club_graphrrrangelenrrrrris rtest_nodes_all_labeledz+TestHarmonicFunction.test_nodes_all_labeled<s]  " '99! S s1v :AQ<1771:j#99 99 :rctj}d}hd}|D]}|j||=tj||}t t t||z }|D]}|||j||k(rJy)Nr?>rrrrr r)rr@rrrsetrArB)rrr label_removedrDrlabel_not_removeds r"test_labeled_nodes_are_not_changedz7TestHarmonicFunction.test_labeled_nodes_are_not_changedCs  " 0  'A :& ''99! S c!f .>" :AQ<1771:j#99 99 :rN) __name__ __module__ __qualname__rr%r*r/r8r=rErMrrrr s* #5 5 5 5#: :rrc6eZdZdZdZdZdZdZdZdZ y) TestLocalAndGlobalConsistencyctjd}d}d|jd|<d|jd|<tj||}|ddk(sJ|ddk(sJ|d dk(sJ|ddk(sJyr rrrrlocal_and_global_consistencyrs rrz-TestLocalAndGlobalConsistency.test_path_graphPs MM!  !$ :!$ :'DD * |s"""|s"""|s"""|s"""rctjtj5tjd}t j |dddy#1swYyxYwr )r!r"rr#rrrVr$s rr%z,TestLocalAndGlobalConsistency.test_no_labels]sI ]]2++ , @ a A  < rasE GG3B:B:J=:=:r