btj) jdZddlZddlZddlZddlZddlZejdZejdZ e dejdZ de j d<ej d d Zid dd d dddddddddddddddd ddddddddd d d!dd"dd#dddddd$d%d%d&ddd'd d( d)d*d+dd,d*d-dd.ddd/Zejd0ddedfd1d2fd3Zejd4d5d6Zd7Zd8Zejd9d:ejd;ejejfd<Zejd;ejejfd=Zejd;ejejfd>Zd?Zejd;ejejfd@Zejd;ejejfdAZejd4ded!ed!gfdBdBdBgfdCdBdDgffejd;ejejfdEZejd4ded)ed)gfdFdFdFgfdGd*dFgffejd;ejejfdHZdIZ dJZ!dKZ"dLZ#dMZ$dNZ%ejddOdPZ&ejdQdd gdRZ'dSZ(dTZ)ej*dUZ+ejdVej,ej-ej.ej/ej0ej1ej2ej3gdWZ4dXZ5dYZ6dZZ7d[Z8d\Z9ejd9d]d^d_gd`fdadbgdcfdddegdffdgdhgdiff ejdjddkgfdlZ:ejd9d]d^d_gd`fdadbgdcfdddegdffdgdhgdiff ejdjddkgfdmZ;dnZdqZ?drZ@ejd;ejejfdsZAejdtdudvgdvffdwZBejdtdudvgdvffdxZCejdjgdyddkdzgfd{ZDd|ZEd}ZFejdGd~dZGejddgdgdggddd*ggdfdZHdZIdZJdZKdZLdZMdZNdZOdZPdZQdZRdZSdZTejddgdfdZUejddgdfdZVdZWdZXdZYejddOdZZejddOdZ[dZ\dZ]dZ^dZ_ejdej`ejafdZbejdej`ejafdZcdZdejdejdgejed gejfd gfdZgejdejdgejed gejfd gfdZhdZidZjdZkejddd*iddiddiddDiddDifdZlejdej`ej,fdZmejddd gejdejaejnejoejpejqgdZrejdGgddZsejdddgdZtdS)z,Unit tests for matplotlib drawing functions.N matplotlibnumpyPSzmatplotlib.pyplotFz text.usetexnode_pos node_visibleT node_colorz#1f78b4 node_sizei, node_label z#000000z sans-serifnormal?center) sizecolorfamilyweightalphabackground_colorbackground_alphah_alignv_alignbbox node_shapeo node_alphanode_border_widthnode_border_colorface edge_visible edge_width edge_color edge_labelround)rrr)boxstyleecfc?) rrrrrrrrposrotate edge_style- edge_alphaedge_arrowstyleedge_arrowsizearc3)edge_curvatureedge_source_marginedge_target_margin) param_name param_valueexpected)r #FF0000red)r rlimectjd}tj|ddtd}tj|fd|i||itj | d d|sJt dS)Nrz#00FF00rocanvasr) nx path_graphset_node_attributespltfigure add_subplotdisplaymplcolors same_color get_childrenget_edgecolorsclose)r5r6r7Gr=s _/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/networkx/drawing/tests/test_pylab.py$test_display_arg_handling_node_colorrMCs aA1i111 ZZ\\ % %c * *FJq===J #<=== : !4!4!6!6q!9!H!H!J!J1!Mx X XXX XIIKKKKK)r6r7))N)rOrOrO)r))r)r)r)r))n_alpha)rr)gUUUUUU??ctjd}tj|d|Ddtd}tj|||t| d dddf|ksJt dS)Nrc i|] }|d|dzz  S)rO.0ns rL z8test_display_arg_handling_node_alpha..^s"AAA!q!q1u+AAArNrPr<)r=rr) r>r?r@nodesrArBrCrDallrHget_fcrJ)r6r7rKr=s rL$test_display_arg_handling_node_alphar]Ts aA1AAqwwyyAAA9MMM ZZ\\ % %c * *FJqK8888 a ''))!!!Q$/8;   IIKKKKKrNctjd}tj|d|Ddtd}tj||dt | d j ddgddgddgd d ggksJt dS) Nrci|]}|||fSrTrTrUs rLrXz.test_display_node_position..i<<r?r@rZrArBrCrDnpr[rH get_offsetsdatarJ)rKr=s rLtest_display_node_positionregs aA1<r-rTrVuvs rLrXz0test_display_line_collection..us3 H H HdaQFa!eq[1EEc H H HrN arrowstyler< )r=r0cPg|]#}t|tjj!|$SrT isinstancerE collectionsLineCollectionrVls rL z0test_display_line_collection..zs=    a7 8 8     rNrc(g|]\}}||zdz dS)rarOrTris rLruz0test_display_line_collection..s)&N&N&NTQ1q5A+&Nq&N&N&NrN) r>karate_club_graphset_edge_attributesedgesrArBrCrDrHlen get_pathssumrJ)rKr=lcs rLtest_display_line_collectionr~rs A H Haggii H H H,ZZ\\ % %c * *FJq3333  $$&&     B r||~~  #&N&NQWWYY&N&N&N"O"O O O O OIIKKKKKrN)r#r7)Nblackrr9rryellowrrOrrOr:)rblue)#0000FFr graph_typec tjd|}tj|ddtd}tj||||rd|D}n)d|j Dd  }tfd |DsJt dS) NrY create_usingrrr<)r#r=ctg|]5}t|tjj!|6SrTrprEpatchesFancyArrowPatchr\rVfs rLruz2test_display_edge_single_color..E   !S[899 HHJJ   rNcPg|]#}t|tjj!|$SrTrorss rLruz2test_display_edge_single_color..=   !S_;<<    rNrc3XK|]$}tj|V%dSN)rErFrG)rVcr7s rL z1test_display_edge_single_color..s5BBasz$$Q11BBBBBBrN) r>r?rxrArBrCrD is_directedrHrq get_colorsr[rJ)r#r7rrKr=rFs ` rLtest_display_edge_single_colorrs aj111A1i111 ZZ\\ % %c * *FJqZ7777}}   ((**      '     Z\\  BBBB6BBB B BBB BIIKKKKKrNctjd|}tj|ddddtd}tj||d d g}|rd |D}n)d |j Dd  }tj ||sJtdS)NrYrr8rrrOrrOrOrarr<r=r9rctg|]5}t|tjj!|6SrTrrs rLruz5test_display_edge_multiple_colors..rrNcPg|]#}t|tjj!|$SrTrorss rLruz5test_display_edge_multiple_colors..s<   As7U)V)V    rNr)r>r?rxrArBrCrDrrHrqrrErFrGrJ)rrKaxr7rFs rL!test_display_edge_multiple_colorsrs  aj111A1y)DDgNNN  ! !# & &BJqvH}}   __&&      ~    Z\\  :  2 222 2IIKKKKKrNctjd|}tj|d|Ddtd}tj|||rd| D}n5d|j Dd }d | D}d d g}d }t||D][\}}|\} } |\} } tt| | z ||fkr%tt| | z ||fksJ\tdS)NrYrci|]}|||fSrTrTrUs rLrXz.test_display_edge_position..r`rNr*r<rcg|]g}t|tjj!|jdddf|jdddffhS)rN)rprErrget_pathverticesrs rLruz.test_display_edge_position..sr   !S[899 ZZ\\ "1aaa4 (!**,,*?AAA*F G   rNcPg|]#}t|tjj!|$SrTrorss rLruz.test_display_edge_position..s<   As7U)V)V    rNrcRg|]$}|jdddf|jdddff%S)rN)rrVps rLruz.test_display_edge_position..sH   67QZ111 qz"aaa%0 1   rN)rrrOrO)rrarag?)r>r?r@rZrArBrCrDrrHrqr{zipr[absrJ) rrKr end_pointsline_collectionr7 thresholdae act_startact_end exp_startexp_ends rLtest_display_edge_positionrs aj111A1< spring_layoutrKs rL fixed_layoutz4test_display_position_function..fixed_layouts////rNr<)rr=c4i|]\}}|t|SrTtuple)rVrWrs rLrXz2test_display_position_function..s11588rNr) r>rwrArBrCrDrrZrHrcrdr[rJ)rKrr*ract_posrWs rLtest_display_position_functionrs A000 ,q//C  ! !# & &BJq<3333 #AGGIIr/@/@/C/O/O/Q/Q/V W WGWWYY))3q6WQZ'((((((IIKKKKKrNctjd|}tj|ddddtjd}tj|dd|d t d }tj || tj | }| dd| d| dg}|rd|D}n)d|jDd}tj|d|jddsJtj|d|jddsJtj||sJt dS)NrYrrrOrrplasmarF)rZr<rcmapctg|]5}t|tjj!|6SrT)rprErr get_facecolorrs rLruz/test_display_edge_colormaps..G   !S[899 OO     rNcPg|]#}t|tjj!|$SrTrorss rLruz/test_display_edge_colormaps..rrNrr)r>r?rxrE colormapsapply_matplotlib_colorsrArBrCrDcmScalarMappableset_climto_rgbarrHrqrrFrGryrJrrKrr=mapperr7rFs rLtest_display_edge_colormapsrs aj111A1q!44h??? = "Dq(GTGGGG ZZ\\ % %c * *FJq    V " " " - -F OOAqq!!6>>!#4#45H}}   ((**      '     Z\\  : !agdmG.D E EEE E : !agdmG.D E EEE E : 6 2 222 2IIKKKKKrNcBtjd|}tj|dddddtjd}tj|dd |t d }tj || tj | }| dd| d| d| dg}d |jDd}tj|d|jdd sJtj|d|jdd sJtj|d|jdd sJtj||sJt dS)NrYrrr)rOrrOrarrrr<rrcPg|]#}t|tjj!|$SrTrprErqPathCollectionrVss rLruz/test_display_node_colormaps..s<As7U)V)V rNra)r>r?r@rErrrArBrCrDrrrrrqrIrFrGrZrJrs rLtest_display_node_colormapsrs aj111A1!22H=== = "Dq(GT::: ZZ\\ % %c * *FJq    V " " " - -F OOAqq!!6>>##6#6q8I8IJH%     : !agaj.A B BBB B : !agaj.A B BBB B : !agaj.A B BBB B : 6 2 222 2IIKKKKKrNwidthrmctjd|}tj|ddddtd}tj||||rd |D}n6td |j Dd  }||ksJdS) NrYrrrmrrr<)r"r=ctg|]5}t|tjj!|6SrT)rprErr get_linewidthrs rLruz+test_display_edge_width..,rrNcPg|]#}t|tjj!|$SrTrorss rLruz+test_display_edge_width..3s=   a!?@@    rNr) r>r?rxrArBrCrDrrHlistrqget_linewidths)r6r7rrKr=widthss rLtest_display_edge_widthrs aj111A1q"55w??? ZZ\\ % %c * *FJq[8888}}    ((**      +      !!    X      rN:stylectjd|}tj|ddddtd}tj||||rd |D}n:ddd fd d |j Dd  D}||ksJt dS)NrYrr-rrrr<)r,r=ctg|]5}t|tjj!|6SrT)rprErr get_linestylers rLruz+test_display_edge_style..KrrN))rN)r)rOgffffff?cjg|]/}|d|dt|dndf0S)rrONr)rVr linestyless rLruz+test_display_edge_style..SsN    !QqT-=eAaDkkk4H I   rNcPg|]#}t|tjj!|$SrTrorss rLruz+test_display_edge_style..Us=a!?@@rNr) r>r?rxrArBrCrDrrHrqget_linestylesrJ)r6r7rrKr=stylesrs @rLtest_display_edge_styler<s2 aj111A1sC88'BBB ZZ\\ % %c * *FJq[8888}}   ((**   "%c::     +  !!     X    IIKKKKKrNctjd}td}tj||ddid|D}t||D]F\}}| t|ksJ| dksJGt dS)Nrr<r)r=r cPg|]#}t|tjj!|$SrTrprEtextTextrVts rLruz,test_display_node_labels..c+ O O OA*Q 2N2N Oa O O OrNg4@) r>r?rArBrCrDrHrrZget_textstrget_sizerJ)rKr=labelsrWrts rLtest_display_node_labelsr _s aA ZZ\\ % %c * *FJqVRL9999 O O,,.. O O OFAGGIIv&&$$1zz||s1vv%%%%zz||t#####IIKKKKKrNctjd}td}tj|d|Dtj||ddidd|D}t|t||D]U\}}| t|d|d zksJ| dksJVtdS) Nrr<c(i|]\}}||fd||ziS)labelrTris rLrXz,test_display_edge_labels..os+NNNDAq1vQ'7NNNrNrr)r=r$r cPg|]#}t|tjj!|$SrTrrs rLruz,test_display_edge_labels..qrrNrrO)r>r?rArBrCrxryrDrHprintrrr get_colorrJ)rKr=r rrts rLtest_display_edge_labelsrjs aA ZZ\\ % %c * *F1NNAGGIINNNOOOJqWcNtLLLL O O,,.. O O OF &MMMAGGIIv&&$$1zz||s1Q4!A$;//////{{}}#####IIKKKKKrNctj}|gd|gdtj|d|dDdt d}tj ||d | D}|gd ksJt dS) N)ABCD))rr0)rr1)rr-1)rrr)rrr)rrrc Fi|]}|ddt|dzS)z arc3,rad=皙?raintrVrs rLrXz.s1 I I IA,3QqT?,, I I IrNT)keys curvaturer<rc~g|]:}t|tjj!|j;SrT)rprErrget_connectionstyleradrs rLruz.sL    a4 5 5  #   rN)rr皙ɿrr$r) r> MultiGraphadd_nodes_fromadd_edges_fromrxryrArBrCrDrHrJ)rKr=radss rL(test_display_multigraph_non_integer_keysr)ys A)))***       I Iagg4g6H6H I I I;ZZ\\ % %c * *FJq      $$&&   D 333 3 3 3 3IIKKKKKrNctj}tjtj5tj|dt ddddS#1swxYwYdS)Nbad_arg)r+)r>rwpytestraises NetworkXErrorrDrArJrs rLtest_display_raises_for_bad_argr/s A r' ( ( 1i(((( s0A//A36A3ctjdtj}tj|d|Ddt }tj||gdd|DksJt dS)Nrrc0i|]\}}||f||zdzdzS)rarTris rLrXz+test_display_arrow_size..s/ ; ; ;AQFQUQY1$ ; ; ;rN arrowsizer) 1ctg|]5}t|tjj!|6SrT)rprErrget_mutation_scalers rLruz+test_display_arrow_size..sI a4 5 5 rN) r>r?DiGraphrxryrAaxesrDrHrJ)rKrs rLtest_display_arrow_sizer:s abj111A ; ; ; ; ;[ BJq ;;""     IIKKKKKrNc^tjd}tj|ddddddtj|d|Dd t jtj5tj|d d d d S#1swxYwYd S) zP This test ensures that a error is raised for incomplete position data. rrrr)rr)rrOrarr*ci|] }||dk S)rrTrUs rLrXz9test_display_mismatched_edge_position..s;;;Aq!a%;;;rNvisibleN)r>r?r@rZr,r-r.rDrs rL%test_display_mismatched_edge_positionr>s aA1&V6JJERRR1;;;;;YGGG r' ( ( 1 sB""B&)B&)rrcHtd}tjdg}tj|ddddtj|||d|Dd}| d d d df}td}tj||d d | d |Dd}| d d d df}|d|dksJ|d|dksJt d S)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 os min_{source/target}_margin edge attributes result is 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 r<rrrr*)r=rcPg|]#}t|tjj!|$SrTrprErrrs rLruz-test_display_edge_margins..s< 1ck6Q(R(R rNrNrad)r=r3r4rcPg|]#}t|tjj!|$SrTrArs rLruz-test_display_edge_margins..s< 1ck6Q(R(R rNrO) rArBrCr>r8r@rDrH get_extentscornersrJ)rrrK default_arrowdefault_extent padded_arrow padded_extents rLtest_display_edge_marginsrJs*  ! !# & &B F8A1&V44e<<<Jq 3333??$$ M#..0088::33Q36BN  ! !# & &BJ  ??$$ L!,,..66881a@M  nQ/ / / / /  nQ/ / / / /IIKKKKKrNticksctjd}tj|d|Ddt}tj|||j|jfD])}t| |ksJ*t dS)NrYci|]}|||fSrTrTrUs rLrXz+test_display_hide_ticks..r`rNr*) hide_ticks) r>r?r@rZrAr9rDxaxisyaxisboolget_ticklabelsrJ)rKrKraxiss rLtest_display_hide_ticksrTs aA1<.s<    1ck6Q(R(R     rN)rAr9r>r8add_nodeadd_edger@rDrHrDrrheightdelaxesrJ)rrKarrowrs rLtest_display_self_loopr\s B AJJqMMMJJq!1q&k5111Jq  ??$$    E     D $* $+ :>>dkAooo -KKOOOIIKKKKKrNctj}tj|tj|diksJdS)z If the pos attribute isn't provided or is a function, display computes the layout and adds it to the graph. We need to ensure that this new attribute is removed from the returned graph. z!display's position attribute nameN)r>rwrDget_node_attributesrs rLtest_display_remove_pos_attrr_ sD AJqMMM !!%H I IR O O O O O OrNc#Kt\}}||fVt|tdSr)rAsubplotsrZrJ)figrs rLrarasDllnnGC r'MMMKKOOOIIKKKKKrNfunctionc|tjkrtjdd|\}}dddd}|tfi|||dz dS) Nscipyz draw_kamada_kawai requires scipy)reasonrrBrY)r r rtest.ps)r>draw_kamada_kawair, importorskipbarbellsavefig)rcratmp_pathrb_optionss rL test_drawro sv2'''G,NOOOO FC$3CCG HW     KK9$%%%%%rNc |\}}ttdttddttddg}tjt|||dz dS)Nrrm)nlistrg)rranger> draw_shellrjrk)rarlrbrmrrs rLtest_draw_shell_nlistru6su FC %((^^T%2,,//eBmm1D1D EEM'''''KK9$%%%%%rNc|\}}tjdd}tj|||dz dS)Nrarrg)r>complete_bipartite_graphdraw_bipartiterk)rarlrbrmrKs rLtest_draw_bipartitery=sK FC #Aq))AaKK9$%%%%%rNctt}tjt|dt jjddS)NrT)r#r edge_cmap with_labels)rsrjnumber_of_edgesr> draw_springrArBlues)rFs rLtest_edge_colormaprDsN 7**,, - -FNF!sv|QUrNc|\}}d}tj|g}d|D}tj||tj|||di||dz dS)Nrci|]}|||fSrTrTrUs rLrXz2test_draw_networkx_edge_labels..P 1q!f rN)r*edge edge_labelsrg)r>r8drawdraw_networkx_edge_labelsrk)rarlrbrmrrKr*s rLtest_draw_networkx_edge_labelsrLs FC D D6A a CGA3 CdF^DDDDKK9$%%%%%rNc\tjtdSr)r>r~rj to_directedrTrNrL test_arrowsrVs$N7&&(()))))rNrrrr9rrrrrr:)#0000ffrrredgelistrctjd}tj|tj|||}tj||sJdS)z`Tests ways of specifying all edges have a single color for edges drawn with a LineCollectionrYr*rr#N)r>r?draw_networkx_edges random_layoutrErFrGr)r#r7rrK drawn_edgess rL!test_single_edge_color_undirectedr[sp& aA( r""X*K : !6!6!8!8( C CCC CCCrNc tjdtj}tj|tj|||}|D]6}t j||sJ7dS)zaTests ways of specifying all edges have a single color for edges drawn with FancyArrowPatchesrYrrN) r>r?r8rrrErFrG get_edgecolor)r#r7rrKrfaps rLtest_single_edge_color_directedrus& abj111A( r""X*KDDz$$S%6%6%8%8(CCCCCCDDrNcftjdtj}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.rrci|]}|||fSrTrTrUs rLrXz8test_edge_color_tuple_interpretation.. , , ,1q!f , , ,rN)rrrrOrOr#rr)rr#rrrarYrr)rrr)rYrrrrOrarYN) r>r?r8rsrzrrErFrGr)rKr*r'rrs rL$test_edge_color_tuple_interpretationrs abj111A , ,eCFFmm , , ,C( B B,QCCC  B BC:(():):)<)rrr itertoolscyclerErFrGrrKr* edgecolorsrrr7s rL.test_fewer_edge_colors_than_num_edges_directedrs A 7 # #C J(CJGGGK[)/**E*EFFDD Xz$$S%6%6%8%8(CCCCCCDDrNcNtjdtj}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.rr)rrrrrNr) r>r?r8rrjrrrErFrGrrs rL-test_more_edge_colors_than_num_edges_directedrs abj111A 7 # #C%J(CJGGGK[*SbS/::DD Xz$$S%6%6%8%8(CCCCCCDDrNc~tjttjtddgdd}|}t |dksJtj |dddsJ|ddksJdSNrrpurpler)r*rr#rrar) r>rrjrrsqueezerzr{rErFrG)edge_collectionr's rL3test_edge_color_string_with_global_alpha_undirectedrs,  W % %&! O  " " $ $ , , . .B ((** + +q 0 0 0 0 : CRC( 3 333 3 b6S======rNcdtjttjtddgdd}t |dksJ|D]N}|}tj |dddsJ|ddksJOdSr) r>rrjrrrzrrErFrG)rrr's rL1test_edge_color_string_with_global_alpha_directedrs(  W % %&! K {  q     z$$RWh77777"v}}}}}rNctjd|}dtt|D}tj||}t |t r|d}|dksJdS)z^Test the default linewidth for edges drawn either via LineCollection or FancyArrowPatches.rarci|]}|||fSrTrTrUs rLrXz1test_edge_width_default_value..rrNrrON)r>r?rsrzrrprr)rrKr*rs rLtest_edge_width_default_valuers aj111A , ,eCFFmm , , ,C(C00K+t$$%!!n  $ $ & &! + + + + + +rN) edgewidthr7rYrYrYc*tjd}dtt|D}tj|||}t|dksJ||ksJdS)Nrci|]}|||fSrTrTrUs rLrXz;test_edge_width_single_value_undirected..rrNrrY)r>r?rsrzrr{r)rr7rKr*rs rL'test_edge_width_single_value_undirectedrs aA , ,eCFFmm , , ,C(CyAAAK {$$&& ' '1 , , , ,  $ $ & &( 2 2 2 2 2 2rNc(tjdtj}dtt |D}tj|||}t |dksJ|D]}||ksJdS)Nrrci|]}|||fSrTrTrUs rLrXz9test_edge_width_single_value_directed.. rrNrrY)r>r?r8rsrzrr)rr7rKr*rrs rL%test_edge_width_single_value_directedrs abj111A , ,eCFFmm , , ,C(CyAAAK {  q //  ""h.....//rNrrct}tj|}d}tj||||}t |t j|D]\}}||ksJ dS)N)r)g@g(@)rr) rjrr>rrrrrr)rrKr*rrrexpected_widths rLtest_edge_width_sequencer's A 1  C F(C(&QQQK"; 0G0GHH55^  ""n4444455rNcJtjdtj}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).rYrrrrc6g|]}|SrTrrs rLruz7test_edge_color_with_edge_vmin_vmax..?s":::1??$$:::rNrg?)r# edge_vmin edge_vmaxc6g|]}|SrTrrs rLruz7test_edge_color_with_edge_vmin_vmax..Ds"<<<1Q__&&<<r?r8rrrErFrG)rKr*r orig_colors scaled_colorss rL#test_edge_color_with_edge_vmin_vmaxr8s abj111A 1  C(CQHEEEK::k:::K( 3C:K=< <<Test default linestyle for edges drawn with FancyArrowPatches.rrci|]}|||fSrTrTrUs rLrXz9test_directed_edges_linestyle_default..KrrNrYsolidNr>r?r8rsrzrr)rKr*rrs rL%test_directed_edges_linestyle_defaultrHs abj111A , ,eCFFmm , , ,C(C00K {  q ..  ""g-----..rN)dashed--rOrc(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).rrci|]}|||fSrTrTrUs rLrXz>test_directed_edges_linestyle_single_value..brrNrrYNr)rrKr*rrs rL*test_directed_edges_linestyle_single_valuerTs abj111A , ,eCFFmm , , ,C(Cu===K {  q ,,  ""e+++++,,rN style_seqrrr)rr-r)rr-rz-.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).rrci|]}|||fSrTrTrUs rLrXz:test_directed_edges_linestyle_sequence..zrrNrrYN) r>r?r8rsrzrrrrr)rrKr*rrrs rL&test_directed_edges_linestyle_sequencerjs abj111A , ,eCFFmm , , ,C(CyAAAK {  q +yy'A'ABB,, U  ""e+++++,,rNcPddlm}m}ddlm}t jt j}t jt j}t j |d}t j |d}t j ||}t||sJt j ||d}t|tsJt|dkrt|d|sJt j ||d }t||sJt j ||d}t||sJt j ||d}t|tsJt|dkrt|d|sJdSdS) Nr)rrr)rr*rTarrowsF)matplotlib.collectionsrrrmatplotlib.patchesrr> frucht_graphGraphr8rdraw_networkx_nodesrprrrz) rrrrrKdGr*dposrZrys rLtest_return_typesrsEEEEEEEE222222 RX...A bj 1 1 1B 12 & & &C BR ( ( (D  "1c * *E e^ , ,,, ,  "2tD 9 9 9E eT " """ " 5zzA~~%(O44444  "2tE : : :E e^ , ,,, ,  "1d4 8 8 8E e^ , ,,, ,  "2s4 8 8 8E eT " """ " 5zzA~~%(O44444~44rNc tj}tj|d}tj||gddddtj||gdd dgd tj||d d tj||gddd dtj||gddd d tj||gddd ddd i}d|d<d|d<d|d<d|d<d|d<d|d<d |d!<d"|d#<d$t dD}tj|||d%&tj|||d%|'tj||dd()tj||d*d+i,dS)-Nrr)rrOrarYri?)nodelistr r r)rrrr)rQr)rrrr))rr)rrr)rYr)rrrr#)rr)rr)rr)rrT)rrmin_source_marginmin_target_marginrr#z$a$rz$b$rOz$c$raz$d$rYz$\alpha$rz$\beta$rz$\gamma$rz$\delta$rc*i|]}||dzdkrdndS)rarkrrTrUs rLrXz*test_labels_and_colors..s* > > >aA  > > >rN) font_size)r font_colorF)rr+rz4-5r)r> cubical_graphrrrrsdraw_networkx_labelsr)rKr*r rFs rLtest_labels_and_colorsrs A 12 & & &C 3#D  $$$ 1cC8888  111   111   111    FF1IF1IF1IF1IF1IF1IF1IF1I > >U1XX > > >FAsFb9999AsFbVLLLL CT%HHHH Cfe_EEEEEErNc|\}}tjt|tjttjt|dS)Nr)r>rrjrcircular_layout)rarbrs rL test_axesrsIGCGG "*rrrs rLtest_empty_graphrs  AGAJJJJJrNcVddl}tjddg}tjddg}tj|}t tj||gtjj sJt tj||gtjj sJtj ||gdgksJtj ||gdgksJtj ||gdgksJtj ||gdgksJdS)Nrrr)rT)rrF) rr>rr8rrprrErqrr)rrKDGr*s rL#test_draw_empty_nodes_return_valuesrsT!!!! &&!""A VV$ % %B Q  C  q#333S_5S     r3444co6T     !!S2d C C Cr I I I I !!S2e D D D J J J J !"cBu E E E K K K K !"cBt D D D J J J J J JrNctjdtj}tj|dgtj|dggddS)NrYr)rrOrr)rmrr)rr )r>r? MultiDiGraph draw_networkxrs rLtest_multigraph_edgelist_tuplesrsT abo666AQ)----Q) DDDDDDrNcHtjt}t}|dtjt|ddgttjfdtD}t}|dtjt|||| d|d tjt||dS) N皙?r)rcg|]}|z SrTrT)rVx num_nodess rLruz#test_alpha_iter.. s 5 5 5qQ] 5 5 5rN)r rrO) r>rrjrArBrCrrzrZrsappend)r*rbrrFr s @rLtest_alpha_iterrs 7 # #C **,,COOC7CSz::::GM""I 5 5 5 5E)$4$4 5 5 5E 9  FOOC7CF%HHHH LLOOOOOC7Cu555555rNc|\}}tjd}tj|gd|d|D}t |dksJdS)Nr)rhr^)rrcPg|]#}t|tjj!|$SrTrrs rLruz-test_multiple_node_shapes..s< 1co6T(U(U rN)r>r?rrHrz)rarbrrKscatterss rLtest_multiple_node_shapesrsGC aAGA...26666??$$H x==A      rNc2tj}|\}}tj||d|Dd|Dt |d|DD]o\}}|dzrdnd}t j| |sJt|dz d z| ksJpdS) Nc"i|] }||dzrdnd S)rarrrTrUs rLrXz7test_individualized_font_attributes..#s'@@@Aa!e,ss@@@rNc:i|]}|t|dz dzS)""""""@rrrUs rLrXz7test_individualized_font_attributes..$s+@@@1c!w-!+,,@@@rN)rrrcg|]H}t|tjjr't |dkF|ISr)rprErrrzrrs rLruz7test_individualized_font_attributes..(sU   !SX]++ 14AJJLL0A0AA0E0E 0E0E0ErNrarr9rr) r>rwrrZrrHrErFrGrrr)rarKrbrrWrr7s rL#test_individualized_font_attributesrs+ AGCG @@aggii@@@@@aggii@@@      __&&    6 61 !e.77z$$Q[[]]H=====1=1$%%55555 6 6rNctj}|\}}d|D}d|D}tj||d||d|D}t ||D]\}}|d|ddz|d dzzzdzksJt|dzrtj j j ntj j j } t|| sJdS) Nc4g|]\}}||zdzdkrdndS)rarrh-[rTris rLruz7test_individualized_edge_attributes..7s2MMM41aQUaK1,,55$MMMrNc6g|]\}}d|dz|dzzzdzS)rmrarTris rLruz7test_individualized_edge_attributes..8s2BBB1"AA &+BBBrNT)rrrlr2cPg|]#}t|tjj!|$SrTrArs rLruz7test_individualized_edge_attributes..:s< 1ck6Q(R(R rNrmrrarO)r>rwryrrHrr7r|rEr ArrowStyleBracketB CurveFilledBrpget_arrowstyle) rarKrbr arrowstyles arrowsizesrrrr7s rL#test_individualized_edge_attributesr(4sW AGCMM17799MMMKBB BBBJGA"TkZPPPP??$$FAGGIIv&&881##%%qtax!A$(/B)Cb)HHHHH1vvz 5CK " + +'4  !**,,h77777788rNctjtd5tjt dddddS#1swxYwYdS)NzReceived invalid argumentmatchbar)foo)r,r- ValueErrorr>rrjrTrNrLtest_error_invalid_kwdsr/Gs z)D E E E$$ U####$$$$$$$$$$$$$$$$$$sAA  A ctjgd}gd}tjtd5tj||ddddS#1swxYwYdS)N)rrrarrYrOrY)rOrarYz1arrowsize should have the same length as edgelistr*)r2)r>r8r,r-r.r)rKr2s rL+test_draw_networkx_arrowsize_incorrect_sizer4Ls 33344A I M   (( Y''''((((((((((((((((((sAA!Ar2)rmrr5cdtjgd}dddd}tj|||}t|trt j|n|}t||D]@\}}t|tj j sJ| |ksJAdS)Nrr1rrrrOrr)r*r2) r>r8rrprrrepeatrrErrr7)r2rKr*ryrr7s rLtest_draw_edges_arrowsizer:Us +++,,AF + +C  "1# C C CE/9)S/I/IX  +++yIUI..44 X#s{:;;;;;%%''83333344rNrlrhrhrz<|-|>ctjgd}dddd}tj|||}t|trt j|n|}tjj j tjj j tjj j d}t||D]P\}}t|tjjsJt|||sJQdS)Nr7rrr8r)r*rlr;)r>r8rrprrr9rErr"r$r# CurveFilledABrrr%)rlrKr*ry arrow_objectsrr7s rLtest_draw_edges_arrowstyler?bs +++,,AF + +C  "1#* E E EE)3:s(C(CS $$$ {%2k$-'5M UJ//II X#s{:;;;;;#,,.. h0GHHHHHHIIrNcptjttddgdS)Nr1r2r)r>rrjrbarrayrTrNrLtest_np_edgelistrBws/Wrxx0@'A'ABBBBBBrNctjd}ddd}tjtjd5tj||ddddS#1swxYwYdS)NrYrrrzhas no positionr*)r>r?r,r-r.rrKr*s rL*test_draw_nodes_missing_node_from_positionrE|s aA C r'/@ A A A'' q#&&&''''''''''''''''''sAA!$A!c|\}}tjd}d|D}gd}tj||d||}t|dksJdS)zEnsure draw_networkx_nodes works when node_shape is a Python list. This covers the case where node_shape is a sequence (list) and node_color is a single scalar color, which should be supported. rci|]}|||fSrTrT)rVis rLrXzNtest_draw_networkx_nodes_node_shape_list_with_scalar_color..rrN)rrrrrr9)r rrrN)r> empty_graphrrzrc)rarbrrKr*shapesrZs rL:test_draw_networkx_nodes_node_shape_list_with_scalar_colorrKs GC qA a C & & &F  "      E u  "" # #a ' ' ' ' ' 'rNc|\}}tjdg}ddd}tj||||d}|ddddf}tj||||dd d}|ddddf} | d|dksJ| d |d ksJdS) 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 rrr8rrrNrarBrrrrrO)r>r8rrDrE) rrarbrrKr* default_patchrG padded_patchrIs rL)test_draw_edges_min_source_target_marginsrRs&GC F8A C*1cbZPPPQRSM"..0088::33Q36BN)    L!,,..66881a@M  nQ/ / / / /  nQ/ / / / / / /rNc `|\}}tjddg}dddd}tj||||}d|D}tj||||d d gd d g }d |D} t|| D]-\} } | d | d ksJ| d| dksJ.dS)rMrrrr8)rarrrNcpg|]3}|ddddf4SNrarrDrE)rVds rLruzHtest_draw_edges_min_source_target_margins_individual..s;OOOAammoo--//!Q7OOOrNbfrOcpg|]3}|ddddf4SrUrVrs rLruzHtest_draw_edges_min_source_target_margins_individual..s;MMM1Q]]__,,..sssAv6MMMrNrrON)r>r8rr) rrarbrrKr*rPrGrQrIrWrs rL4test_draw_edges_min_source_target_margins_individualr[s&GC FF#$$AF + +C*1cbZPPPMOOOOON)  s)s) LNM MMMMNM221tad{{{{tad{{{{{ rNc|\}}tj}|d|ddtj|ddid}|}|jdkr |jdksJdS)zDEnsure that selfloop extent is non-zero when there is only one node.rrN)r>r8rWrXrrDrrY)rarbrrKpatchrs rL&test_nonzero_selfloop_with_single_noder^sGC AJJqMMMJJq!  "1q&k 2 21 5E     D :>>dkAooo -oorNc2|\}}tjdtj}|ddd|jD}tj||dgd}|}|jdkr |jdksJdS) zfEnsure that selfloop extent is non-zero when only a single edge is specified in the edgelist. rarrOci|]}|||fSrTrTrUs rLrXzFtest_nonzero_selfloop_with_single_edge_in_edgelist.. & & &1q!f & & &rNrrrN) r>r?r8rXrZrrDrrY)rarbrrKr*r]rs rL2test_nonzero_selfloop_with_single_edge_in_edgelistrb s GC abj111AJJq! & &ag & & &C  "1cVH = = =a @E     D :>>dkAooo -oorNcgd}gd}d}tjj|||}t |dddf|ksJdS)zfTest apply_alpha when there is a mismatch between the number of supplied colors and elements. rrr)Nr)r>drawingnx_pylab apply_alphar[)r colorlistr rgba_colorss rLtest_apply_alphariscyyHI E*%11)UHMMK {111b5!U* + +++ +++rNcddl}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 rNrYrci|]}|||fSrTrTrUs rLrXz>test_draw_edges_toggling_with_arrows_kwarg..4s ! ! !1q!f ! ! !rNTrF)rrr>r?r8rrzryrprErrrqrr)rUGrr*rKrys rL*test_draw_edges_toggling_with_arrows_kwargrm's"!!! q  B qrz 2 2 2B ! !b ! ! !C"XAA&q#d;;;5zzS\\))))%(CK$?@@@@@@"XAA&q#e<<<%!?@@@@@@  "2s + +E eS_; < <<< <  "2s + +E u::QW % % % % eAh ; < <<< <<.Ps1UUz!S_;<<UUUUUUrN)rr>r?anyrqrrnrarrKrbrs rL,test_draw_networkx_arrows_default_undirectedrtIss!!!! aAGCLr UUbnUUU U UUU Uz>rNcddl}tjdtj}|\}}|||t d|jDrJ|jsJdS)NrrYrrc3TK|]#}t|tjjV$dSrrorqs rLrz=test_draw_networkx_arrows_default_directed..[sB:; 1co455rN)rr>r?r8rrrqrrss rL*test_draw_networkx_arrows_default_directedrwTs!!!! abj111AGCLr?A~  ::rNctjd}|dd|\}}tj|ddg||jrJdS)NrYrrr)rr)r>r?rXrr)rarKrbrs rLtest_edgelist_kwarg_not_ignoredryas] aAJJq!GCGA(R0000z>rN)rKexpected_n_edgesrac BtgdD]1\}\}}|||t|dz d2d|D}dddggdfD]E}tj|||tj|||}t ||ksJFd S) zGDraws edges correctly for 3 types of graphs and checks for valid lengthrrrr1rYrarci|]}|||fSrTrTrUs rLrXzFtest_draw_networkx_edges_multiedge_connectionstyle..rrrN arc3,rad=0.1)rr arc3,rad=0.2connectionstyleN) enumeraterXr%r>rrz)rKrzrHrjrkr* conn_stylers rL2test_draw_networkx_edges_multiedge_connectionstylerjs ???@@11 6Aq 1aa!eQ 0000 a C  (888// q#zBBBB'3 KKK6{{......//rNc tgdD]1\}\}}|||t|dz d2d|D}tj||gd}ddd ggdfD]W}tj|||}t ||ksJ|D]} | jj d ksJXd S) zXDraws labels correctly for 3 types of graphs and checks for valid length and class namesr|rYrar}ci|]}|||fSrTrTrUs rLrXzLtest_draw_networkx_edge_labels_multiedge_connectionstyle..rrN)rrrrrrCurvedArrowTextN) rrXr%r>rrrzvalues __class____name__) rKrzrHrjrkr*rr text_itemstis rL8test_draw_networkx_edge_labels_multiedge_connectionstyler~s% ???@@11 6Aq 1aa!eQ 0000 a C  # 3 P P PF  (888>> 1!S*UUU :"22222##%% > >B<(,====== >>>rNc,tj}|ddd|dddtj|d}d|D}tj|||dd g }t |d ksJdS) NrrOrmr}rrci|]}|||fSrTrTrUs rLrXz;test_draw_networkx_edge_label_multiedge..rrNrrrrra)r>r%rXget_edge_attributesrrz)rKrr*rs rL'test_draw_networkx_edge_label_multiedgers AJJq!BJJJq!BJ(H55K a C-  '8 J z??a      rNctjd}d|jD}tj||iiksJdS)zORegression test for draw_networkx_edge_labels with empty dict. See gh-5372.rYci|]}|||fSrTrTrUs rLrXz.rarNrN)r>r?rZrrDs rL(test_draw_networkx_edge_label_empty_dictrsQ aA & &ag & & &C '3B ? ? ?2 E E E E E ErNcR|\}}gd}gd}tj|}d|jD}tj|||||t |jdksJt |ddt dd gz}t|j|dd|D]c\}} } | | sJtj || sJddS) ztWhen an edgelist is supplied along with a sequence of colors, check that the self-loops have the correct colors.)r3rrrrr)pinkcyanrr9rgreenci|]}|||fSrTrTrUs rLrXzGtest_draw_networkx_edges_undirected_selfloop_colors..rarN)rrr#rYNrr)r>rrZrrzrrbrArrcontains_pointrErFrGr) rarbrr edge_colorsrKr* sl_pointsrclrslps rL3test_draw_networkx_edges_undirected_selfloop_colorsrs5GC???HCCCK A & &ag & & &C1cb8 TTTT rz??a     "##''"((As8*<*<>>>>>??rNfap_only_kwargrrrrrctjd}d|D}|\}}t|d}t jt d|d5tj||fd|i|dddn #1swxYwYtj 5tj d tj d t tj||f|d d |ddddS#1swxYwYdS)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.rYci|]}|||fSrTrTrUs rLrXzEtest_user_warnings_for_unused_edge_drawing_kwargs..rrNrz The z# keyword argument is not applicabler*rNerrorignore)categoryT)rr) r>r?rrr,warns UserWarningrwarningscatch_warnings simplefilterfilterwarningsDeprecationWarning)rrarKr*rbr kwarg_names rL1test_user_warnings_for_unused_edge_drawing_kwargsrs aA a CGCn))++,,Q/J UjUUU   @@ q#??"????@@@@@@@@@@@@@@@  " "MMg&&&3EFFFF q#L"TLL^LLL MMMMMMMMMMMMMMMMMMs%,BBB,AC??DDdraw_fnc|\}}tjd}tjd5}|||dddn #1swxYwYt |dksJdS)NrT)recordrr)r> cycle_graphrrrz)rrarbrrKws rL*test_no_warning_on_default_draw_arrowstylersGC qA   - - -b q66Q;;;;;;sA  A A rNmethodctjd}d|jD}|\}}||||||j|jfD])}t ||ksJ*dS)NrYci|]}|||fSrTrTrUs rLrXz#test_hide_ticks..rarN)r*rrN)r>r?rZrOrPrQrR)rrNrarKr*rmrrSs rLtest_hide_ticksrs aA & &ag & & &C EAr F1#"444428$99D''))**j8888899rN)angleangle3arcz arc3,rad=0.0zbar,fraction=0.1c|\}}d}tj|g}d|D}|dd}tj|||di|}|dd|ddzdz } |dd|ddzdz } |d vr<||js-J||jt j| | fkd S|d kr||j| ksJd Sd S) z Check that FancyArrowPatches with all `connectionstyle`s are supported in edge label rendering. See gh-7735 and gh-8106. rci|]}||dfSrrTrUs rLrXz8test_edge_label_all_connectionstyles..rrN,rrrrOra>rr1r,N)r>r8splitrr yr,approx) rarrbrrrKr*namer hmidvmids rL$test_edge_label_all_connectionstylesrs GC D D6A a C ;;s  A D  ) 3T6NEF F1IAq !Q &D F1IAq !Q &D d|~LLvd|~d|1L1LLLL~LL d|~$$$$ $$rN label_posgg?c|\}}d}tj|g}d|D}tj|||di|d}||js-J||jt j||fkdS)zH Check that label positions can be extrapolated outside [0, 1]. rci|]}|||fSrTrTrUs rLrXz-test_edge_label_label_pos..)rrNrr)rrrN)r>r8rr rr,r)rarrbrrrKr*lbls rLtest_edge_label_label_posr!s GC D D6A a C & 3T6NiQX   C t9;LLD v}i5K'L'LLLL;LLrN)u__doc__rosrr,networkxr>rirErbuserArcParams barbell_graphrjdefaultsmark parametrizerMr]rer~rr8rrrrrrrrr rr)r/r:r>rJrTr\r_fixturera draw_circularrh draw_planar draw_random draw_spectralr~rtdraw_forceatlas2roruryrrrrrrrrrrrrrrrrrrrrrrrrrrrr(r/r4r:r?rBrErKrRr[r^rbrirmrrrtrwryr%rrrrrrrrrrrrrrrrTrNrLrs22  f,''V!! f-..# ] " 1a , , D, ),  ,       , "##, $#%, &', (), *D+, ,#-, .)/, 0$O?SS  1, H#I, J#K, NsO, PaQ, RW, , , ^- tXl34('"  "*'=>>?>  *"*'=>>?>("*'=>>?>>$"*'=>>?>8"*'=>>?>( &(>?@ QF 1b'"*'=>>?>, &(>?@ sCj 3*"*'=>>?>4   6      "z22//32/d5$-0010&PPP           &&  &&&&&&&&&&***  #  H%   f   dVH%566DD76  D #  H%   f   dVH%566 D D76   D6F6F6FrDDDDDD      "*'=>>,,?>, a333 a///    555 = = = . . .  , , ,     s     , ,   ,55549F9F9FzNNN  KKK,EEE666$666.888&$$$ (((r<<<&899 4 4:9 4/E/E/E'FGGIIHGI(CCC '''(((:z22)0)032)0\z22))32)X . . ....",,,===D"'23C)DEEFE"'23C)DEE  FE bjllA!,r/@/@!.DE / /  / bjllA!,r/@/@!.DE>> >(    FFF???6 s b N+ b! b!   MM  M.RWb.>$?@@A@t}55   $      99  659 K K K%%%.tSk22 M M32 M M MrN