ިsgQ dZddlZddlZddlZddlZej dZej dZejdej dZ de jd<ddl Z e jd d Zd Zd Zd ZdZej&j)ddddgdfddgdfddgdfddgdff ej&j)dddgfdZej&j)ddddgdfddgdfddgdfddgdff ej&j)dddgfd Zd!Zd"Zd#Zd$Zd%Zej&j)d&e j8e j:fd'Zej&j)d(d)d*gd*ffd+Zej&j)d(d)d*gd*ffd,Z ej&j)dgd-ddd.gfd/Z!d0Z"d1Z#ej&j)d2d3d4Z$ej&j)d5d6gd7gd8ggd9d7d:ggd;fd<Z%d=Z&d>Z'ej&jPd?Z)d@Z*dAZ+dBZ,dCZ-dDZ.dEZ/dFZ0dGZ1dHZ2dIZ3ej&j)dJdKgdLfdMZ4ej&j)dNdOgdPfdQZ5dRZ6dSZ7ej&j)dTdUdVZ8ej&j)dTdUdWZ9dXZ:dYZ;dZZrIG drawn_edgess r%!test_single_edge_color_undirectedrVFs[& aA(( r"X*K :: !6!6!8( CC Cr8ctjdtj}tj|tj|||}|D]2}t j j|j|r2Jy)zaTests ways of specifying all edges have a single color for edges drawn with FancyArrowPatchesr  create_usingrLN) rrNDiGraphrOrPrQr6rR get_edgecolor)r0r>rIrTrUfaps r%test_single_edge_color_directedr]`sr& abjj1A(( r"X*KDzz$$S%6%6%8(CCCDr8ctjdtj}tt |Dcic]}|||f }}dD]}tj |||}|D]2}t jj|j|r2Jtj ||ddg|}|D]2}t jj|j|r2Jtj ||gdd }|D]2}t jj|jd r2Jtj ||gd d }|D]2}t jj|jd r2Jtj ||gdd }t jj|d j|djsJ|D]2}t jj|jd s2Jtj ||gd d }t jj|d j|djsJt jj|dj|djsJ|D]2}t jj|jd s2Jycc}w)zIf edge_color is a sequence with the same length as edgelist, then each value in edge_color is mapped onto each edge via colormap.rrX)rrrErrrErEr0rJrE)rIr0rJrbrcr r`rH)rJrbre)r rr_rrErcr N) rrNrZr-lenrOrQr6rRr[)rTnrMecrUr\s r%$test_edge_color_tuple_interpretationri{s abjj1A#CFm ,1q!f9 ,C ,( B,,QC  BC::(():):)s r%.test_fewer_edge_colors_than_num_edges_directedrrs A  7 #C J((CJGK[)//**EFD Xzz$$S%6%6%8(CCCDr8cBtjdtj}tjt}d}tj |||}t ||ddD]5\}}tjj|j|r5Jy)z\Test that extra edge colors are ignored when there are more specified colors than edges.rrX)r?rlrmcraN) rrNrZrPrrOrnrQr6rRr[rps r%-test_more_edge_colors_than_num_edges_directedrvs abjj1A  7 #C%J((CJGK[*Sb/:D Xzz$$S%6%6%8(CCCDr8cHtjttjtddgdd}|j j }t |jdk(sJtjj|dddsJ|ddk(sJyNrJrbpurple皙?)rMrIr0alpharcru) rrOrrPrSsqueezerf get_pathsrQr6rR)edge_collectionrhs r%3test_edge_color_string_with_global_alpha_undirectedrs,,   W %&! O  " " $ , , .B ((* +q 00 0 :: CR( 33 3 b6S==r8c:tjtjtjtddgdd}t |dk(sJ|D]A}|j }tjj|dddsJ|ddk(rAJyrx) rrOrr;rPrfr[rQr6rR)rUr\rhs r%1test_edge_color_string_with_global_alpha_directedrs((   W %&! K { q      zz$$RWh777"v}}r8 graph_typectjd|}tt|Dcic]}|||f }}tj||}t |t r|d}|jdk(sJycc}w)z^Test the default linewidth for edges drawn either via LineCollection or FancyArrowPatches.rcrXrrEN)rrNr-rfrO isinstancer, get_linewidth)rrTrgrMrUs r%test_edge_width_default_valuersx aj1A#CFm ,1q!f9 ,C ,((C0K+t$!!n  $ $ &! ++ + -s A=) edgewidthr>r r r ctjd}tt|Dcic]}|||f }}tj|||}t|j dk(sJ|j |k(sJycc}w)Nrrr )rrNr-rfrOr}r)rr>rTrgrMrUs r%'test_edge_width_single_value_undirectedrs aA#CFm ,1q!f9 ,C ,((CyAK {$$& '1 ,, ,  $ $ &( 22 2 -s Bc(tjdtj}tt |Dcic]}|||f }}tj |||}t |dk(sJ|D]}|j |k(rJycc}w)NrrXrr )rrNrZr-rfrOr)rr>rTrgrMrUr\s r%%test_edge_width_single_value_directedrs abjj1A#CFm ,1q!f9 ,C ,((CyAK { q  /  "h.../ - Brdrbctj}tj|}d}tj||||}t |t j|D]\}}|j|k(rJy)N)?@g(@)rIr) rr;rrPrOrnrror)rIrTrMwidthsrUr\expected_widths r%test_edge_width_sequencersx A  1 C F((C(&QK"; 0GH5^  "n4445r8ctjdtj}tj|}tj||ddg}|Dcgc]}|j }}tj||ddgdd}|Dcgc]}|j }}t jj||sJy cc}wcc}w) zyTest that edge_vmin and edge_vmax properly set the dynamic range of the color map when num edges == len(edge_colors).r rXrrBrarzg?)r0 edge_vmin edge_vmaxN) rrNrZrPrOr[rQr6rR)rTrMrUe orig_colors scaled_colorss r%#test_edge_color_with_edge_vmin_vmaxr#s abjj1A  1 C((CQHEK.9:1??$:K:(( 3C:K1<<1Q__&Test default linestyle for edges drawn with FancyArrowPatches.rrXr solidNrrNrZr-rfrO get_linestyle)rTrgrMrUr\s r%%test_directed_edges_linestyle_defaultr3s abjj1A#CFm ,1q!f9 ,C ,((C0K { q  .  "g---. -s B style)dashed--rErErEc(tjdtj}tt |Dcic]}|||f }}tj |||}t |dk(sJ|D]}|j |k(rJycc}w)zTests support for specifying linestyles with a single value to be applied to all edges in ``draw_networkx_edges`` for FancyArrowPatch outputs (e.g. directed edges).rrXrr Nr)rrTrgrMrUr\s r%*test_directed_edges_linestyle_single_valuer?s abjj1A#CFm ,1q!f9 ,C ,((Cu=K { q  ,  "e+++, -r style_seqrrr)r-:r)rrrz-.chtjdtj}tt |Dcic]}|||f }}tj |||}t |dk(sJt |tj|D]\}}|j|k(rJycc}w)zTests support for specifying linestyles with sequences in ``draw_networkx_edges`` for FancyArrowPatch outputs (e.g. directed edges).rrXrr N) rrNrZr-rfrOrnrror)rrTrgrMrUr\rs r%&test_directed_edges_linestyle_sequencerUs abjj1A#CFm ,1q!f9 ,C ,((CyAK { q  +yy'AB, U  "e+++, -s B/cddlm}m}ddlm}t j t j}t j t j}t j|}t j|}t j||}t||sJt j||d}t|tsJt|dkDrt|d|sJt j||d}t||sJt j||d}t||sJt j||d}t|tsJt|dkDrt|d|sJyy)Nr)LineCollectionPathCollection)FancyArrowPatchTarrowsF)matplotlib.collectionsrrmatplotlib.patchesrr cubical_graphGraphrZ spring_layoutdraw_networkx_nodesrrOr,rf) rrrrTdGrMdposnodesedgess r%test_return_typesrmsEE2 "A  "** %B  1 C  B D  " "1c *E e^ ,, ,  " "2tD 9E eT "" " 5zA~%(O444  " "2tE :E e^ ,, ,  " "1d4 8E e^ ,, ,  " "2s4 8E eT "" " 5zA~%(O444r8c "tj}tj|}tj||gddddtj||gdddgdtj||d d tj||gd d d dtj||gdd d dtj||gddd dd di}d|d<d|d<d|d<d|d<d|d<d|d<d|d<d |d!<t d Dcic]}||dzdk(rd"nd}}tj |||d#$tj |||d#|%tj||dd&'tj||d(d)i*ycc}w)+NrrErcr r?i?)nodelistr r r{)rrrm)g?rrrBrBr)rr{)rJrbre)r r)rIrr{r0)rr)rr)rr)rrT)rIrmin_source_marginmin_target_marginrr0z$a$rz$b$rEz$c$rcz$d$r z$\alpha$rz$\beta$rz$\gamma$rz$\delta$rk) font_size)r font_colorF) edge_labelsrotaterz4-5r)rrrrrOr-draw_networkx_labelsdraw_networkx_edge_labels)rTrMlabelsrgr6s r%test_labels_and_colorsrs A  1 C 3#D  $ 1cC8  1   1   1 FF1IF1IF1IF1IF1IF1IF1IF1I5:1X >aA + >F >AsFb9AsFbVL  CT%H  Cfe_E ?sF ctj}tj\}}dddddd}tj||dgdd tj||d d gd tj ||dd|j dtjtjd|S)NrrrErrJr)rr)rrErcr ri rztab:blue)r rr irz tab:orangerr)r{rg)\(?off) r house_graphrsubplotsrrOmargins tight_layoutaxis)rTfigaxrMs r%test_house_with_colorsrs AllnGCFv* EC    3$!1cA6JJtHHUO Jr8ctj\}}tjt|tj ttj t|y)Nr)rrrdrawrrcircular_layout)rrs r% test_axesrs<llnGCGGG  "*<*s r%#test_individualized_font_attributesr"sE A  ! !# &BGG 67ggi@Aa!es,@67ggi@1c!w-!+,,@    __& !SXX]]+AJJL0AA0E   61 !e7zz$$Q[[]H===1=1$%555 6A@ sE= FA F ctj}tjj d}|j Dcgc]\}}||zdzdk(rdnd}}}|j Dcgc]\}}d|dz|dzzzdz}}}tj ||d|||jDcgc])}t|tjjs(|+}}t|j |D]\}} | jd|ddz|d dzzzdzk(sJt|dzr$tjjj n#tjjj"} t| j%| rJycc}}wcc}}wcc}w) Nrrcr-|>-[r(T)rr arrowstyle arrowsizerE)rrrrrrrrrrQpatchesrrnget_mutation_scalesum ArrowStyleBracketB CurveFilledBget_arrowstyle) rTruv arrowstyles arrowsizesfrrar>s r%#test_individualized_edge_attributesr"9s A  ! !# &BCD779M41aQUaK1,5$6MKM89 B1"AA &+BJBGGA"TkZP??$ 1ckk6Q6Q(RFAGGIv&81##%qtax!A$(/B)Cb)HHHH1vz KK " " + +''44  !**,h7778 NBs F3:F9)F?.F?ctjtd5tjt ddddy#1swYyxYw)NzReceived invalid argumentmatchbar)foo)pytestraises ValueErrorrrrr<r8r%test_error_invalid_kwdsr+Ls4 z)D E$ U#$$$s AA ctjgd}gd}tjtd5tj ||dddy#1swYyxYw)N)rJrrcrr rEr )rErcr z1arrowsize should have the same length as edgelistr$)r)rrZr(r)r*r)rTrs r%+test_draw_networkx_arrowsize_incorrect_sizer0QsM 34AI M ( Y'(((s AA!r)r(rr1cZtjgd}dddd}tj|||}t|trt j |n|}t||D]@\}}t|tjjsJ|j|k(r@Jy)NrJr-rbrrJrrrErc)rMr) rrZrOrr rrepeatrnrQrrr)rrTrMrr\r>s r%test_draw_edges_arrowsizer6Zs +,AF +C  " "1# CE/9)S/I   +yIUI.4 X#s{{::;;;%%'83334r8rrrrz<|-|>cFtjgd}dddd}tj|||}t|trt j |n|}tjjjtjjjtjjjd}t||D]J\}}t|tjjsJt|j||rJJy)Nr3rrJrr4)rMrr7)rrZrOrstrrr5rQrrrr CurveFilledABrnrr)rrTrMr arrow_objectsr\r>s r%test_draw_edges_arrowstyler<gs +,AF +C  " "1#* EE)3:s(C $ {{%%22kk$$--''55M UJ/I X#s{{::;;;#,,. h0GHHHIr8cdtjttj ddgy)Nr-r.r)rrrnparrayr<r8r%test_np_edgelistr@|s Wrxx0@'ABr8ctjd}ddd}tjtjd5tj ||dddy#1swYyxYw)Nr rrrJzhas no positionr$)rrNr(r) NetworkXErrorr)rTrMs r%*test_draw_nodes_missing_node_from_positionrCsT aA C r''/@ A' q#&'''s A  A)r)rrctj\}}tjdg}ddd}tj||||d}|j j ddddf}tj||||dd d}|j j ddddf}|d|dkDsJ|d |d ksJy) Test that there is a wider gap between the node and the start of an incident edge when min_source_margin is specified. This test checks that the use of min_{source/target}_margin kwargs result in shorter (more padding) between the edges and source and target nodes. As a crude visual example, let 's' and 't' represent source and target nodes, respectively: Default: s-----------------------------t With margins: s ----------------------- t rJrrrrrNrcr rrrrrE)rrrrZrO get_extentscorners) rrrrTrM default_patchdefault_extent padded_patch padded_extents r%)test_draw_edges_min_source_target_marginsrNs&llnGC F8A C**1cbZPQRSM"..088:3Q36BN))    L!,,.6681a@M  nQ/ // /  nQ/ // /r8c :tj\}}tjddg}dddd}tj||||}|Dcgc](}|j j ddd d f*}}tj||||d d gd d g }|D cgc](} | j j ddd d f*} } t|| D]0\}} td| d|| d |d kDsJ| d|dkr0Jycc}wcc} w)rErJrbrr)rcrr4rFNrcrbfrGzp=z, d=rE) rrrrZrOrHrIrnprint) rrrrTrMrJdrKrLprMs r%4test_draw_edges_min_source_target_margins_individualrUs<&llnGC FF#$AF +C**1cbZPMANOAammo--/!Q7ONO))  s)s) LAMM1Q]]_,,.ssAv6MMMNM21 Eqdmtad{{tad{{ PNs -D$-Dctj\}}tj}|j d|j ddtj |ddid}|j}|jdkDr|jdkDsJtj|tjy)zDEnsure that selfloop extent is non-zero when there is only one node.rrN) rrrrZadd_nodeadd_edgerOrHrheightdelaxesclose)rrrTpatchbboxs r%&test_nonzero_selfloop_with_single_noder^sllnGC AJJqMJJq!  " "1q&k 21 5E    D ::>dkkAo- -KKOIIKr8ctj\}}tjdtj}|j dd|j Dcic]}|||f }}tj||dgd}|j}|jdkDr|jdkDsJtj|tjycc}w)zfEnsure that selfloop extent is non-zero when only a single edge is specified in the edgelist. rcrXrErrrN) rrrrNrZrXrrOrHrrYrZr[)rrrTrgrMr\r]s r%2test_nonzero_selfloop_with_single_edge_in_edgelistr`s llnGC abjj1AJJq!gg &1q!f9 &C &  " "1cVH =a @E    D ::>dkkAo- -KKOIIK 's C!cgd}gd}d}tjjj|||}t |dddf|k(sJy)zfTest apply_alpha when there is a mismatch between the number of supplied colors and elements. r4rkrNru)rdrawingnx_pylab apply_alphaall)r colorlistr{ rgba_colorss r%test_apply_alpharh sMHI E**%%11)UHMK {1b5!U* ++ +r8cZddl}ddl}tjd}tjdtj}|Dcic]}|||f }}||fD]f}tj ||d}t |t |jk(sJt|dtjjrfJ||fD]@}tj ||d}t|tjjr@Jtj ||}t|tjjsJtj ||}t |t jk(sJt|dtjjsJycc}w)a/ The `arrows` keyword argument is used as a 3-way switch to select which type of object to use for drawing edges: - ``arrows=None`` -> default (FancyArrowPatches for directed, else LineCollection) - ``arrows=True`` -> FancyArrowPatches - ``arrows=False`` -> LineCollection rNr rXTrF)rrrrNrZrOrfrrrQrrrr)rUGrrgrMrTrs r%*test_draw_edges_toggling_with_arrows_kwargrksl" q B qrzz 2B !1q!f9 !C !"XA&&q#d;5zS\)))%(CKK$?$?@@@A "XA&&q#e<%!?!?@@@A  " "2s +E eS__;; << <  " "2s +E u:QWW %% % eAh ; ; << <% "s F( drawing_funcc"ddl}tjd}tj \}}|||t d|j DsJ|jrJtj|tjy)Nrr rc3dK|](}t|tjj*ywr:rrQrr.0rts r% z?test_draw_networkx_arrows_default_undirected..As!Uz!S__;;<U.0) rrrNrranyrrrZr[rlrrTrrs r%,test_draw_networkx_arrows_default_undirectedrv:sd! aAllnGCr UbnnU UU Uzz>KKOIIKr8cBddl}tjdtj}tj \}}|||t d|jDrJ|jsJtj|tjy)Nrr rXrc3dK|](}t|tjj*ywr:rorps r%rrz=test_draw_networkx_arrows_default_directed..Ns&:; 1coo445rs) rrrNrZrrrtrrrZr[rus r%*test_draw_networkx_arrows_default_directedryGsw! abjj1AllnGCr?A~~  :::KKOIIKr8c"tjd}|jddtj \}}tj |ddg||j rJtj|tjy)Nr rrJrb)rIr) rrNrXrrrrrZr[)rTrrs r%test_edgelist_kwarg_not_ignoredr{Vsb aAJJq!llnGCGGA(R0zz>KKOIIKr8)rTexpected_n_edgesrcc @tgdD])\}\}}|j||t|dz d+|Dcic]}|||f }}dddggdfD]B}tj|||tj|||}t ||k(rBJycc}w) zGDraws edges correctly for 3 types of graphs and checks for valid lengthrJrJrJr-r rcweight arc3,rad=0.1)rr arc3,rad=0.2connectionstyleN) enumeraterXroundrrOrf) rTr|irrrgrM conn_stylers r%2test_draw_networkx_edges_multiedge_connectionstyleras ?@1 6Aq 1aa!eQ 01 1q!f9 C   (8/ q#zB''3 K6{..../ !s Bc tgdD])\}\}}|j||t|dz d+|Dcic]}|||f }}tj||gd}dddggdfD]Z}tj |||} t | |k(sJ| jD]} | jjd k(rJ\y cc}w) zXDraws labels correctly for 3 types of graphs and checks for valid length and class namesr~r rcr)rrrrrrCurvedArrowTextN) rrXrrrOrrfvalues __class____name__) rTr|rrrrgrMrr text_itemstis r%8test_draw_networkx_edge_labels_multiedge_connectionstylerus ?@1 6Aq 1aa!eQ 01 1q!f9 C  # # 3 PF  (8> 11!S*U :"2222##% >B<<((,== == >> !s C c,tj}|jddd|jdddtj|d}|Dcic]}|||f }}tj|||ddg }t |d k(sJycc}w) NrrEr(rrrrr)rrrc)r MultiGraphrXget_edge_attributesrrf)rTrrgrMrs r%'test_draw_networkx_edge_label_multiedgers AJJq!BJJJq!BJ((H5K 1q!f9 C --  '8 J z?a   !s Bctjd}|jDcic]}|||f }}tj||iik(sJycc}w)zORegression test for draw_networkx_edge_labels with empty dict. See gh-5372.r rN)rrNrr)rTrgrMs r%(test_draw_networkx_edge_label_empty_dictrsS aAgg &1q!f9 &C & ' '3B ?2 EE E 's Actj\}}gd}gd}tj|}|jDcic]}|||f }}tj |||||t |jdk(sJtj|ddtjddgz}t|j|dd|D]W\}} } |jj| sJtjj|j!| rWJtj#|tj%ycc}w) ztWhen an edgelist is supplied along with a sequence of colors, check that the self-loops have the correct colors.)r/rbrerr)rcrc)pinkcyanr r@rHgreen)rrIr0r Nrr)rrrrrrOrfrr>r?rnget_pathcontains_pointrQr6rRr[rZr[) rrrI edge_colorsrTrgrM sl_pointsr\clrslps r%3test_draw_networkx_edges_undirected_selfloop_colorsrs#llnGC?HCK Agg &1q!f9 &C &1cb8 T rzz?a   "#'"((As8*<>>?KKOIIK# 's Efap_only_kwargrrrrr(rc^tjd}|Dcic]}|||f }}tj\}}t |j d}t jtd|d5tj||fd|i|dddtj5tjdtj||f|d d |dddtj|tjycc}w#1swYxYw#1swYDxYw) zUsers should get a warning when they specify a non-default value for one of the kwargs that applies only to edges drawn with FancyArrowPatches, but FancyArrowPatches aren't being used under the hood.r rz The z# keyword argument is not applicabler$rNerrorT)rr)rrNrrr,keysr(warns UserWarningrOwarningscatch_warnings simplefilterrZr[)rrTrgrMrr kwarg_names r%1test_user_warnings_for_unused_edge_drawing_kwargsrs aA 1q!f9 C llnGCn))+,Q/J Xj\1TU @ q#?"??@  "Mg& q#L"TL^LM KKOIIK# ! @@ MMs D:D00D#D #D,draw_fnc8tj\}}tjd}t j d5}|||dddt dk(sJtj|tjy#1swYCxYw)NrT)recordrr) rrr cycle_graphrrrfrZr[)rrrrTws r%*test_no_warning_on_default_draw_arrowstylerstllnGC qA   -b q6Q;;KKOIIK s  BB hide_ticksTmethodc~tjd}|jDcic]}|||f }}tj \}}||||||j |j fD] }t|j|k7r Jtj|tjycc}w)Nr )rMrr) rrNrrrxaxisyaxisboolget_ticklabelsrZr[)rrrTrgrM_rrs r%test_hide_ticksrs aAgg &1q!f9 &C & LLNEAr 1#"4288$9D'')*j8889KKOIIK 's B:)R__doc__rrrr( importorskiprQr>userrcParamsnetworkxr barbell_graphrr&r.r7r=mark parametrizerVr]rirrrvrrrrZrrrrrrrrrrmpl_image_comparerrrrrrrrr"r+r0r6r<r@rCrNrUr^r`rhrkrrrvryr{rrrrrrrrrrrrOrrrr<r8r%rs^2  f,'V! f-.# ] "  1a 0 *  #  H%   f  dVH%56D7 D #  H%   f  dVH%56 D7  D6FrDD  "**'=>,?, a33 a//  55 = .  , ,     s    ,  ,549Fz2N  K,E6$6.8&$ (r<&89 4: 4/E'FGIHI(C 'z2)03)0\z2*3*Z$(,=D"''23C3C)DE F "''23C3C)DE F bjjlA!,r/@!.DE /  / bjjlA!,r/@!.DE> >(  F: s b N+ b! b!   2RWWb.>.>$?@ A t}5   $$       6 r8