_-Ph<dZddlZddlmZddlmZddlm Z m Z m Z ddl m Z mZddlmZmZmZmZmZddlmZdd lmZd Zd Zd ZGd de ZGdde ZGdde ZdZdZdZ dZ!dS)zTest graph routings.N)sparse) bellman_ford)TestCase assert_equalassert_array_almost_equal)poisson load_example)maximal_independent_setvertex_coloringr lloyd_clusterconnected_components)bellman_ford_reference)amg_corectj|}|j|jk}|j||_|j||_|j||_d|jdd<|SN)r coo_matrixrowcoldata)Gmasks V/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyamg/tests/test_graph.pycanonical_graphrs\ !A 5AE>D E$KAE E$KAE VD\AFAF111I Hct|}|jdkr3||j||jzdksJ|||zzdksJdS)Nrr)rnnzrrmaxmin)rmiss r assert_is_misr!suA uqyyAE SZ',,..!3333 !C%K    ! # # # # # #rct|}||j||jksJt j|dksJdS)Nr)rrrallnpbincount)rcs rassert_is_vertex_coloringr''seA aeH!% % % ' '''' KNNQ  # # % %%%%%%rc2eZdZdZdZdZdZdZdZdS) TestGraphc*g}tjdtdD]h}tjdddk}|t ||jzti|tj d|tj d|tj d|tj dtj dz |td |td |td |td d D]*}|t|d+d|D}||_dS)Nig&?)rr)r.)r,r,r.r+)r+r+) r0)r+r+r+)airfoilbarknotAc,g|]}t|Sr.0rs r z#TestGraph.setUp..Ds 333##333r)r$randomseedrangerandappendrTastypefloatzerosoneseyerr cases)selfrF_ir4names rsetUpzTestGraph.setUp1s y!!!(( A AB q!$$s*A LLQS1188?? @ @ @ @ RXf%%&&& RXf%%&&& RXf%%&&& RWV__rvayy0111 WT]]### WV__%%% WX&&''' WY''(((. 2 2D LLd++C0 1 1 1 133U333 rctttjdddgdD]-}|jD]#}t||}t ||$.|jD]b}dD]]}t||}|dkr4|tj|jdz|z}t|}t ||^cdS)Nr.rserialparallelalgo)rr.)kr)rr r$rErFr!shaper)rGrPrr rSs rtest_maximal_independent_setz&TestGraph.test_maximal_independent_setHs,RVAYY77!Q@@@* & &DZ & &-ad;;;a%%%% & & &A! & &-a1555q55RVAGAJ///!3A'**Aa%%%%  & & &rc tttjddgtttjdgdtt ttjdgddD]-}|jD]#}t||}t||$.dS) NrrrQ)rrr)rQrQ)rrr.MISJPLDFmethod)rr r$rEsortedrDrFr'rGr\rr&s rtest_vertex_coloringzTestGraph.test_vertex_coloringYs_RVAYY//!555_RVAYY//;;;VOBGFOO<<==yyyIII* 0 0FZ 0 0#Af555)!Q//// 0 0 0rctjd|jD]R}tj|j|_|jd}t|dz t|dz |dz |fD]}||jdks|dkrtj |d|}t||\}}t||\}}t||t|||jddkrWt|j } | dd|fd } t|| TdS) z4Test pile of cases against reference implementation.iar r.rN)csgraph)axis)r$r;r<rFr>rrrTint permutationrrrbellman_ford_scipyr@rravel) rGrNn_seedsseedsD_resultS_result D_expected S_expectedDD_scipys rtest_bellman_fordzTestGraph.test_bellman_forddsp z""" A AAY^^AE**AF A"IIs1R4yy!A#q9 A AQWQZ''7Q;; --a00':%1!U%;%;"()?5)I)I& J)(J???)(J???71:??*13777A5koo1o55;;==G-h@@@! A  A Arc tjddgddgddgddgddgddgddgddgg}tjgdt}tj||dddf|dddfff}tjgdtjd d d d gtjtjd tjtjgtjd d d d gtjd dd d gg}t dD]V}t||g\}}t|||t||g\}}t|||WdS)NrrRrQrr.)r.rr.rrRr+rQrdtype)?@@rzrwrzg@rxryg@r+) r$arrayrBrrinfr=rrr)rGEdgeswrdistances_FROM_seedr<distancenearests rtest_bellman_ford_referencez%TestGraph.test_bellman_ford_reference~s1a&a&a&a&a&a&a&a&"## H---U ; ; ;  q5A;aaad "<= > > h(H(H(H)+RR(H)+RVRV(L)+RR(H)+RR(H (JKK !HH > >D 6q4& A A Hg #6t#< = = = ,Q 7 7 Hg #6t#< = = = =  > >rctjd|jD]V}tj|j|_dD](}||jdkrt||\}}})WdS)Nltr/r r$r;r<rFr>rrrTr rGrrk distancesclusterscenterss rtest_lloyd_clusterzTestGraph.test_lloyd_clusters z""" I IAY^^AE**AF I IQWQZ''/>>0 I I I I Irr)c&eZdZdZdZdZdZdS)TestComplexGraphc^g}tjdtdD]h}tjdddk}|t ||jztid|D}||_ dS)Nlkor+r,r-cRg|]$}t|dt|zz%S?r7r8s rr:z*TestComplexGraph.setUp..s2KKK##D););$;;KKKr) r$r;r<r=r>r?rr@rArBrF)rGrFrHr4s rrJzTestComplexGraph.setUps z"""(( A AB q!$$s*A LLQS1188?? @ @ @ @KKUKKK rctttjdddgdD]-}|jD]#}t||}t ||$.dS)Nr.rrLrO)rr r$rErFr!)rGrPrr s rrUz-TestComplexGraph.test_maximal_independent_sets{,RVAYY77!Q@@@* & &DZ & &-ad;;;a%%%% & & &rcfdD]-}|jD]#}t||}t||$.dS)NrWr[)rFr r'r^s rr_z%TestComplexGraph.test_vertex_coloringsV* 0 0FZ 0 0#Af555)!Q//// 0 0 0rcPtjd|jD]~}tj|jdtj|jzz|_dD](}||jdkrt||\}}})dS)Ni$}rr/rrrs rrz#TestComplexGraph.test_lloyd_clusters z""" I IAY^^AE**T")..2G2G-GGAF I IQWQZ''/zJtest_connected_components..array_to_set_of_sets..b555QIaLL555rset enumerateaddvaluesarrrqins rarray_to_set_of_setsz7test_connected_components..array_to_set_of_sets[uAXX  uu!!#  1! 55!((**555 5rr?rrr rrreference_connected_componentsrFrresultrexpecteds rtest_connected_componentsr s E LL"LLL$0LL$0LL$0LL$233444  LL"LLL$0LL$0LL$0LL$233444  LL"LLL$0LL$0LL$0LL$233444  LL"LLL$0LL$0LL$0LL$233444 LL"$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6 $899::: LL"$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6 $899::: LL"$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6 $899:::''%a((VZZ\\1%%% 6 6 6&%f--1!44VX&&&&#''rc g}|tjgdgdgdgdg|tjgdgdgdgdg|tjgdgdgdgdg|tjgdgdgdgdg|tjgdgd gd gd gd gd g|tjgdgdgd gd gd gd g|tjgdgdgd gdgd gd gd|D}|D]`}t|}t |dd}||}t |}t ||adS)Nrrrrrrrrrrrrrrrrrcg|] }|d|zz Srr6r8s rr:z5test_complex_connected_components..s % % %!QtAvX % % %rrci}t|D]}t||<t|D] \}}|||!d|DS)rc,h|]}t|Sr6rrs rrzRtest_complex_connected_components..array_to_set_of_sets..rrrrs rrz?test_complex_connected_components..array_to_set_of_setsrrrrs r!test_complex_connected_componentsrjs E LL"LLL$0LL$0LL$0LL$233444  LL"LLL$0LL$0LL$0LL$233444  LL"LLL$0LL$0LL$0LL$233444  LL"LLL$0LL$0LL$0LL$233444 LL"$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6 $899::: LL"$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6 $899::: LL"$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6$6 $899::: & %u % % %E''%a((VZZ\\1%%% 6 6 6&%f--1!44VX&&&&#''rc4|}|jd}fdt}t}t|D]D}||vr>t}|||||t |E|S)Nrc||vrc|||||j|j||j|dzD]}||||dSdSr)rrr)rr componentvisitedjDFSs rrz+reference_connected_components..DFSs G   MM!    KKNNNYqx{18AaC=89 . .Aq)W----   . .r)tocsrrTrr=rr)rrjr componentsrrrs @rrrs  A  A.....eeGJ 1XX11 G  I C1i ) ) ) NN9Y// 0 0 0 r)"__doc__numpyr$scipyrscipy.sparse.csgraphrrh numpy.testingrrr pyamg.galleryrr rr r r r pyamg.graph_refrpyamgrrr!r'r)rrrrrrr6rrrsCCCCCCKKKKKKKKKK////////LLLLLLLLLLLLLL222222    $$$&&&pIpIpIpIpIpIpIpIf&I&I&I&I&Ix&I&I&IR7373737373(737373t0004G'G'G'TJ'J'J'\r