o  hQ @sdZddlZddlZddlZddlZedZedZededZ 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&dRdSZ'ejj(dTdUZ)dVdWZ*dXdYZ+dZd[Z,d\d]Z-d^d_Z.d`daZ/dbdcZ0dddeZ1dfdgZ2dhdiZ3ejdjdkgdlfdmdnZ4ejdodpgdqfdrdsZ5dtduZ6dvdwZ7ejdxdydzd{Z8ejdxdyd|d}Z9d~dZ:ddZ;ddZe j?fddZ@ejde j>e j?fddZAddZBejde dge Cd ge Dd gfddZEejde dge Cd ge Dd gfddZFddZGddZHddZIejddodLidjdiddiddiddifddZJejde j>e jKfddZLejdddgejde j?e jMe jNe jOe jPgddZQdS)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#U/var/www/vscode/kcb/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_colormap9s  r4cCsttdSN)rrr to_directedr#r#r#r$ test_arrowsAsr7)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_undirectedFs 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| |sIJqz8test_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_interpretation{sT     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$rZ6r[z9test_directed_edges_linestyle_default..r solidNrrGrRr+rcrH get_linestyle)rMrFrNrTr#r#r$%test_directed_edges_linestyle_default3s 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$rZMr[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$rZer[z:test_directed_edges_linestyle_sequence..rr N) rrGrRr+rcrHrirrjr)rrMrFrNrTrr#r#r$&test_directed_edges_linestyle_sequenceUsrc Csddlm}m}ddlm}ttj}ttj}t |}t |}t ||}t ||s1Jtj ||dd}t |t s@Jt|dkrOt |d|sOJtj ||dd}t ||s^Jtj ||dd}t ||smJtj ||dd}t |t s|Jt|dkrt |d|sJdSdS)Nr)LineCollectionPathCollection)FancyArrowPatchTarrowsF)matplotlib.collectionsrrmatplotlib.patchesrr cubical_graphGraphrR spring_layoutdraw_networkx_nodesr{rHr*rc) rrrrMdGrFdposnodesedgesr#r#r$test_return_typesms,        rc CsTt}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!<d"d#td D}tj|||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)rCrmin_source_marginmin_target_marginr r-z$a$rz$b$r?z$c$r`z$d$r z$\alpha$rz$\beta$rz$\gamma$rz$\delta$rcSs"i|] }||ddkr dndqS)r`rkr9r#rWr#r#r$rZs"z*test_labels_and_colors..) font_size)r font_colorF) edge_labelsrotaterz4-5r)rrrrrHr+draw_networkx_labelsdraw_networkx_edge_labels)rMrFlabelsr3r#r#r$test_labels_and_colorssl   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)rrrrMr#r#r$test_empty_graphsrcCsddl}tddg}tddg}t|}ttj||gdtjj s&Jttj||gdtjj s5Jtj ||gddgksBJtj ||gddgksOJtj ||gddgks\Jtj ||gddgksiJdS)Nrr_rb)rT)rCrF) rrrrRrr{rrJ collectionsrrH)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_subplotrrcrr+append)rFrrtr3r#rr$test_alpha_iters      rcCsRtd}td}tj|gd|ddd|D}t|dks'JdS)Nro)ohs^) node_shapercSg|] }t|tjjr|qSr#)r{rJrr)rXrr#r#r$r z-test_multiple_node_shapes..)rrGrrrr get_childrenrc)rMrscattersr#r#r$test_multiple_node_shapess rcCst}td}tj||dd|Ddd|Ddt|dd|DD]%\}}|dr:d nd }t j | |sGJt |d d |ksUJq0dS) NrcSsi|] }||dr dndqS)r`rr9r#rWr#r#r$rZ(z7test_individualized_font_attributes..cSsi|] }|t|ddqS)""""""@r)intrWr#r#r$rZ)r)rrrcSs.g|]}t|tjjrt|dkr|qS)r)r{rJtextTextrcget_text)rXtr#r#r$r-s z7test_individualized_font_attributes..r`rr:rr)rkarate_club_graphrrrrrrirrJr3rKrLrget_size)rMrrYrr8r#r#r$#test_individualized_font_attributes"s$ rcCst}td}dd|D}dd|D}tj||d||ddd|D}t||D]3\}}| d|d d |d d dksQJt |d r\t j j jnt j j j}t||sjJq7dS) NrcSs(g|]\}}||ddkrdndqS)r`r-|>-[r#rXuvr#r#r$r<(z7test_individualized_edge_attributes..cSs(g|]\}}d|d|ddqS)r&r`r#rr#r#r$r=rT)rr arrowstyle arrowsizecSrr#)r{rJpatchesr)rXfr#r#r$r?rr&rr`r?)rrrrrrrrriget_mutation_scalesumrJr ArrowStyleBracketB CurveFilledBr{get_arrowstyle)rMr arrowstyles arrowsizesrrar8r#r#r$#test_individualized_edge_attributes9s ,  rcCsBtjtddtjtddWddS1swYdS)NzReceived invalid argumentmatchbar)foo)pytestraises ValueErrorrrrr#r#r#r$test_error_invalid_kwdsLs"rcCsXtgd}gd}tjtddtj||dWddS1s%wYdS)N)rDrr`rr r?r )r?r`r z1arrowsize should have the same length as edgelistr)r)rrRrrrr)rMrr#r#r$+test_draw_networkx_arrowsize_incorrect_sizeQs"rr)r&rrcCs|tgd}dddd}tj|||d}t|trt|n|}t||D]\}}t|tj j s3J| |ks;Jq&dS)NrDrr_rrDrrr?r`)rFr) rrRrHr{rrrepeatrirJrrr )rrMrFrrTr8r#r#r$test_draw_edges_arrowsizeZs r#rrrrz<|-|>cCstgd}dddd}tj|||d}t|trt|n|}tjj j tjj j tjj j d}t ||D]\}}t|tjjsBJt|||sMJq5dS)Nr rrDrr!)rFrr$)rrRrHr{strrr"rJrr r r  CurveFilledABrirr)rrMrFr arrow_objectsrTr8r#r#r$test_draw_edges_arrowstylegs r(cCstjttddgddS)Nrrr)rrrnparrayr#r#r#r$test_np_edgelist|sr+cCsVtd}ddd}tjtjddt||WddS1s$wYdS)Nr rrrDzhas no positionr)rrGrr NetworkXErrorrrMrFr#r#r$*test_draw_nodes_missing_node_from_positions  "r.r)rrc Cst\}}tdg}ddd}tj||||dd}|ddddf}tj||||ddd d}|ddddf}|d|dksMJ|d |d ksWJdS) 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 rDrrrrrNr`rrrrrr?)rrrrRrH get_extentscorners) rrrrMrF default_patchdefault_extent padded_patch padded_extentr#r#r$)test_draw_edges_min_source_target_marginss$   r8c Cst\}}tddg}dddd}tj||||d}dd |D}tj||||d d gd d gd }d d |D}t||D]"\} } td| d| | d| dksVJ| d| dks`Jq>dS)r/rDr_rr)r`rr!r0cS&g|]}|ddddfqSNr`rr2r3)rXdr#r#r$r&zHtest_draw_edges_min_source_target_margins_individual..bfr1cSr9r:r;)rXpr#r#r$rr=zp=z, d=rr?N)rrrrRrHriprint) rrrrMrFr4r5r6r7r<r@r#r#r$4test_draw_edges_min_source_target_margins_individuals&  rBcCstt\}}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_edgerHr2r heightdelaxesclose)rrrMpatchbboxr#r#r$&test_nonzero_selfloop_with_single_nodes     rJcCst\}}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$rZr[zFtest_nonzero_selfloop_with_single_edge_in_edgelist..rrrN) rrrrGrRrDrrHr2r rErFrG)rrrMrFrHrIr#r#r$2test_nonzero_selfloop_with_single_edge_in_edgelists    rKcCsFgd}gd}d}tjj|||}t|dddf|ks!JdS)zfTest apply_alpha when there is a mismatch between the number of supplied colors and elements. r!rfrNro)rdrawingnx_pylab apply_alphaall)r colorlistrt rgba_colorsr#r#r$test_apply_alpha s  rRcCs 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$rZ%r[z>test_draw_edges_toggling_with_arrows_kwarg..TrF)rrrrGrRrHrcrr{rJrrrr)rUGrrFrMrr#r#r$*test_draw_edges_toggling_with_arrows_kwargs"     rT drawing_funccCsbddl}td}t\}}|||dtdd|jDs!J|jr&Jt|t dS)Nrr rcs|] }t|tjjVqdSr5r{rJrrrXrnr#r#r$ Asz?test_draw_networkx_arrows_default_undirected..) rrrGrranyrrrFrGrUrrMrrr#r#r$,test_draw_networkx_arrows_default_undirected:s      r\cCshddl}tjdtjd}t\}}|||dtdd|jDr$J|js)Jt |t dS)Nrr rPrcsrVr5rWrXr#r#r$rYNs z=test_draw_networkx_arrows_default_directed..) rrrGrRrrrZrrrFrGr[r#r#r$*test_draw_networkx_arrows_default_directedGs     r]cCsVtd}|ddt\}}tj|ddg|d|jr Jt|tdS)Nr rrDr_)rCr) rrGrDrrrrrFrG)rMrrr#r#r$test_edgelist_kwarg_not_ignoredVs      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$rZir[zFtest_draw_networkx_edges_multiedge_connectionstyle.. arc3,rad=0.1)rcrc arc3,rad=0.2connectionstyleN) enumeraterDroundrrHrc)rMr_irrrF conn_stylerr#r#r$2test_draw_networkx_edges_multiedge_connectionstyleasrkc 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`racSrVr#r#rWr#r#r$rZ}r[zLtest_draw_networkx_edge_labels_multiedge_connectionstyle..)rcrcrcrercrdCurvedArrowTextN) rgrDrhrrHrrcvalues __class____name__) rMr_rirrrFrrj text_itemstir#r#r$8test_draw_networkx_edge_labels_multiedge_connectionstyleus   rrcCslt}|jdddd|jddddt|d}dd|D}tj|||d d gd }t|d ks4JdS) Nrr?r&rarrbcSrVr#r#rWr#r#r$rZr[z;test_draw_networkx_edge_label_multiedge..rcrd)rrfr`)r MultiGraphrDget_edge_attributesrrc)rMrrFrpr#r#r$'test_draw_networkx_edge_label_multiedges rucCs6td}dd|jD}tj||idiksJdS)zORegression test for draw_networkx_edge_labels with empty dict. See gh-5372.r cSrVr#r#rWr#r#r$rZr[z.rN)rrGrrr-r#r#r$(test_draw_networkx_edge_label_empty_dicts rvc 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)rrrrrrHrcrr)r*riget_pathcontains_pointrJr3rKrSrFrG) rrrC edge_colorsrMrF sl_pointsrTclrslpr#r#r$3test_draw_networkx_edges_undirected_selfloop_colorss      rfap_only_kwargrrfrdrr&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 simplefilterrFrG)rrMrFrr kwarg_namer#r#r$1test_user_warnings_for_unused_edge_drawing_kwargss       rdraw_fncCstt\}}td}tjdd}|||dWdn1s"wYt|dks/Jt|tdS)NrT)recordrr) rrr cycle_graphrrrcrFrG)rrrrMwr#r#r$*test_no_warning_on_default_draw_arrowstyles    r hide_ticksTmethodcCsrtd}dd|jD}t\}}|||||d|j|jfD] }t||ks-Jq!t |t dS)Nr cSrVr#r#rWr#r#r$rZr[z#test_hide_ticks..)rFrr) rrGrrrxaxisyaxisboolget_ticklabelsrFrG)rrrMrF_rrr#r#r$test_hide_tickss    r)R__doc__rrrr importorskiprJr)userrcParamsnetworkxr barbell_graphrr%r,r4r7mark parametrizerOrUrermrprxryrrRr}rrrrrrrrrmpl_image_comparerrrrrrrrrrrr#r(r+r.r8rBrJrKrRrTrrr\r]r^rsrrkrrrurvrrrrrrHrrrr#r#r#r$sD           9             =      . - "