o ]ZhQ @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 matplotlibnumpyZPSzmatplotlib.pyplotFz text.usetexcCszOz1tjtjtjtjtjtjtjg}ddddg}t ||D]\}}|t fi|t dqWn t y;YnwWztdWdStyOYdSwztdWwty`Yww)Nblackd) node_color node_sizewidthtest.ps)nx draw_circularZdraw_kamada_kawaiZ draw_planarZ draw_randomZ draw_spectral draw_spring draw_shell itertoolsproductbarbellpltsavefigModuleNotFoundErrorosunlinkOSError)Z functionsoptionsfunctionoptionrP/var/www/auris/lib/python3.10/site-packages/networkx/drawing/tests/test_pylab.py test_draws8     rcCsz3ttdttddttddg}tjt|dtdWztdWdSt y3YdSwztdWwt yDYww)Nr Znlistr ) listranger rrrrrrrr"rrrtest_draw_shell_nlist-s&   r%cCs(tt}tjt|dtjjdddS)NrT) edge_colorr Z edge_cmapZ with_labels)r$rZnumber_of_edgesr rrcmZBlues)colorsrrrtest_edge_colormap9s  r)cCsttdSN)r rr to_directedrrrr test_arrowsAsr,)r&expected)Nr)rredr.r/)?r1gyellowr0r2)rrr4limer3r5)#0000ffbluer6r7edgelistrr4cCs<td}tj|t|||d}tj||sJdS)z`Tests ways of specifying all edges have a single color for edges drawn with a LineCollectionrposr8r&N)r path_graphdraw_networkx_edges random_layoutmplr( same_color get_color)r&r-r8G drawn_edgesrrr!test_single_edge_color_undirectedFs rDcCsLtjdtjd}tj|t|||d}|D] }tj||s#JqdS)zaTests ways of specifying all edges have a single color for edges drawn with FancyArrowPatchesrZ create_usingr:N) r r<DiGraphr=r>r?r(r@ get_edgecolor)r&r-r8rBrCfaprrrtest_single_edge_color_directed`srIcCstjdtjd}ddtt|D}dD]5}tj|||d}|D] }tj| |s.Jq!tj||ddg|d }|D] }tj| |sIJqz8test_edge_color_tuple_interpretation..)rrr4rrr4r4r&r9r4)r8r&r9rSrTrrQr7)r9rSrV)rrrPrr4rTrN) r r<rFr$lenr=r?r(r@rG)rBr;ecrCrHrrr$test_edge_color_tuple_interpretation{sT     rYcCsZt}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.r.gbrRN) rr+r r>r=ziprcycler?r(r@rGrBr;Z edgecolorsrCrHr-rrr.test_fewer_edge_colors_than_num_edges_directeds r`cCsdtjdtjd}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.rrE)r.r[r\crRN) r r<rFr>rr=r]r?r(r@rGr_rrr-test_more_edge_colors_than_num_edges_directeds rccCsltjtttddgddd}|}t|dksJtj |ddds,J|ddks4JdSNr9rSpurple皙?)r;r8r&alpharTrb) r r=rr>rAZsqueezerW get_pathsr?r(r@)Zedge_collectionrXrrr3test_edge_color_string_with_global_alpha_undirecteds ricCsrtjtttddgddd}t|dksJ|D]}|}tj |ddds.J|ddks6JqdSrd) r r=rr+r>rWrGr?r(r@)rCrHrXrrr1test_edge_color_string_with_global_alpha_directedsrj graph_typecCsVtjd|d}ddtt|D}t||}t|tr!|d}|dks)JdS)z^Test the default linewidth for edges drawn either via LineCollection or FancyArrowPatches.rTrEcSrJrrrKrrrrNrOz1test_edge_width_default_value..rr4N)r r<r$rWr= isinstancer# get_linewidth)rkrBr;rCrrrtest_edge_width_default_values   rn) edgewidthr-rrrcCsXtd}ddtt|D}tj|||d}t|dks"J||ks*JdS)NrcSrJrrrKrrrrNrOz;test_edge_width_single_value_undirected..r r)r r<r$rWr=rhrm)ror-rBr;rCrrr'test_edge_width_single_value_undirecteds rrcCsdtjdtjd}ddtt|D}tj|||d}t|dks#J|D] }||ks/Jq%dS)NrrEcSrJrrrKrrrrN rOz9test_edge_width_single_value_directed..rqr)r r<rFr$rWr=rm)ror-rBr;rCrHrrr%test_edge_width_single_value_directedsrsrUrScCsVt}t|}d}tj||||d}t|t|D] \}}||ks(JqdS)N)?@g(@)r8r ) rr+r r>r=r]rr^rm)r8rBr;widthsrCrHZexpected_widthrrrtest_edge_width_sequences rwcCsxtjdtjd}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).rrErr1rRcSg|]}|qSrrGrLerrr *z7test_edge_color_with_edge_vmin_vmax..rfg?)r&Z edge_vminZ edge_vmaxcSrxrryrzrrrr|/r}N)r r<rFr>r=r?r(r@)rBr;rCZ orig_colorsZ scaled_colorsrrr#test_edge_color_with_edge_vmin_vmax#s r~cCs`tjdtjd}ddtt|D}t||}t|dks!J|D] }|dks-Jq#dS)z>Test default linestyle for edges drawn with FancyArrowPatches.rrEcSrJrrrKrrrrN6rOz9test_directed_edges_linestyle_default..rZsolidNr r<rFr$rWr= get_linestyle)rBr;rCrHrrr%test_directed_edges_linestyle_default3s rstyle)dashed--r4r4r4cCsdtjdtjd}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).rrEcSrJrrrKrrrrNMrOz>test_directed_edges_linestyle_single_value..rrNr)rrBr;rCrHrrr*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).rrEcSrJrrrKrrrrNerOz:test_directed_edges_linestyle_sequence..rrN) r r<rFr$rWr=r]rr^r)rrBr;rCrHrrrr&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_graphGraphrF spring_layoutdraw_networkx_nodesrlr=r#rW) rrrrBZdGr;Zdposnodesedgesrrrtest_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),Nrr4rTrr.i?)nodelistr r rg)rrr\)g?rtrr1r1rt)r rg)r9rSrV)rr)r8r rgr&)rr)rr)rr)rrT)r8rmin_source_marginmin_target_marginr r&z$a$rz$b$r4z$c$rTz$d$rz$\alpha$rz$\beta$rz$\gamma$rz$\delta$rcSs"i|] }||ddkr dndqS)rTrkr.rrKrrrrNs"z*test_labels_and_colors..) font_size)r font_colorF) edge_labelsrotaterz4-5r)r rrrr=r$draw_networkx_labelsdraw_networkx_edge_labels)rBr;labelsr(rrrtest_labels_and_colorssl   rcCst}t\}}dddddd}tj||dgdd d tj||d d gd d tj||ddd|dttd|S)Nrrr4rr9r)rtru)rr4rTrri rztab:blue)r rr irz tab:orangertr)rgr g)\(?off) r Z house_graphrsubplotsrr=ZmarginsZ tight_layoutaxis)rBfigaxr;rrrtest_house_with_colorss$    rcCs4t\}}tjt|dtjttt|ddS)Nr)rrr drawrrcircular_layout)rrrrr test_axess rcCst}t|dSr*)r rrrBrrrtest_empty_graphsrcCsddl}tddg}tddg}t|}ttj||gdtjj s&Jttj||gdtjj s5Jtj ||gddgksBJtj ||gddgksOJtj ||gddgks\Jtj ||gddgksiJdS)NrrSrV)rT)r8rF) rr rrFrrlrr? collectionsrr=)rrBDGr;rrr#test_draw_empty_nodes_return_valuess rcCs:tjdtjd}tj|dgdtj|dggdddS)NrrE)rr4rr8)r r)r8r )r r< MultiDiGraph draw_networkxrrrrtest_multigraph_edgelist_tuplessrcstt}t}|dtjt|ddgdttjfddt D}t }|dtjt|||d| d |d tjt||ddS) N皙?rf)rgcsg|]}|qSrr)rLxZ num_nodesrrr|r}z#test_alpha_iter..)r rgr4) r r>rrfigure add_subplotrrWrr$append)r;rrgr(rrrtest_alpha_iters      rcCsRtd}td}tj|gd|ddd|D}t|dks'JdS)Nro)ohs^) node_shapercSg|] }t|tjjr|qSr)rlr?rr)rLrrrrr| z-test_multiple_node_shapes..)r r<rrrr get_childrenrW)rBrZscattersrrrtest_multiple_node_shapess rcCst}td}tj||dd|Ddd|Ddt|dd|DD]%\}}|dr:d nd }t j | |sGJt |d d |ksUJq0dS) NrcSsi|] }||dr dndqS)rTrr.rrKrrrrN(z7test_individualized_font_attributes..cSsi|] }|t|ddqS)""""""@r)intrKrrrrN)r)rrrcSs.g|]}t|tjjrt|dkr|qS)r)rlr?textTextrWZget_text)rLtrrrr|-s z7test_individualized_font_attributes..rTrr/rr)r karate_club_graphrrrrrr]rr?r(r@rArget_size)rBrrMrr-rrr#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)rTr-|>-[rrLuvrrrr|<(z7test_individualized_edge_attributes..cSs(g|]\}}d|d|ddqS)r rTrrrrrr|=rT)rr arrowstyle arrowsizecSrr)rlr?patchesr)rLfrrrr|?rr rrTr4)r rrrrrrrr]get_mutation_scalesumr?r ArrowStyleBracketB CurveFilledBrlget_arrowstyle)rBrZ arrowstylesZ arrowsizesrr{ar-rrr#test_individualized_edge_attributes9s ,  rcCsBtjtddtjtddWddS1swYdS)NzReceived invalid argumentmatchbar)Zfoo)pytestraises ValueErrorr rrrrrrtest_error_invalid_kwdsLs"rcCsXtgd}gd}tjtddtj||dWddS1s%wYdS)N)r9rrTrrr4r)r4rTrz1arrowsize should have the same length as edgelistr)r)r rFrrrr)rBrrrr+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)Nr9rrSrr9rrr4rT)r;r) r rFr=rlrrrepeatr]r?rrr)rrBr;rrHr-rrrtest_draw_edges_arrowsizeZs rrrrrz<|-|>cCstgd}dddd}tj|||d}t|trt|n|}tjj j tjj j tjj j d}t ||D]\}}t|tjjsBJt|||sMJq5dS)Nrrr9rr)r;rr)r rFr=rlstrrrr?rrrrZ CurveFilledABr]rr)rrBr;rZ arrow_objectsrHr-rrrtest_draw_edges_arrowstylegs rcCstjttddgddS)Nrrr)r rrnparrayrrrrtest_np_edgelist|sr cCsVtd}ddd}tjtjddt||WddS1s$wYdS)Nrrrr9zhas no positionr)r r<rrZ NetworkXErrorrrBr;rrr*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 r9rrrrrNrTrrrrrr4)rrr rFr= get_extentscorners) rrrrBr; default_patchdefault_extent padded_patch padded_extentrrr)test_draw_edges_min_source_target_marginss$   rc 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 r9rSrr)rTrrr cS&g|]}|ddddfqSNrTrrr)rLdrrrr|&zHtest_draw_edges_min_source_target_margins_individual..bfrcSrrr)rLprrrr|rzp=z, d=rr4N)rrr rFr=r]print) rrrrBr;rrrrrrrrr4test_draw_edges_min_source_target_margins_individuals&  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) rrr rFadd_nodeadd_edger=rr heightdelaxesclose)rrrBpatchbboxrrr&test_nonzero_selfloop_with_single_nodes     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. rTrEr4cSrJrrrKrrrrNrOzFtest_nonzero_selfloop_with_single_edge_in_edgelist..rrrN) rrr r<rFr!rr=rr r"r#r$)rrrBr;r%r&rrr2test_nonzero_selfloop_with_single_edge_in_edgelists    r(cCsFgd}gd}d}tjj|||}t|dddf|ks!JdS)zfTest apply_alpha when there is a mismatch between the number of supplied colors and elements. rrZrtNrb)r ZdrawingZnx_pylabZ apply_alphaall)rZ colorlistrgZ rgba_colorsrrrtest_apply_alpha s  r*cCs 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 rNrrEcSrJrrrKrrrrN%rOz>test_draw_edges_toggling_with_arrows_kwarg..TrF)rrr r<rFr=rWrrlr?rrrr)rZUGrr;rBrrrr*test_draw_edges_toggling_with_arrows_kwargs"     r+ drawing_funccCsbddl}td}t\}}|||dtdd|jDs!J|jr&Jt|t dS)Nrrrcs|] }t|tjjVqdSr*rlr?rrrLrarrr Asz?test_draw_networkx_arrows_default_undirected..) rr r<rranyrrr#r$r,rrBrrrrr,test_draw_networkx_arrows_default_undirected:s      r3cCshddl}tjdtjd}t\}}|||dtdd|jDr$J|js)Jt |t dS)NrrrErcsr-r*r.r/rrrr0Ns z=test_draw_networkx_arrows_default_directed..) rr r<rFrrr1rrr#r$r2rrr*test_draw_networkx_arrows_default_directedGs     r4cCsVtd}|ddt\}}tj|ddg|d|jr Jt|tdS)Nrrr9rS)r8r) r r<r!rrrrr#r$)rBrrrrrtest_edgelist_kwarg_not_ignoredVs      r5)rBexpected_n_edgesrTcCstgdD]\}\}}|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 lengthr9r9r9rrrTweightcSrJrrrKrrrrNirOzFtest_draw_networkx_edges_multiedge_connectionstyle.. arc3,rad=0.1)r:r: arc3,rad=0.2connectionstyleN) enumerater!roundr r=rW)rBr6irrr; conn_stylerrrr2test_draw_networkx_edges_multiedge_connectionstyleasrBc 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 namesr7rrTr8cSrJrrrKrrrrN}rOzLtest_draw_networkx_edge_labels_multiedge_connectionstyle..)r:r:r:r<r:r;ZCurvedArrowTextN) r>r!r?r r=rrWvalues __class____name__) rBr6r@rrr;rrA text_itemstirrr8test_draw_networkx_edge_labels_multiedge_connectionstyleus   rHcCslt}|jdddd|jddddt|d}dd|D}tj|||d d gd }t|d ks4JdS) Nrr4r r8rr9cSrJrrrKrrrrNrOz;test_draw_networkx_edge_label_multiedge..r:r;)rr=rT)r MultiGraphr!Zget_edge_attributesrrW)rBrr;rFrrr'test_draw_networkx_edge_label_multiedges rJcCs6td}dd|jD}tj||idiksJdS)zORegression test for draw_networkx_edge_labels with empty dict. See gh-5372.rcSrJrrrKrrrrNrOz.rN)r r<rrr rrr(test_draw_networkx_edge_label_empty_dicts rKc 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.)rrSrVrrp)rTrT)Zpinkcyanrr/r7greencSrJrrrKrrrrNrOzGtest_draw_networkx_edges_undirected_selfloop_colors..)rr8r&rNrr)rrr rrr=rWrrrr]get_pathZcontains_pointr?r(r@rGr#r$) rrr8Z edge_colorsrBr;Z sl_pointsrHZclrZslprrr3test_draw_networkx_edges_undirected_selfloop_colorss      rPfap_only_kwargrr=r;rr rcCstd}dd|D}t\}}t|d}tjtd|ddtj ||fd|i|Wd n1s.rz The z# keyword argument is not applicablerrNerrorT)rr)r r<rrr#keysrZwarns UserWarningr=warningscatch_warnings simplefilterr#r$)rQrBr;rrZ kwarg_namerrr1test_user_warnings_for_unused_edge_drawing_kwargss       rXdraw_fncCstt\}}td}tjdd}|||dWdn1s"wYt|dks/Jt|tdS)NrT)recordrr) rrr Z cycle_graphrUrVrWr#r$)rYrrrBwrrr*test_no_warning_on_default_draw_arrowstyles    r\ hide_ticksTmethodcCsrtd}dd|jD}t\}}|||||d|j|jfD] }t||ks-Jq!t |t dS)NrcSrJrrrKrrrrNrOz#test_hide_ticks..)r;rr]) r r<rrrZxaxisZyaxisboolZget_ticklabelsr#r$)r^r]rBr;_rrrrrtest_hide_tickss    ra)R__doc__rrrUrZ importorskipr?rZuserZrcParamsZnetworkxr Z barbell_graphrrr%r)r,markZ parametrizerDrIrYr`rcrirjrrFrnrrrsrwr~rrrrrZmpl_image_comparerrrrrrrrrrrrrr r rrr'r(r*r+rrr3r4r5rIrrBrHrJrKrPrXrr\rr=rrrarrrrsD           9             =      . - "