U 9v f¸kã @sˆdZddlZddlZddlZddlZe d¡Ze d¡Ze d¡e d¡Z de j d<ddl Z e   d d ¡Zd d „Zd d„Zdd„Zdd„Zej ddddgdfddgdfddgdfddgd ff ¡ej d!dd"gf¡d#d$„ƒƒZej ddddgdfddgdfddgdfddgd ff ¡ej d!dd"gf¡d%d&„ƒƒZd'd(„Zd)d*„Zd+d,„Zd-d.„Zd/d0„Zej d1e je jf¡d2d3„ƒZej d4d5d6gd6ff¡d7d8„ƒZej d4d5d6gd6ff¡d9d:„ƒZ ej d!d"d;d„ƒZ!d?d@„Z"dAdB„Z#ej dCdD¡dEdF„ƒZ$ej dGdHgdIgdJgdIdKdLgdIdKgdIdKdLdMgf¡dNdO„ƒZ%dPdQ„Z&ejj'dRdS„ƒZ(dTdU„Z)dVdW„Z*dXdY„Z+dZd[„Z,d\d]„Z-d^d_„Z.d`da„Z/ej dbdcdddedcgf¡dfdg„ƒZ0dhdi„Z1djdk„Z2ej dldm¡dndo„ƒZ3dpdq„Z4drds„Z5dtdu„Z6dvdw„Z7ej dxe j8e j9f¡dydz„ƒZ:ej dxe j8e j9f¡d{d|„ƒZ;d}d~„Zdƒd„„Z?ej d…d†dKidbdeid‡dˆid‰ddidŠddif¡d‹dŒ„ƒZ@dS)z,Unit tests for matplotlib drawing functions.éNÚ matplotlibZnumpyZPSzmatplotlib.pyplotFz text.usetexéécCsŠz^tjtjtjtjtjtj tj g}ddddœg}t   ||¡D]\}}|t f|Žt d¡q}sz8test_edge_color_tuple_interpretation..)©rrr1©rrr1r1©r#r6©r1é)r5r#©rPrrMr4)rrrLrr1rPrN) rr9rDr!Úlenr:r<r%r=rEr?)r@r8ÚecrArFrrrÚ$test_edge_color_tuple_interpretationystÿÿ ÿÿ ÿ  ÿ ÿ  ÿ rTcCsZt ¡}t t¡}d}tj|||d}t|t |¡ƒD]\}}tj   |  ¡|¡s6t ‚q6dS)zZTest that the edge colors are cycled when there are fewer specified colors than edges.)r+ÚgÚbrNN) rr(rr;r:ÚziprÚcycler<r%r=rEr?©r@r8Z edgecolorsrArFr*rrrÚ.test_fewer_edge_colors_than_num_edges_directed²s  rZcCsdtjdtjd}t t¡}d}tj|||d}t||dd…ƒD]\}}tj  |  ¡|¡s@t ‚q@dS)z\Test that extra edge colors are ignored when there are more specified colors than edges.rrC)r+rUrVÚcrNNéÿÿÿÿ) rr9rDr;rr:rWr<r%r=rEr?rYrrrÚ-test_more_edge_colors_than_num_edges_directed½s  r]cCsltjtt t¡ddgddd}| ¡ ¡}t| ¡ƒdks>t‚t j   |dd…d¡sXt‚|ddksht‚dS©Nr6rOZpurpleçš™™™™™É?)r8r5r#ÚalpharPr\) rr:rr;r>ZsqueezerRÚ get_pathsr?r<r%r=)Zedge_collectionrSrrrÚ3test_edge_color_string_with_global_alpha_undirectedÈsû rbcCsrtjt ¡t t¡ddgddd}t|ƒdks2t‚|D]6}| ¡}tj   |dd…d¡s\t‚|ddks6t‚q6dSr^) rr:rr(r;rRr?rEr<r%r=)rArFrSrrrÚ1test_edge_color_string_with_global_alpha_directedÖsûrcÚ graph_typecCsVtjd|d}dd„tt|ƒƒDƒ}t ||¡}t|tƒrB|d}| ¡dksRt‚dS)z^Test the default linewidth for edges drawn either via LineCollection or FancyArrowPatches.rPrCcSsi|]}|||f“qSrrrHrrrrKêsz1test_edge_width_default_value..rr1N) rr9r!rRr:Ú isinstancer Ú get_linewidthr?)rdr@r8rArrrÚtest_edge_width_default_valueås   rg)Ú edgewidthr*©rrrcCsXt d¡}dd„tt|ƒƒDƒ}tj|||d}t| ¡ƒdksDt‚| ¡|ksTt‚dS)NrcSsi|]}|||f“qSrrrHrrrrKúsz;test_edge_width_single_value_undirected..©r r)rr9r!rRr:rar?rf)rhr*r@r8rArrrÚ'test_edge_width_single_value_undirectedñs  rkcCsdtjdtjd}dd„tt|ƒƒDƒ}tj|||d}t|ƒdksFt‚|D]}| ¡|ksJt‚qJdS)NrrCcSsi|]}|||f“qSrrrHrrrrK sz9test_edge_width_single_value_directed..rjr)rr9rDr!rRr:r?rf)rhr*r@r8rArFrrrÚ%test_edge_width_single_value_directeds rlrOrQcCsVt ¡}t |¡}d}tj||||d}t|t |¡ƒD]\}}| ¡|ks8t ‚q8dS)N)çà?ç@g(@)r5r ) rr(rr;r:rWrrXrfr?)r5r@r8ZwidthsrArFZexpected_widthrrrÚtest_edge_width_sequences  rocCsxtjdtjd}t |¡}tj||ddgd}dd„|Dƒ}tj||dd gdd d }d d„|Dƒ}tj ||¡stt‚d S) zyTest that edge_vmin and edge_vmax properly set the dynamic range of the color map when num edges == len(edge_colors).rrCrr.rNcSsg|] }| ¡‘qSr©rE©rIÚerrrÚ (sz7test_edge_color_with_edge_vmin_vmax..r_gš™™™™™é?)r#Z edge_vminZ edge_vmaxcSsg|] }| ¡‘qSrrprqrrrrs-sN) rr9rDr;r:r<r%r=r?)r@r8rAZ orig_colorsZ scaled_colorsrrrÚ#test_edge_color_with_edge_vmin_vmax!s ÿrtcCs`tjdtjd}dd„tt|ƒƒDƒ}t ||¡}t|ƒdksBt‚|D]}| ¡dksFt‚qFdS)z>Test default linestyle for edges drawn with FancyArrowPatches.rrCcSsi|]}|||f“qSrrrHrrrrK4sz9test_directed_edges_linestyle_default..rZsolidN©rr9rDr!rRr:r?Ú get_linestyle)r@r8rArFrrrÚ%test_directed_edges_linestyle_default1s  rwÚstyle)Údashedú--©r1©r1r1cCsdtjdtjd}dd„tt|ƒƒDƒ}tj|||d}t|ƒdksFt‚|D]}| ¡|ksJt‚qJdS)z¬Tests 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).rrCcSsi|]}|||f“qSrrrHrrrrKKsz>test_directed_edges_linestyle_single_value..©rxrNru)rxr@r8rArFrrrÚ*test_directed_edges_linestyle_single_value=s r~Ú style_seqryrzr{ú-ú:z-.cCsttjdtjd}dd„tt|ƒƒDƒ}tj|||d}t|ƒdksFt‚t|t  |¡ƒD]\}}|  ¡|ksVt‚qVdS)zˆTests support for specifying linestyles with sequences in ``draw_networkx_edges`` for FancyArrowPatch outputs (e.g. directed edges).rrCcSsi|]}|||f“qSrrrHrrrrKcsz:test_directed_edges_linestyle_sequence..r}rN) rr9rDr!rRr:r?rWrrXrv)rr@r8rArFrxrrrÚ&test_directed_edges_linestyle_sequenceSs r‚c CsFt ¡}t |¡}tj||ddddgddddtj||d d d d gd dddddgdtj||dddtj||ddddgddddtj||ddddgddd dtj||ddddgddddd di}d|d<d|d<d |d<d!|d<d"|d <d#|d <d$|d <d%|d <tj|||d&d'tj||dd(d)tj||dd*id+dS),Nrr1rPrr+iôgè?)Únodelistr r r`rérérVgÐ?rmr.)r r`r6rOrQ)rré)r5r r`r#)rr„)r„r)rr…)r…rT)r5ÚarrowsÚmin_source_marginÚmin_target_marginr r#z$a$z$b$z$c$z$d$z$\alpha$z$\beta$z$\gamma$z$\delta$é)Z font_sizeF)Ú edge_labelsÚrotatez4-5©r‹)rZ cubical_graphZ spring_layoutÚdraw_networkx_nodesr:Zdraw_networkx_labelsÚdraw_networkx_edge_labels)r@r8ÚlabelsrrrÚtest_labels_and_colorsksr  ÿ  ú  ú ú ø r‘cCsˆt ¡}t ¡\}}ddddddœ}tj||ddd d d gd d tj||ddgdd tj||ddd| d¡t ¡t d¡|S)N©rr©r1rr6r|)rmrn)rr1rPrri¸ rr1rPrztab:blue)r rƒr iÐrz tab:orangermr)r`r g)\Âõ(¼?Úoff) rZ house_graphrÚsubplotsrŽr:ZmarginsZ tight_layoutZaxis)r@ÚfigÚaxr8rrrÚtest_house_with_colors¦s,  ûÿ  r˜cCs4t ¡\}}tjt|dtjtt t¡|ddS)N©r—)rr•rÚdrawrrÚcircular_layout)r–r—rrrÚ test_axesÀs rœcCst ¡}t |¡dSr')rÚGraphrš©r@rrrÚtest_empty_graphÆsrŸcCsÖddl}t ddg¡}t ddg¡}t |¡}ttj||gdtjj ƒsLt ‚ttj||gdtjj ƒsjt ‚tj ||gddgks„t ‚tj ||gddgksžt ‚tj ||gddgks¸t ‚tj ||gddgksÒt ‚dS)NrrOrQ)rƒT)r5r‡F) Úmatplotlib.collectionsrrrDr›rerŽr<Ú collectionsZPathCollectionr?r:)rr@ÚDGr8rrrÚ#test_draw_empty_nodes_return_valuesËs  ÿÿr£cCs<tjdtjd}tj|dgdtj|dgdddgddS) NrrC)rr1r©r5rér)r5r )rr9Z MultiDiGraphÚ draw_networkxržrrrÚtest_multigraph_edgelist_tuplesásr§csœt t¡}t ¡}| d¡tjt|ddgdttjƒ‰‡fdd„t ˆƒDƒ}t ˆƒ}| d¡tjt|||d|  d ¡| d ¡tjt||ddS) Néƒçš™™™™™¹?r_)r`csg|] }|ˆ‘qSrr)rIÚx©Z num_nodesrrrsðsz#test_alpha_iter..é„)r r`r1é…) rr;rrÚfigureZ add_subplotrŽrRÚnodesr!Úappend)r8r–r`r%rr«rÚtest_alpha_iterès      r±c Cs,tjtddtjtddW5QRXdS)NzReceived invalid argument©ÚmatchÚbar)Zfoo)ÚpytestÚraisesÚ ValueErrorrršrrrrrÚtest_error_invalid_kwdsúsr¸c CsHt ddddg¡}dddg}tjtdd tj||d W5QRXdS) Nr6©rrP©rr©r1rr1rPrz1arrowsize should have the same length as edgelistr²)Ú arrowsize)rrDrµr¶r·rš)r@r¼rrrÚ+test_draw_networkx_arrowsize_incorrect_sizeÿs ÿr½r¼érr¥cCs~t dddg¡}ddddœ}tj|||d}t|tƒr@t |¡n|}t||ƒD]*\}}t|tj j ƒsht ‚|  ¡|ksNt ‚qNdS)Nr6r¹rOr’r“)rr1rP)r8r¼) rrDr:reÚintrÚrepeatrWr<ÚpatchesÚFancyArrowPatchr?Zget_mutation_scale)r¼r@r8ÚedgesrFr*rrrÚtest_draw_edges_arrowsizes rÄcCstjtt ddg¡ddS)Nr¹rºr¤)rr¦rÚnpÚarrayrrrrÚtest_np_edgelistsrÇc Cs@t d¡}dddœ}tjtjddt ||¡W5QRXdS)Nrr’r|r6zhas no positionr²)rr9rµr¶Ú NetworkXErrorrŽ©r@r8rrrÚ*test_draw_nodes_missing_node_from_positions  rÊÚ node_shape)ÚoÚsc Cs²t ¡\}}t dg¡}dddœ}tj||||dd}| ¡ ¡ddd…df}tj||||ddd d}| ¡ ¡ddd…df}|d|dksšt‚|d |d ks®t‚dS) aõ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 r6r’r“)r—rËrNrPr)r—rËrˆr‰r1)rr•rrDr:Ú get_extentsZcornersr?) rËr–r—r@r8Z default_patchZdefault_extentZ padded_patchZ padded_extentrrrÚ)test_draw_edges_min_source_target_margins#s$   úùrÏcCslt ¡\}}t ¡}| d¡| dd¡t |ddi¡d}| ¡}|jdkrZ|j dks^t ‚t  |¡dS)zDEnsure that selfloop extent is non-zero when there is only one node.rr’N) rr•rrDÚadd_nodeÚadd_edger:rÎr Úheightr?Údelaxes)r–r—r@ÚpatchÚbboxrrrÚ&test_nonzero_selfloop_with_single_nodePs   rÖcCs|t ¡\}}tjdtjd}| dd¡dd„|jDƒ}tj||dgdd}| ¡}|j dkrj|j dksnt ‚t  |¡d S) zfEnsure that selfloop extent is non-zero when only a single edge is specified in the edgelist. rPrCr1cSsi|]}|||f“qSrrrHrrrrKjszFtest_nonzero_selfloop_with_single_edge_in_edgelist..r|r¤rN) rr•rr9rDrÑr¯r:rÎr rÒr?rÓ)r–r—r@r8rÔrÕrrrÚ2test_nonzero_selfloop_with_single_edge_in_edgelistas  r×cCsJdddg}dddg}d}tjj |||¡}t|dd…d f|kƒsFt‚dS) zfTest apply_alpha when there is a mismatch between the number of supplied colors and elements. rr1rPr+rUrVrmNr\)rZdrawingZnx_pylabZ apply_alphaÚallr?)rƒZ colorlistr`Z rgba_colorsrrrÚtest_apply_alphats   rÙcCsddl}ddl}t d¡}tjdtjd}dd„|Dƒ}||fD]@}tj||dd}t|ƒt|jƒksjt‚t |dt j j ƒs@t‚q@||fD]&}tj||d d}t |t j jƒsŠt‚qŠt ||¡}t |t j jƒsÐt‚t ||¡}t|ƒt|jƒksòt‚t |dt j j ƒs t‚dS) 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 rNrrCcSsi|]}|||f“qSrrrHrrrrKŒsz>test_draw_edges_toggling_with_arrows_kwarg..T)r‡F)r Zmatplotlib.patchesrr9rDr:rRrÃr?rer<rÁrÂr¡ÚLineCollection)rZUGr¢r8r@rÃrrrÚ*test_draw_edges_toggling_with_arrows_kwargs"     rÛÚ drawing_funccCsZddl}t d¡}t ¡\}}|||dtdd„|jDƒƒsBt‚|jrLt‚t  |¡dS)Nrrr™css|]}t|tjjƒVqdSr'©rer<r¡rÚ©rIr[rrrÚ ¨sz?test_draw_networkx_arrows_default_undirected..) r rr9rr•Úanyr¡r?rÁrÓ©rÜrr@r–r—rrrÚ,test_draw_networkx_arrows_default_undirected¡s    râcCs`ddl}tjdtjd}t ¡\}}|||dtdd„|jDƒƒrHt‚|j sRt‚t  |¡dS)NrrrCr™css|]}t|tjjƒVqdSr'rÝrÞrrrrß´sz=test_draw_networkx_arrows_default_directed..) r rr9rDrr•ràr¡r?rÁrÓrárrrÚ*test_draw_networkx_arrows_default_directed­s  ÿ  rãcCsNt d¡}| dd¡t ¡\}}tj|ddg|d|jr@t‚t |¡dS)Nrrr6rO)r5r—) rr9rÑrr•ršrÁr?rÓ)r@r–r—rrrÚtest_edgelist_kwarg_not_ignored»s     räc Csvd}t ¡}|jdddd|jddddt |d¡}dd „|Dƒ}tjtj|d tj|||d W5QRXd S) zs draw_networkx_edge_labels should raise an informative error message when the edge label includes keys z5draw_networkx_edge_labels does not support multiedgesrr1r)Úweightr¥råcSsi|]}|||f“qSrrrHrrrrKÏszEtest_draw_networkx_edge_label_multiedge_exception..r²rN)rZ MultiGraphrÑZget_edge_attributesrµr¶rÈr)Z exception_msgr@r‹r8rrrÚ1test_draw_networkx_edge_label_multiedge_exceptionÅs ræcCs6t d¡}dd„|jDƒ}tj||idiks2t‚dS)zORegression test for draw_networkx_edge_labels with empty dict. See gh-5372.rcSsi|]}|||f“qSrrrHrrrrKØsz.rN)rr9r¯rr?rÉrrrÚ(test_draw_networkx_edge_label_empty_dictÔs rçc Csät ¡\}}ddddddg}ddd d d d g}t |¡}d d„|jDƒ}tj|||||dt|jƒdkslt‚t   |dd…¡t   ddg¡}t |j|dd…|ƒD]2\}}} |  ¡  | ¡s¾t‚tj | ¡|¡s¢t‚q¢t |¡dS)ztWhen an edgelist is supplied along with a sequence of colors, check that the self-loops have the correct colors.r»rOrQr|ri)rPrPZpinkZcyanrr,r4ZgreencSsi|]}|||f“qSrrrHrrrrKåszGtest_draw_networkx_edges_undirected_selfloop_colors..)r—r5r#réýÿÿÿNrr©)rr•rrr¯r:rRrÁr?rÅrÆrWÚget_pathZcontains_pointr<r%r=rErÓ) r–r—r5Z edge_colorsr@r8Z sl_pointsrFZclrZslprrrÚ3test_draw_networkx_edges_undirected_selfloop_colorsÜs    rêÚfap_only_kwargZ arrowstyleZconnectionstylez arc3,rad=0.2rˆr‰c Cs´t d¡}dd„|Dƒ}t ¡\}}t| ¡ƒd}tjtd|›ddtj ||fd|i|—ŽW5QRXt   ¡*t   d ¡tj ||f|d d œ|—ŽW5QRXt  |¡d S) zÏUsers 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.rcSsi|]}|||f“qSrrrHrrrrKszEtest_user_warnings_for_unused_edge_drawing_kwargs..rz The z# keyword argument is not applicabler²r—ÚerrorT)r—r‡N)rr9rr•r ÚkeysrµZwarnsÚ UserWarningr:ÚwarningsÚcatch_warningsÚ simplefilterrÓ)rër@r8r–r—Z kwarg_namerrrÚ1test_user_warnings_for_unused_edge_drawing_kwargsøs   ÿ"  $rò)AÚ__doc__rr rïrµZ importorskipr<rÅZuserZrcParamsZnetworkxrZ barbell_graphrrr"r&r)ÚmarkZ parametrizerBrGrTrZr]rbrcrrDrgrkrlrortrwr~r‚r‘Zmpl_image_comparer˜rœrŸr£r§r±r¸r½rÄrÇrÊrÏrÖr×rÙrÛršr¦rârãrärærçrêròrrrrÚs       ÷þ ÷þ 9   þþ þþ ýþ  þ  úþ ;     , "   ûþ