o hv @s@dZddlZddlZddlZddlZedZedZededZ de j d<ddl Z e d d Zd d Zd dZddZddZejddddgdfddgdfddgdfddgd ff ejd!dd"gfd#d$Zejddddgdfddgdfddgdfddgd ff ejd!dd"gfd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zejd1e je jfd2d3Zejd4d5d6gd6ffd7d8Zejd4d5d6gd6ffd9d:Z ejd!gd;dd"dZ!d?d@Z"dAdBZ#ejdCdDdEdFZ$ejdGdHgdIgdJggdKdIdLggdMfdNdOZ%dPdQZ&ejj'dRdSZ(dTdUZ)dVdWZ*dXdYZ+dZd[Z,d\d]Z-d^d_Z.d`daZ/ejdbdcgddfdedfZ0dgdhZ1didjZ2ejdkdldmdnZ3dodpZ4dqdrZ5dsdtZ6dudvZ7ejdwe j8e j9fdxdyZ:ejdwe j8e j9fdzd{Z;d|d}Zd gfddZ?ejd~e dge =d ge >d gfddZ@ddZAddZBddZCejdddLidbdiddiddiddifddZDejde j8e jEfddZFejdddgejde j9e jGe jHe jIe jJgddZKdS)z,Unit tests for matplotlib drawing functions.N matplotlibnumpyPSzmatplotlib.pyplotFz text.usetexcCszOz1tjtjtjtjtjtjtjg}ddddg}t ||D]\}}|t fi|t dqWn t y;YnwWztdWdStyOYdSwztdWwty`Yww)Nblackd) node_color node_sizewidthtest.ps)nx draw_circulardraw_kamada_kawai draw_planar draw_random draw_spectral draw_spring draw_shell itertoolsproductbarbellpltsavefigModuleNotFoundErrorosunlinkOSError) functionsoptionsfunctionoptionr#\/var/www/html/env_mimamsha/lib/python3.10/site-packages/networkx/drawing/tests/test_pylab.py test_draws8     r%cCsz3ttdttddttddg}tjt|dtdWztdWdSt y3YdSwztdWwt yDYww)Nr nlistr ) listrangerrrrrrrrr(r#r#r$test_draw_shell_nlist,s&   r,cCs(tt}tjt|dtjjdddS)NrT) edge_colorr edge_cmap with_labels)r+rnumber_of_edgesrrrcmBlues)colorsr#r#r$test_edge_colormap8s  r4cCsttdSN)rrr to_directedr#r#r#r$ test_arrows@sr7)r-expected)Nr)rredr9r:)?r<gyellowr;r=)rrr?limer>r@)#0000ffbluerArBedgelistrr?cCs<td}tj|t|||d}tj||sJdS)z`Tests ways of specifying all edges have a single color for edges drawn with a LineCollectionr posrCr-N)r path_graphdraw_networkx_edges random_layoutmplr3 same_color get_color)r-r8rCG drawn_edgesr#r#r$!test_single_edge_color_undirectedEs rOcCsLtjdtjd}tj|t|||d}|D] }tj||s#JqdS)zaTests ways of specifying all edges have a single color for edges drawn with FancyArrowPatchesr  create_usingrEN) rrGDiGraphrHrIrJr3rK get_edgecolor)r-r8rCrMrNfapr#r#r$test_single_edge_color_directed_srUcCstjdtjd}ddtt|D}dD]5}tj|||d}|D] }tj| |s.Jq!tj||ddg|d }|D] }tj| |sIJq~z8test_edge_color_tuple_interpretation..)rrr?rrr?r?r-rDr?)rCr-rDr_r`r r]rB)rDr_rb)r rr\rr?r`r N) rrGrRr+lenrHrJr3rKrS)rMrFecrNrTr#r#r$$test_edge_color_tuple_interpretationzsT     recCsZt}tt}d}tj|||d}t|t|D]\}}tj | |s*JqdS)zZTest that the edge colors are cycled when there are fewer specified colors than edges.r9gbr^N) rr6rrIrHziprcyclerJr3rKrSrMrF edgecolorsrNrTr8r#r#r$.test_fewer_edge_colors_than_num_edges_directeds rmcCsdtjdtjd}tt}d}tj|||d}t||ddD]\}}tj | |s/Jq dS)z\Test that extra edge colors are ignored when there are more specified colors than edges.rrP)r9rgrhcr^N) rrGrRrIrrHrirJr3rKrSrkr#r#r$-test_more_edge_colors_than_num_edges_directeds rpcCsltjtttddgddd}|}t|dksJtj |ddds,J|ddks4JdSNrDr_purple皙?)rFrCr-alphar`ro) rrHrrIrLsqueezerc get_pathsrJr3rK)edge_collectionrdr#r#r$3test_edge_color_string_with_global_alpha_undirecteds rxcCsrtjtttddgddd}t|dksJ|D]}|}tj |ddds.J|ddks6JqdSrq) rrHrr6rIrcrSrJr3rK)rNrTrdr#r#r$1test_edge_color_string_with_global_alpha_directedsry graph_typecCsVtjd|d}ddtt|D}t||}t|tr!|d}|dks)JdS)z^Test the default linewidth for edges drawn either via LineCollection or FancyArrowPatches.r`rPcSrVr#r#rWr#r#r$rZr[z1test_edge_width_default_value..rr?N)rrGr+rcrH isinstancer* get_linewidth)rzrMrFrNr#r#r$test_edge_width_default_values   r}) edgewidthr8r r r cCsXtd}ddtt|D}tj|||d}t|dks"J||ks*JdS)NrcSrVr#r#rWr#r#r$rZr[z;test_edge_width_single_value_undirected..r r )rrGr+rcrHrvr|)r~r8rMrFrNr#r#r$'test_edge_width_single_value_undirecteds rcCsdtjdtjd}ddtt|D}tj|||d}t|dks#J|D] }||ks/Jq%dS)NrrPcSrVr#r#rWr#r#r$rZ r[z9test_edge_width_single_value_directed..rr )rrGrRr+rcrHr|)r~r8rMrFrNrTr#r#r$%test_edge_width_single_value_directedsrrar_cCsVt}t|}d}tj||||d}t|t|D] \}}||ks(JqdS)N)?@g(@)rCr ) rr6rrIrHrirrjr|)rCrMrFwidthsrNrTexpected_widthr#r#r$test_edge_width_sequences rcCsxtjdtjd}t|}tj||ddgd}dd|D}tj||dd gdd d }d d|D}tj||s:Jd S) zyTest that edge_vmin and edge_vmax properly set the dynamic range of the color map when num edges == len(edge_colors).r rPrr<r^cSg|]}|qSr#rSrXer#r#r$ )z7test_edge_color_with_edge_vmin_vmax..rsg?)r- edge_vmin edge_vmaxcSrr#rrr#r#r$r.rN)rrGrRrIrHrJr3rK)rMrFrN orig_colors scaled_colorsr#r#r$#test_edge_color_with_edge_vmin_vmax"s rcCs`tjdtjd}ddtt|D}t||}t|dks!J|D] }|dks-Jq#dS)z>Test default linestyle for edges drawn with FancyArrowPatches.rrPcSrVr#r#rWr#r#r$rZ5r[z9test_directed_edges_linestyle_default..r solidNrrGrRr+rcrH get_linestyle)rMrFrNrTr#r#r$%test_directed_edges_linestyle_default2s rstyle)dashed--r?r?r?cCsdtjdtjd}ddtt|D}tj|||d}t|dks#J|D] }||ks/Jq%dS)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).rrPcSrVr#r#rWr#r#r$rZLr[z>test_directed_edges_linestyle_single_value..rr Nr)rrMrFrNrTr#r#r$*test_directed_edges_linestyle_single_value>s r style_seqrrr)r-:r)rrrz-.cCsttjdtjd}ddtt|D}tj|||d}t|dks#Jt|t|D] \}}| |ks7Jq+dS)zTests support for specifying linestyles with sequences in ``draw_networkx_edges`` for FancyArrowPatch outputs (e.g. directed edges).rrPcSrVr#r#rWr#r#r$rZdr[z:test_directed_edges_linestyle_sequence..rr N) rrGrRr+rcrHrirrjr)rrMrFrNrTrr#r#r$&test_directed_edges_linestyle_sequenceTsrc Cs.t}t|}tj||gdddddtj||gdddgddtj||d d d tj||gd d d ddtj||gdd d ddtj||gddd dd ddi}d|d<d|d<d|d<d|d<d|d<d|d<d|d<d |d!<tj|||d"d#tj||dd$d%tj||d&d'id(dS))Nrr?r`r r9i?)nodelistr r rt)rrrh)g?rrr<r<r)r rt)rDr_rb)r r)rCr rtr-)rr)rr)rr)rrT)rCarrowsmin_source_marginmin_target_marginr r-z$a$rz$b$r?z$c$r`z$d$r z$\alpha$rz$\beta$rz$\gamma$rz$\delta$r) font_sizeF) edge_labelsrotaterz4-5r)r cubical_graph spring_layoutdraw_networkx_nodesrHdraw_networkx_labelsdraw_networkx_edge_labels)rMrFlabelsr#r#r$test_labels_and_colorslsh   rcCst}t\}}dddddd}tj||dgdd d tj||d d gd d tj||ddd|dttd|S)Nrrr?rrDr)rr)rr?r`r ri rztab:blue)r rr irz tab:orangerr)rtr g)\(?off) r house_graphrsubplotsrrHmargins tight_layoutaxis)rMfigaxrFr#r#r$test_house_with_colorss$    rcCs4t\}}tjt|dtjttt|ddS)Nr)rrrdrawrrcircular_layout)rrr#r#r$ test_axess rcCst}t|dSr5)rGraphrrMr#r#r$test_empty_graphsrcCsddl}tddg}tddg}t|}ttj||gdtjj s&Jttj||gdtjj s5Jtj ||gddgksBJtj ||gddgksOJtj ||gddgks\Jtj ||gddgksiJdS)Nrr_rb)rT)rCrF) matplotlib.collectionsrrrRrr{rrJ collectionsPathCollectionrH)rrMDGrFr#r#r$#test_draw_empty_nodes_return_valuess rcCs:tjdtjd}tj|dgdtj|dggdddS)Nr rP)rr?rrC)r&r)rCr )rrG MultiDiGraph draw_networkxrr#r#r$test_multigraph_edgelist_tuplessrcstt}t}|dtjt|ddgdttjfddt D}t }|dtjt|||d| d |d tjt||ddS) N皙?rs)rtcsg|]}|qSr#r#)rXx num_nodesr#r$rrz#test_alpha_iter..)r rtr?) rrIrrfigure add_subplotrrcnodesr+append)rFrrtr3r#rr$test_alpha_iters      rcCsBtjtddtjtddWddS1swYdS)NzReceived invalid argumentmatchbar)foo)pytestraises ValueErrorrrrr#r#r#r$test_error_invalid_kwdss"rcCsXtgd}gd}tjtddtj||dWddS1s%wYdS)N)rDrr`rr r?r )r?r`r z1arrowsize should have the same length as edgelistr) arrowsize)rrRrrrr)rMrr#r#r$+test_draw_networkx_arrowsize_incorrect_sizes"rr)r&rrcCs|tgd}dddd}tj|||d}t|trt|n|}t||D]\}}t|tj j s3J| |ks;Jq&dS)N)rDrr_rrDrrr?r`)rFr) rrRrHr{intrrepeatrirJpatchesFancyArrowPatchget_mutation_scale)rrMrFedgesrTr8r#r#r$test_draw_edges_arrowsize s rcCstjttddgddS)Nrrr)rrrnparrayr#r#r#r$test_np_edgelistsrcCsVtd}ddd}tjtjddt||WddS1s$wYdS)Nr rrrDzhas no positionr)rrGrr NetworkXErrorrrMrFr#r#r$*test_draw_nodes_missing_node_from_positions  "r node_shape)osc Cst\}}tdg}ddd}tj||||dd}|ddddf}tj||||ddd d}|ddddf}|d|dksMJ|d |d ksWJdS) 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 rDrr)rrrNr`r)rrrrr?)rrrrRrH get_extentscorners) rrrrMrF default_patchdefault_extent padded_patch padded_extentr#r#r$)test_draw_edges_min_source_target_margins$s$   rcCstt\}}t}|d|ddt|ddid}|}|jdkr-|j dks/Jt |t dS)zDEnsure that selfloop extent is non-zero when there is only one node.rrN) rrrrRadd_nodeadd_edgerHrr heightdelaxesclose)rrrMpatchbboxr#r#r$&test_nonzero_selfloop_with_single_nodeQs     r cCst\}}tjdtjd}|dddd|jD}tj||dgdd}|}|j dkr5|j dks7Jt |t d S) zfEnsure that selfloop extent is non-zero when only a single edge is specified in the edgelist. r`rPr?cSrVr#r#rWr#r#r$rZlr[zFtest_nonzero_selfloop_with_single_edge_in_edgelist..rrrN) rrrrGrRrrrHrr rr r )rrrMrFr r r#r#r$2test_nonzero_selfloop_with_single_edge_in_edgelistcs    rcCsFgd}gd}d}tjj|||}t|dddf|ks!JdS)zfTest apply_alpha when there is a mismatch between the number of supplied colors and elements. rrfrNro)rdrawingnx_pylab apply_alphaall)r colorlistrt rgba_colorsr#r#r$test_apply_alphaws  rcCs ddl}ddl}td}tjdtjd}dd|D}||fD] }tj||dd}t|t|jks5Jt|dt j j s@Jq ||fD]}tj||d d}t|t j j sXJqEt||}t|t j j shJt||}t|t|jksyJt|dt j 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 rPcSrVr#r#rWr#r#r$rZr[z>test_draw_edges_toggling_with_arrows_kwarg..T)rF)rmatplotlib.patchesrrGrRrHrcrr{rJrrrLineCollection)rUGrrFrMrr#r#r$*test_draw_edges_toggling_with_arrows_kwargs"     r drawing_funccCsbddl}td}t\}}|||dtdd|jDs!J|jr&Jt|t dS)Nrr rcs|] }t|tjjVqdSr5r{rJrrrXrnr#r#r$ sz?test_draw_networkx_arrows_default_undirected..) rrrGrranyrrr r rrrMrrr#r#r$,test_draw_networkx_arrows_default_undirecteds      r!cCshddl}tjdtjd}t\}}|||dtdd|jDr$J|js)Jt |t dS)Nrr rPrcsrr5rrr#r#r$rs z=test_draw_networkx_arrows_default_directed..) rrrGrRrrrrrr r r r#r#r$*test_draw_networkx_arrows_default_directeds     r"cCsVtd}|ddt\}}tj|ddg|d|jr Jt|tdS)Nr rrDr_)rCr) rrGrrrrrr r )rMrrr#r#r$test_edgelist_kwarg_not_ignoreds      r#)rMexpected_n_edgesr`cCstgdD]\}\}}|j||t|dddqdd|D}dddggdfD]}tj|||d tj|||d }t||ksDJq*d S) zGDraws edges correctly for 3 types of graphs and checks for valid lengthrDrDrDrr r`weightcSrVr#r#rWr#r#r$rZr[zFtest_draw_networkx_edges_multiedge_connectionstyle.. arc3,rad=0.1)r(r( arc3,rad=0.2connectionstyleN) enumeraterroundrrHrc)rMr$iuvrF conn_stylerr#r#r$2test_draw_networkx_edges_multiedge_connectionstylesr2c CstgdD]\}\}}|j||t|dddqdd|D}tj||gdd}d d d ggdfD]!}tj|||d}t||ksFJ|D] } | jj d ksTJqJq4d S) zXDraws labels correctly for 3 types of graphs and checks for valid length and class namesr%r r`r&cSrVr#r#rWr#r#r$rZr[zLtest_draw_networkx_edge_labels_multiedge_connectionstyle..)r(r(r(r*r(r)CurvedArrowTextN) r,rr-rrHrrcvalues __class____name__) rMr$r.r/r0rFrr1 text_itemstir#r#r$8test_draw_networkx_edge_labels_multiedge_connectionstyles   r9cCslt}|jdddd|jddddt|d}dd|D}tj|||d d gd }t|d ks4JdS) Nrr?r&r&rr'cSrVr#r#rWr#r#r$rZr[z;test_draw_networkx_edge_label_multiedge..r(r))rr+r`)r MultiGraphrget_edge_attributesrrc)rMrrFr7r#r#r$'test_draw_networkx_edge_label_multiedges r<cCs6td}dd|jD}tj||idiksJdS)zORegression test for draw_networkx_edge_labels with empty dict. See gh-5372.r cSrVr#r#rWr#r#r$rZ r[z.rN)rrGrrrr#r#r$(test_draw_networkx_edge_label_empty_dicts r=c Cst\}}gd}gd}t|}dd|jD}tj|||||dt|jdks.Jt |ddt d d g}t |j|dd|D]\}}} | | sWJt j||sbJqIt|tdS) ztWhen an edgelist is supplied along with a sequence of colors, check that the self-loops have the correct colors.)rr_rbrr)r`r`)pinkcyanrr:rBgreencSrVr#r#rWr#r#r$rZr[zGtest_draw_networkx_edges_undirected_selfloop_colors..)rrCr-r Nrr)rrrrrrHrcrrrriget_pathcontains_pointrJr3rKrSr r ) rrrC edge_colorsrMrF sl_pointsrTclrslpr#r#r$3test_draw_networkx_edges_undirected_selfloop_colorss      rHfap_only_kwarg arrowstylerr+r)rr&rcCstd}dd|D}t\}}t|d}tjtd|ddtj ||fd|i|Wd n1s.rz The z# keyword argument is not applicablerrNerrorT)rr)rrGrrr*keysrwarns UserWarningrHwarningscatch_warnings simplefilterr r )rIrMrFrr kwarg_namer#r#r$1test_user_warnings_for_unused_edge_drawing_kwargs+s       rSdraw_fncCstt\}}td}tjdd}|||dWdn1s"wYt|dks/Jt|tdS)NrT)recordrr) rrr cycle_graphrOrPrcr r )rTrrrMwr#r#r$*test_no_warning_on_default_draw_arrowstyleNs    rX hide_ticksTmethodcCsrtd}dd|jD}t\}}|||||d|j|jfD] }t||ks-Jq!t |t dS)Nr cSrVr#r#rWr#r#r$rZhr[z#test_hide_ticks..)rFrrY) rrGrrrxaxisyaxisboolget_ticklabelsr r )rZrYrMrF_rrr#r#r$test_hide_ticks[s    r`)L__doc__rrrOr importorskiprJruserrcParamsnetworkxr barbell_graphrr%r,r4r7mark parametrizerOrUrermrprxryrrRr}rrrrrrrrmpl_image_comparerrrrrrrrrrrrr rrrrrr!r"r#r:rr2r9r<r=rHrSrrXrrHrrr`r#r#r#r$s4           9             ;    , "