G8c/dZddlZddlZddlmZddlmZddlZddlm Z ddl Z ddl Z ddl Z ddl mZddlmZddlmZddlmZddlmZddlmZmZd Ze j<j?d d Z d Z!edgddZ"dZ#dZ$dZ%dZ&edgddZ'eddZ(dZ)dZ*edgddd Z+ed!gdd"Z,d#Z-ed$gd%&d'Z.d(Z/ed)Z0e j<jcd*d+d,gedd-Z2d.Z3d/Z4ed0gddd1Z5ed2Z6d3Z7ed4Z8ed5gd6Z9y)7z% Tests specific to the lines module. N)SimpleNamespace)cycler)assert_array_equal) MarkerStyle)Path)image_comparisoncheck_figures_equalc d\}}tjddgtjddg}}d}ttj|||||dgy)zTest a problematic case.)i)igH@gW@g@g(\@rN)nparrayrmlines segment_hits)cxcyxyradiuss =/usr/lib/python3/dist-packages/matplotlib/tests/test_lines.pytest_segment_hitsrsR FB 88T4L !288S$K#8qA Fv**2r1a@1#F)rerunsc&d}tjdd|}tjj|}t j }t j }tj||}|jdttj|jjdd}|j|ttj|jjdd}||z }d} || ksJy ) ac GitHub issue #1256 identified a bug in Line.draw method Despite visibility attribute set to False, the draw method was not returning early enough and some pre-rendering code was executed though not necessary. Consequence was an excessive draw time for invisible Line instances holding a large number of points (Npts> 10**6) irsizeFr)numberrepeatN)r linspacerandomnormalpltfiguresubplotr Line2D set_visiblemintimeitrcanvasdrawadd_line) Nrrfigaxl t_no_linet_invisible_lineslowdown_factorslowdown_thresholds rtest_invisible_Line_renderingr5"s A Aq!A a A **,C B  aAMM% FMM#**//!AFGIKKN6==1MN'2O / // /rctj\}}tjj d|j tjj dddgyNrr)rr) linestylesr#subplotsr r!seedcontourrandnr.r/s rtest_set_line_coll_dashrBNsDllnGCIINN1JJryyr2&K=JAr line_dashesT) remove_textcltj\}}|jtdddy)N r: linestylelw)r#r=plotrangerAs rtest_line_dashesrMVs'llnGCGGE"IG3rc~tj\}}|jtdd|jtdd|jtdd|jtdd|jtdd|jj y)NrFnone)colorrz.3)rrrr)rrrr#r=rKrLr*r+rAs rtest_line_colorsrS]sllnGCGGE"IVG$GGE"ISG!GGE"ITG"GGE"I\G*GGE"IYG'JJOOrctjgg}tjt5|j ddddy#1swYyxYwNfoobar)r r&pytestraises ValueError set_colorlines rtest_valid_colorsr]gs? ==R D z "! x !!! A  Ac tj\}}ddddddddd d d gff D]}|jtd |  |jj y)N-solid--dashedz-.dashdot:dotted)rN)rrrF)rIrR)r.r/lss rtest_linestyle_variantsrimsellnGCGT8YX'Ar7) b R( ) JJOOrctjgg}tjt5|j ddddy#1swYyxYw)Naardvark)r r&rWrXrY set_linestyler[s rtest_valid_linestylesrmwsA ==R D z "' :&'''r^zdrawstyle_variants.pngctjd\}}gd}t|j|D]4\}}|j t d||j dd6y)N)defaultz steps-mid steps-prez steps-poststepsN drawstyle)rr)xlimylim)r#r=zipflatrKrLset)r.axsdssr/dss rtest_drawstyle_variantsr~}s\||AHC LC chh$)B d r* F()r)png) extensionsc"|j}tjd}|j|dz||j}t j j dd}|j||||jzy)Nrsg@@) transform) add_subplotr arangerK mtransformsAffine2D translate transData)fig_reffig_testr/rts rtest_no_subslice_with_transformrsr    B $AGGAHa    B((5AGGAqAbllNG+rctjgg}tjt5|j ddddy#1swYyxYwrU)r r&rWrXrY set_drawstyler[s rtest_valid_drawstylesrsA ==R D z "% 8$%%%r^ctjddtjzd}tj|}t j \}}|j ||\}|jdt|jjdt|zdz k(sJ|jdt|jjt|k(sJy)NrrrFrqrrp) r r pisinr#r=rKrlenget_pathvertices)rrr.r/r\s rtest_set_drawstylers Aqw#A q AllnGC GGAqMED{# t}}'' (Ac!fHQJ 66 6y! t}}'' (CF 22 2rline_collection_dashesmpl20)rDstylectj\}}tjj d|j tjj dddgyr7r<rAs rtest_set_line_coll_dash_imagersBllnGCIINN1JJryyr2&K=JArzmarker_fill_styles.pngcbtjgddddddtjgdg}d}tjd d g}tjd d g}t j \}}t tjjD]i\}}t tjjD]>\}} t|} |j|d z|z||zd |dzzz|d|| | d| | d @k|jd dg|jddgy)Nrrrgz#ff0000cmr)rrr lightgreenrr rF?rr8rG) marker markersizemarkerfacecoloralt fillstylelabel linewidthrPmarkeredgecolormarkeredgewidthg@) itertoolscycler r r#r= enumerater r&filled_markers fillStylesnextrKset_ylimset_xlim) colorsaltcolorrrr.r/jrifsrPs rtest_marker_fill_stylesrs __iic3 hhy134FH !QA !QAllnGCv}};;< ' 6v}}778 'EArLE GGAFQJAa!e 4!!'/ $)$%  ' ' 'KKCKKS rctjgdtddd\}|jdk(sJ|j dk(sJy)z=Test that markerfacecolor does not override fillstyle='none'.rrrorO)rred)rmarkerfacecolorN)r#rKr get_fillstyleget_markerfacecolor)r0s rtest_markerfacecolor_fillstylersL )Kv$F"' )BA ??  && & "f ,, ,r scaled_linesrp)rc |tjdd}tj\}}gd}t t j d}tt||D]W\}\}}tjdddD]5}|j||tjdzd|zzf||d |7Yy) Nr )rcrfrdaxes.prop_cyclerrF2皙?rH) r r r#r=r matplotlibrcParamsrrxrKones) thr.r/ lins_stylesrrrhstyrJs rtest_lw_scalingrs Q BllnGC1K  ##$56 7B!#k2"67M 9B++b"b) MB BGGB"''"+ R/ L2" L L MMrcLtjgg}|jtjgdsJ|jtjdtj dgsJ|jddgtj gdzzddgzrJy)NrrrrrrGdrr)r r& _is_sortedr r nanr[s rtest_nan_is_sortedrs ==R D ??288I. // / ??288QN3 44 41v36!Q?@@ @@rc|jjddgd|jjgdgddddgy)Nrrz-or)rrrr markevery)r=steprKrrs rtest_step_markersrsC aVT* Iy$1a&Irparentr$axesc  tjjdtjddd}tjj t |}dddgdddgt d d d tjt ||d kDd dg }gd}|dk(r|dd}|dd}fd}fd}nY|dk(rTtjddj tjddj fd} fd}|D] } ||| |D]Q} tjt| jtjt} ||| Sy)N*rr)rrG)rrG rGrFrr333333?)rg?) 111111111111111000100010001000100001000010010001000001001000000000000100000101010000011100011101101101101101111001010011011101r$cVjtj||d|yNrrr add_artistr r&)rrrrs radd_testz test_markevery..add_tests$    a3)D FrcVjtj||d|yrr)rrrrs radd_refztest_markevery..add_refs$    a3)D FrrrcBtj||d|yNz-gDrrrK)rrraxs_tests rrz test_markevery..add_test s N  1ey  ArcBtj||d|yrr)rrraxs_refs rrztest_markevery..add_refs M  q!Ui  @rr)r r!r>r randrsliceriterr=ryr listastypeintbool) rrrrr cases_test cases_refrrcasemerrs `` @@rtest_markeveryr sbIINN2 Aq"A s1vA6:b'5B?))CF#AG,z#JGI_ crN  F F 6 ))!Q/445w''1-223 B A'A&'$ XXd4j ! ( ( - 4 4T :1#$rctj}|jtjddgddgddt j t5|jjdddy#1swYyxYw)Nrrrrr) r#r$rr r&rWrXrYr*r+)r.s r.test_markevery_figure_line_unsupported_relsizer s_ **,CNN6==!Q!QrJK z " s A>>Bctj\}}|jgdtd\}|jj |j dk(sJ|jd|jj |jtd|jj tjddgddgdd gg}|jgd t|d \}|jgd |d \}t|j j|jt|j j|jy)N)rrrD)rsrrrrrr)rms)rrr) r#r=rKrr*r+ get_marker set_markerr_create_closedrr)r.r/r\ triangle1line2line3s rtest_marker_as_markerstyler sllnGC GGIk#&6G 7EDJJOO ??  ## # OOCJJOOOOK$%JJOO##b"X2wA$?@I WWY{9'="W EFE WWYyRW 8FEu'')22I4F4FGu'')22I4F4FGrzstriped_line.pngc 2tjjd}tj\}}|j |j dddddd |j |j dd d d ddd |jdy)Ni!N, rorangebluerbrG )rPgapcolorrIrJrrblack)r)rrGrrr)rPrrIrJralpha) handlelength)r r! default_rngr#r=rKuniformlegend)rng_r/s rtest_striped_linesr'4s ))   )C LLNEArGGCKKRK 6q -GGCKKRK 'AS EII1Irc|jjddggd|jjddggdy)Nrrr)dashes)rrrrrr)rrKrs rtest_odd_dashesr*?sC Ay9 1v.@Arcntj\}}t|jjdz|jj dzdz}|j ddgddgd\}|j|\}}|rJ|j ddgddgdd \}|j|\}}|sJt|d dg|j ddgddgd\}|j|\}}|rJ|jd|j|\}}|sJt|d dgy) Nr)rrrrT)pickerr8)r- pickradiusind) r#r=rbboxwidthheightrKcontainsrset_pickradius)r.r/ mouse_eventl0foundindicesl1l2s r test_pickingr;Es5llnGC!CHHNNa$7$'HHOOq$82$=?K ''1a&1a&' .CB[[-NE79 ''1a&1a&"' =CB[[-NE7 L5wu~s+ ''1a&1a&' .CB[[-NE79b[[-NE7 L5wu~s+rctjddd}|jj||d\}t d|dd|j d|jjgdgdddy) Nrrorz.-rrr)rrrrt)r rrrKrLr)rrrr0s rtest_input_copyr=_sn !QA     $ $Q4 0BA 8AaDOOG y)TWMrrc dddgdddgtddd d d d d dg }tjd}|tjddt |}tjdd}d|dzz}|j }t|D] \}} |j||z d| ||"t||tjd<|j } t|D]\}} | j||z dy)z*Test that we can set markevery prop_cycle.N)r9r?)r9rrrrrrg?)rr)g?rjetg?g?rrGrzo-)rrPr) rmpl colormapsr r rrrrKrrr) rrcasescmaprrrr{rrr/r&s rtest_markevery_prop_cyclerHjs1g|aW 3Q c3  &E == D "++c3E 3 4F BA AqDA    C!%(D 9 Q CD.4e:@.BJ)*    B% 1 Atr):__doc__rr)typesrrnumpyr numpy.testingrrWrrDmatplotlib.lineslinesr matplotlib.markersrmatplotlib.pathrmatplotlib.pyplotpyplotr#matplotlib.transforms transformsrmatplotlib.testing.decoratorsrr rmarkflakyr5rBrMrSr]rirmr~rrrrrrrrr parametrizer r rr'r*r;r=rHrgrrrYs !, !* +OG!(0(0VB=/t4454 ! ' +,$? )@ )),*,% 3+,$gNBOB +,$?@6->")4M5MAJJ Hf#56))$*7)$XH(%&DHIBB ,4NN()r