Gdk dZddlZddlZddlZddlZejdZejdZedejdZ de j d<ddl Z e j d d Zd Zd Zd ZdZejddddgdfddgdfddgdfddgdff ejdddgfdZejddddgdfddgdfddgdfddgdff ejdddgfd Zd!Zd"Zd#Zd$Zd%Zejd&e je jfd'Zejd(d)d*gd*ffd+Zejd(d)d*gd*ffd,Z ejdgd-ddd.gfd/Z!d0Z"d1Z#ejd2d3d4Z$ejd5d6gd7gd8ggd9d7d:ggd;fd<Z%d=Z&ejj'd>Z(d?Z)d@Z*dAZ+dBZ,dCZ-dDZ.dEZ/ejdFdGgdHfdIZ0dJZ1dKZ2ejdLdMdNZ3dOZ4dPZ5dQZ6dRZ7ejdSe j8e j9fdTZ:ejdSe j8e j9fdUZ;dVZdYZ?ejdZd[d:idFd\id]d^id_d`idad`ifdbZ@dS)cz,Unit tests for matplotlib drawing functions.N matplotlibnumpyPSzmatplotlib.pyplotFz text.usetexc tjtjtjtjtjtjtjg}ddddg}tj ||D]-\}}|tfi|t d. tjddS#t$rYdSwxYw# tjdw#t$rYwwxYwxYw)Nblackd) node_color node_sizewidthtest.ps)nx draw_circulardraw_kamada_kawai draw_planar draw_random draw_spectral draw_spring draw_shell itertoolsproductbarbellpltsavefigosunlinkOSError) functionsoptionsfunctionoptions Alib/python3.11/site-packages/networkx/drawing/tests/test_pylab.py test_drawr%s    N N   N M #*aHHI ) 1)W E E # # Hf HW ' ' ' ' ' KK " " " " #   Ii     DD   Ii     D sBBCB33 CCC+CC+ C(%C+'C((C+c ttdttddttddg}tjt|t d tjddS#t$rYdSwxYw# tjdw#t$rYwwxYwxYw)Nr nlistr) listrangerrrrrrrrr)s r$test_draw_shell_nlistr-+seAhheArll!3!3T%B--5H5HI gU++++ I  Ii     DD   Ii     D sBB B4 B## B10B14C6C  C CCCCctt}tjt|dt jjddS)NrT) edge_colorr edge_cmap with_labels)r,rnumber_of_edgesrrrcmBlues)colorss r$test_edge_colormapr67sN 7**,, - -FNF!sv|QUc\tjtdSN)rrr to_directedr7r$ test_arrowsr<?s$N7&&(()))))r7)r/expected)Nr )rredr>r?)?rAgyellowr@rB)rrrDlimerCrE)#0000ffbluerFrGedgelistrrDctjd}tj|tj|||}tj||sJdS)z`Tests ways of specifying all edges have a single color for edges drawn with a LineCollectionr posrHr/N)r path_graphdraw_networkx_edges random_layoutmplr5 same_color get_color)r/r=rHG drawn_edgess r$!test_single_edge_color_undirectedrUDsp& aA( r""X*K : !6!6!8!8( C CCCCCCr7c tjdtj}tj|tj|||}|D]6}t j||sJ7dS)zaTests ways of specifying all edges have a single color for edges drawn with FancyArrowPatchesr  create_usingrKN) rrMDiGraphrNrOrPr5rQ get_edgecolor)r/r=rHrSrTfaps r$test_single_edge_color_directedr\^s& abj111A( r""X*KDDz$$S%6%6%8%8(CCCCCCDDr7cftjdtj}dtt |D}dD]}tj|||}|D]6}t j| |sJ7tj||ddg|}|D]6}t j| |sJ7tj||gd d }|D]6}t j| d sJ7tj||gd d }|D]6}t j| d sJ7tj||gd d }t j|d |d sJ|D]6}t j| d rJ7tj||gd d }t j|d |d sJt j|d |d sJ|D]6}t j| d rJ7dS)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.rrWci|]}|||fSr;r;.0ns r$ z8test_edge_color_tuple_interpretation..} , , ,1q!f , , ,r7)rrrDrrrDrDr/rIrD)rHr/rIrgrhr rerG)rIrgrj)r rrdrrDrhr N) rrMrYr,lenrNrPr5rQrZ)rSrLecrTr[s r$$test_edge_color_tuple_interpretationrmys abj111A , ,eCFFmm , , ,C( B B,QCCC  B BC:(():):)<)gbrfN) rr:rrOrNziprcyclerPr5rQrZrSrL edgecolorsrTr[r=s r$.test_fewer_edge_colors_than_num_edges_directedrvs A 7 # #C J(CJGGGK[)/**E*EFFDD Xz$$S%6%6%8%8(CCCCCCDDr7cNtjdtj}tjt}d}tj|||}t ||ddD]9\}}tj | |sJ:dS)z\Test that extra edge colors are ignored when there are more specified colors than edges.rrW)r>rprqcrfN) rrMrYrOrrNrrrPr5rQrZrts r$-test_more_edge_colors_than_num_edges_directedrzs abj111A 7 # #C%J(CJGGGK[*SbS/::DD Xz$$S%6%6%8%8(CCCCCCDDr7c~tjttjtddgdd}|}t |dksJtj |dddsJ|ddksJdSNrIrgpurple皙?)rLrHr/alpharhry) rrNrrOrRsqueezerk get_pathsrPr5rQ)edge_collectionrls r$3test_edge_color_string_with_global_alpha_undirectedrs,  W % %&! O  " " $ $ , , . .B ((** + +q 0 0 0 0 : CRC( 3 3333 b6S======r7cdtjttjtddgdd}t |dksJ|D]N}|}tj |dddsJ|ddksJOdSr|) rrNrr:rOrkrZrPr5rQ)rTr[rls r$1test_edge_color_string_with_global_alpha_directedrs(  W % %&! K {  q     z$$RWh77777"v}}}}}r7 graph_typectjd|}dtt|D}tj||}t |t r|d}|dksJdS)z^Test the default linewidth for edges drawn either via LineCollection or FancyArrowPatches.rhrWci|]}|||fSr;r;r_s r$rbz1test_edge_width_default_value..rcr7rrDN)rrMr,rkrN isinstancer+ get_linewidth)rrSrLrTs r$test_edge_width_default_valuers aj111A , ,eCFFmm , , ,C(C00K+t$$%!!n  $ $ & &! + + + + + +r7) edgewidthr=r r r c*tjd}dtt|D}tj|||}t|dksJ||ksJdS)Nrci|]}|||fSr;r;r_s r$rbz;test_edge_width_single_value_undirected..rcr7rr )rrMr,rkrNrr)rr=rSrLrTs r$'test_edge_width_single_value_undirectedrs aA , ,eCFFmm , , ,C(CyAAAK {$$&& ' '1 , , , ,  $ $ & &( 2 2 2 2 2 2r7c(tjdtj}dtt |D}tj|||}t |dksJ|D]}||ksJdS)NrrWci|]}|||fSr;r;r_s r$rbz9test_edge_width_single_value_directed.. rcr7rr )rrMrYr,rkrNr)rr=rSrLrTr[s r$%test_edge_width_single_value_directedrs abj111A , ,eCFFmm , , ,C(CyAAAK {  q //  ""h.....//r7rirgct}tj|}d}tj||||}t |t j|D]\}}||ksJ dS)N)?@g(@)rHr) rr:rrOrNrrrrsr)rHrSrLwidthsrTr[expected_widths r$test_edge_width_sequencers A 1  C F(C(&QQQK"; 0G0GHH55^  ""n4444455r7cJtjdtj}tj|}tj||ddg}d|D}tj||ddgdd }d |D}t j||sJd S) zyTest that edge_vmin and edge_vmax properly set the dynamic range of the color map when num edges == len(edge_colors).r rWrrArfc6g|]}|Sr;rZr`es r$ z7test_edge_color_with_edge_vmin_vmax..(s":::1??$$:::r7r~g?)r/ edge_vmin edge_vmaxc6g|]}|Sr;rrs r$rz7test_edge_color_with_edge_vmin_vmax..-s"<<<1Q__&&<<Test default linestyle for edges drawn with FancyArrowPatches.rrWci|]}|||fSr;r;r_s r$rbz9test_directed_edges_linestyle_default..4rcr7r solidNrrMrYr,rkrN get_linestyle)rSrLrTr[s r$%test_directed_edges_linestyle_defaultr1s abj111A , ,eCFFmm , , ,C(C00K {  q ..  ""g-----..r7style)dashed--rDrDrDc(tjdtj}dtt |D}tj|||}t |dksJ|D]}||ksJdS)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).rrWci|]}|||fSr;r;r_s r$rbz>test_directed_edges_linestyle_single_value..Krcr7rr Nr)rrSrLrTr[s r$*test_directed_edges_linestyle_single_valuer=s abj111A , ,eCFFmm , , ,C(Cu===K {  q ,,  ""e+++++,,r7 style_seqrrr)r-:r)rrrz-.cntjdtj}dtt |D}tj|||}t |dksJt |tj|D]\}}| |ksJ dS)zTests support for specifying linestyles with sequences in ``draw_networkx_edges`` for FancyArrowPatch outputs (e.g. directed edges).rrWci|]}|||fSr;r;r_s r$rbz:test_directed_edges_linestyle_sequence..crcr7rr N) rrMrYr,rkrNrrrrsr)rrSrLrTr[rs r$&test_directed_edges_linestyle_sequencerSs abj111A , ,eCFFmm , , ,C(CyAAAK {  q +yy'A'ABB,, U  ""e+++++,,r7c 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!<tj|||d"#tj||dd$%tj||d&d'i(dS))NrrDrhr r>i?)nodelistr r r)rrrq)g?rrrArAr)rr)rIrgrj)r r)rHrrr/)rr)rr)rr)rrT)rHarrowsmin_source_marginmin_target_marginrr/z$a$rz$b$rDz$c$rhz$d$r z$\alpha$rz$\beta$rz$\gamma$rz$\delta$r) font_sizeF) edge_labelsrotaterz4-5r)r cubical_graph spring_layoutdraw_networkx_nodesrNdraw_networkx_labelsdraw_networkx_edge_labels)rSrLlabelss r$test_labels_and_colorsrks A 1  C 3#D  $$$ 1cC8888  111   111   111    FF1IF1IF1IF1IF1IF1IF1IF1IAsFb9999 CT%HHHH Cfe_EEEEEEr7ctj}t\}}dddddd}tj||dgdd tj||d d gd tj||dd|dttd|S)NrrrDrrIr)rr)rrDrhr ri rztab:blue)r rr irz tab:orangerr)rrg)\(?off) r house_graphrsubplotsrrNmargins tight_layoutaxis)rSfigaxrLs r$test_house_with_colorsrs AllnnGCFv* E EC    3$!1cA6666JJtHHUOOO Jr7ct\}}tjt|tjttjt|dS)Nr)rrrdrawrrcircular_layout)rrs r$ test_axesrsQllnnGCGG "*.s 5 5 5qQ] 5 5 5r7)r rrD) rrOrrfigure add_subplotrrknodesr,append)rLrrr5rs @r$test_alpha_iterrs 7 # #C **,,COOC7CSz::::GM""I 5 5 5 5E)$4$4 5 5 5E 9  FOOC7CF%HHHH LLOOOOOC7Cu555555r7ctjtd5tjt dddddS#1swxYwYdS)NzReceived invalid argumentmatchbar)foo)pytestraises ValueErrorrrrr;r7r$test_error_invalid_kwdsr s z)D E E E$$ U####$$$$$$$$$$$$$$$$$$sAA  A ctjgd}gd}tjtd5tj||ddddS#1swxYwYdS)N)rIrrhrr rDr )rDrhr z1arrowsize should have the same length as edgelistr) arrowsize)rrYrrrr)rSrs r$+test_draw_networkx_arrowsize_incorrect_sizers 33344A I M   (( Y''''((((((((((((((((((sAA!Ar)r'rrcdtjgd}dddd}tj|||}t|trt j|n|}t||D]@\}}t|tj j sJ| |ksJAdS)N)rIr rgrrIrrrDrh)rLr) rrYrNrintrrepeatrrrPpatchesFancyArrowPatchget_mutation_scale)rrSrLedgesr[r=s r$test_draw_edges_arrowsizers +++,,AF + +C  "1# C C CE/9)S/I/IX  +++yIUI..44 X#s{:;;;;;%%''83333344r7cptjttddgdS)Nr r r)rrrnparrayr;r7r$test_np_edgelistrs/Wrxx0@'A'ABBBBBBr7ctjd}ddd}tjtjd5tj||ddddS#1swxYwYdS)Nr rrrIzhas no positionr)rrMrr NetworkXErrorrrSrLs r$*test_draw_nodes_missing_node_from_positionr!s aA C r'/@ A A A'' q#&&&''''''''''''''''''sAA!$A! node_shape)osct\}}tjdg}ddd}tj||||d}|ddddf}tj||||dd d}|ddddf}|d|dksJ|d |d ksJdS) aTest 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 rIrr)rr"rNrhr )rr"rrrD)rrrrYrN get_extentscorners) r"rrrSrL default_patchdefault_extent padded_patch padded_extents r$)test_draw_edges_min_source_target_marginsr,#s!&llnnGC F8A C*1cbZPPPQRSM"..0088::33Q36BN)    L!,,..66881a@M  nQ/ / / / /  nQ/ / / / / / /r7ct\}}tj}|d|ddtj|ddid}|}|jdkr |j dksJt |dS)zDEnsure that selfloop extent is non-zero when there is only one node.rrN) rrrrYadd_nodeadd_edgerNr&rheightdelaxes)rrrSpatchbboxs r$&test_nonzero_selfloop_with_single_noder4PsllnnGC AJJqMMMJJq!  "1q&k 2 21 5E     D :>>dkAooooKKOOOOOr7ct\}}tjdtj}|ddd|jD}tj||dgd}|}|j dkr |j dksJt |dS) zfEnsure that selfloop extent is non-zero when only a single edge is specified in the edgelist. rhrWrDci|]}|||fSr;r;r_s r$rbzFtest_nonzero_selfloop_with_single_edge_in_edgelist..j & & &1q!f & & &r7rrrN) rrrrMrYr/rrNr&rr0r1)rrrSrLr2r3s r$2test_nonzero_selfloop_with_single_edge_in_edgelistr8as llnnGC abj111AJJq! & &ag & & &C  "1cVH = = =a @E     D :>>dkAooooKKOOOOOr7cgd}gd}d}tjj|||}t |dddf|ksJdS)zfTest apply_alpha when there is a mismatch between the number of supplied colors and elements. rrorNry)rdrawingnx_pylab apply_alphaall)r colorlistr rgba_colorss r$test_apply_alphar@tscyyHI E*%11)UHMMK {111b5!U* + ++++++r7cddl}ddl}tjd}tjdtj}d|D}||fD]g}tj||d}t |t |jksJt|dtj j sJh||fD]:}tj||d}t|tj j sJ;tj||}t|tj j sJtj||}t |t |jksJt|dtj j sJdS) 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 rWci|]}|||fSr;r;r_s r$rbz>test_draw_edges_toggling_with_arrows_kwarg..s ! ! !1q!f ! ! !r7T)rF)rmatplotlib.patchesrrMrYrNrkrrrPrrrLineCollection)rUGrrLrSrs r$*test_draw_edges_toggling_with_arrows_kwargrFs"!!! q  B qrz 2 2 2B ! !b ! ! !C"XAA&q#d;;;5zzS\\))))%(CK$?@@@@@@"XAA&q#e<<<%!?@@@@@@  "2s + +E eS_; < <<<<  "2s + +E u::QW % % % % eAh ; < <<<<<z?test_draw_networkx_arrows_default_undirected..s1UUz!S_;<<UUUUUUr7) rrrMrranyrrr1rGrrSrrs r$,test_draw_networkx_arrows_default_undirectedrOs!!!! aAllnnGCLr UUbnUUU U UUUUzKKOOOOOr7c&ddl}tjdtj}t\}}|||t d|jDrJ|jsJt |dS)Nrr rWrc3TK|]#}t|tjjV$dSr9rJrKs r$rLz=test_draw_networkx_arrows_default_directed..sB:; 1co455r7) rrrMrYrrrMrrr1rNs r$*test_draw_networkx_arrows_default_directedrRs!!!! abj111AllnnGCLr?A~ :KKOOOOOr7c tjd}|ddt\}}tj|ddg||jrJt|dS)Nr rrIrg)rHr)rrMr/rrrrr1)rSrrs r$test_edgelist_kwarg_not_ignoredrTss aAJJq!llnnGCGA(R0000zKKOOOOOr7crd}tj}|ddd|dddtj|d}d|D}t jtj| 5tj||| d d d d S#1swxYwYd 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 multiedgesrrDr')weightrrVci|]}|||fSr;r;r_s r$rbzEtest_draw_networkx_edge_label_multiedge_exception.. 1q!f r7rrN)r MultiGraphr/get_edge_attributesrrrr) exception_msgrSrrLs r$1test_draw_networkx_edge_label_multiedge_exceptionr\s LM AJJq!BJJJq!BJ(H55K a C r'} = = =FF $QEEEEFFFFFFFFFFFFFFFFFFsB,,B03B0ctjd}d|jD}tj||iiksJdS)zORegression test for draw_networkx_edge_labels with empty dict. See gh-5372.r ci|]}|||fSr;r;r_s r$rbz.r7r7rN)rrMrrr s r$(test_draw_networkx_edge_label_empty_dictr_sQ aA & &ag & & &C '3B ? ? ?2 E E E E E Er7ct\}}gd}gd}tj|}d|jD}tj|||||t |jdksJt |ddt dd gz}t|j|dd|D]c\}}} | | sJtj||sJdt|dS) ztWhen an edgelist is supplied along with a sequence of colors, check that the self-loops have the correct colors.)r rgrjrr)rhrh)pinkcyanr r?rGgreenci|]}|||fSr;r;r_s r$rbzGtest_draw_networkx_edges_undirected_selfloop_colors..r7r7)rrHr/r Nrr)rrrrrrNrkrrrrrget_pathcontains_pointrPr5rQrZr1) rrrH edge_colorsrSrL sl_pointsr[clrslps r$3test_draw_networkx_edges_undirected_selfloop_colorsrlsFllnnGC???HCCCK A & &ag & & &C1cb8 TTTT rz??a     "##''"((As8*<*<>>>>>KKOOOOOr7fap_only_kwarg arrowstylerconnectionstylez arc3,rad=0.2rr'rcBtjd}d|D}t\}}t |d}t jtd|d5tj ||fd|i|dddn #1swxYwYtj 5tj d tj ||f|d d |dddn #1swxYwYt |dS) 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 ci|]}|||fSr;r;r_s r$rbzEtest_user_warnings_for_unused_edge_drawing_kwargs..rXr7rz The z# keyword argument is not applicablerrNerrorT)rr)rrMrrr+keysrwarns UserWarningrNwarningscatch_warnings simplefilterr1)rmrSrLrr kwarg_names r$1test_user_warnings_for_unused_edge_drawing_kwargsrzs aA a CllnnGCn))++,,Q/J UjUUU   @@ q#??"????@@@@@@@@@@@@@@@  " "MMg&&& q#L"TLL^LLLMMMMMMMMMMMMMMM KKOOOOOs$B%%B),B)+C::C>C>)A__doc__rrrvr importorskiprPruserrcParamsnetworkxr barbell_graphrr%r-r6r<mark parametrizerUr\rmrvrzrrrrYrrrrrrrrrmpl_image_comparerrrrrrr rrrr!r,r4r8r@rFrrrOrRrTr\r_rlrzr;r7r$rs22  f,''V!! f-..# ] " 1a .   ***  #  H%   f   dVH%566DD76  D #  H%   f   dVH%566 D D76   D6F6F6FrDDDDDD      "*'=>>,,?>, a333 a///    555 = = = . . .  , , ,     s     , ,   ,7F7F7Fv2NNN  KKK,EEE666$$$$ (((r<<<&899 4 4:9 4CCC '''z22)0)032)0X"&,,,===D"'23C)DEEFE"'23C)DEE  FE  F F FFFF8 s b N+ b! b!     r7