(?ev<ddlZddlZddlmZmZmZmZddlZddl m Z ddl m Z ddlmZddlmZddlmZmZmZddlmZddlmZmZdZdZed gd d d dZdZdZ dZ!dZ"dZ#dZ$dZ%Gddej&Z'GddZ(GddZ)dZ*dZ+dZ,dZ-d Z.d!Z/d"Z0d#Z1d$Z2d%Z3d&Z4d'Z5d(Z6ej78d)gd*d+Z9d,Z:d-Z;d.Z<ed/g0d1Z=d2Z>d3Z?dS)4N)assert_allcloseassert_almost_equalassert_array_equalassert_array_almost_equal)scale)Affine2DBboxTransformedBbox)Path)image_comparisoncheck_figures_equalc\Gddtj}|}tj}tjt jd||jztjd|_ |j tjdS)NcBeZdZdZdxZZdZfdZdZeZ dZ e Z xZ S).AssertingNonAffineTransformz This transform raises an assertion error when called when it shouldn't be and ``self.raise_on_transform`` is True. Fctj|i|d|_tjdd|_dS)NF )super__init__raise_on_transform mtransformsrrunderlying_transform)selfargskwargs __class__s @lib/python3.11/site-packages/matplotlib/tests/test_transforms.pyrzEtest_non_affine_caching..AssertingNonAffineTransform.__init__sN EGG d -f - - -&+D #(3(<(>(>(D(DR(L(LD % % %cX|jr Jd|j|SNz3Invalidated affine part of transform unnecessarily.)rrtransform_pathrpaths rtransform_path_non_affinezVtest_non_affine_caching..AssertingNonAffineTransform.transform_path_non_affine s@. F FE F F F,;;DAA ArcX|jr Jd|j|Sr )rr transformr"s rtransform_non_affinezQtest_non_affine_caching..AssertingNonAffineTransform.transform_non_affine&s@. F FE F F F,66t<< >>>HJJJ#'HLHJJJJJrcGdd}tj}tjtjd|d\}|dd|ddt|j tj d dS)NceZdZdZdZdS)-test_external_transform_api..ScaledByc||_dSN) _scale_factor)r scale_factors rrz6test_external_transform_api..ScaledBy.__init__9s!-D   rchtj|j|jzSrD)rrrrEr9)rr5s r_as_mpl_transformz?test_external_transform_api..ScaledBy._as_mpl_transform<s.(**001CDDn% &rN)r(r)r*rrHrrScaledByrB8s2 . . . & & & & &rrJrr2rd)r4r5r6r7r8set_xlimset_ylimr get_transform_a get_matrixrrr)rJr>lines rtest_external_transform_apirR7s&&&&&&&& B HRYr]]hhrll ; ; ;EDKK3KK3D&&((+6688(**0044??AACCCCCrpre_transform_dataTmpl20g?) remove_textstyletolc <tj}tjd}|t jddd||j z| t j dddt j d dd t jddd||j z| t j ddt j dd||j zt j ddd }t j d d d }dt j |zt j|ddt jfz}t j |t j|ddt jfz }|||||||j zt j|||ddd|ddd}}|ddddddf|ddddddf}}|||d z||||j z||dz |d z|dz|dz||j zdS)Nr0r2rg@ r)r& linewidth)r4r5rrrcontourfr7r8reshaper9 pcolormeshlinspacescattersincosnewaxis streamplothypotquiverbarbs)r>times10xyuvs rtest_pre_transform_plottingruIso B"$$**2..GKK " %%a++w7MKNNNMM"+aA&&+c1a(()B--''1--#bl2444 JJr{1b!!2;r1#5#5 2</111 Ar2A Aq"A "&)) bfQqqq"*}-...A q BF1QQQ ]+,,,AMM!Q1",(>HQNN,,, SSqS61SSqS6qA SSqS##A#X;##A#sss( qAIIaQ1",(>I???HHQUAE1a4A2<1GHHHHHHrctj}tjtjdddtjddd\}}|||tj||ztj d|j ztj ddgd d gg}t||j dS) Nr_(@)@皙?r2?ףp= ?@?)r4r5r7meshgridrgrdlogrrrr9arrayrdataLim get_pointsr>xsysexpecteds r!test_contour_pre_transform_limitsrns B [RR00"+dD"2M2M N NFBKKBrBw%.0066s;;blJLLLx#td%&&H"*"7"7"9"9:::::rctj}tjtjdddtjddd\}}|||tj||zddddftj d|j ztj dd gd d gg}t||j dS Nrwr_rxryrzr2r{r|r}r~)r4r5r7rrgpcolorrrrrr9rrrrrs r test_pcolor_pre_transform_limitsrys B [RR00"+dD"2M2M N NFBIIb"bfR"Woocrc3B3h/#,..44S99BLHJJJx#td%&&H"*"7"7"9"9:::::rctj}tjtjdddtjddd\}}|||tj||zddddftj d|j ztj dd gd d gg}t||j dSr)r4r5r7rrgrfrrrrr9rrrrrs r$test_pcolormesh_pre_transform_limitsrs B [RR00"+dD"2M2M N NFBMM"b"&b//#2#ss(3'02288== LNNNx#td%&&H"*"7"7"9"9:::::rc tjdtjddd}tjddd}tj||\}}tjtjj|jdz}tj }| dd }| d d| |||d |j dS)N!N,rrcrKiropolar projectionegouraud)shading)r7randomseedrgrradiansrandshaper4figure add_subplotset_rlimrfcanvasr:)valuesradiizrrrqfigr>s rtest_pcolormesh_gouraud_nansrsINN8 [C # #F KT2 & &E ;vu % %DAq 29>17+c122A *,,C  1 1BKKTMM!Q9M---JOOrctjddgddgddgg}tjdddddd}||}tjddgddgddgg}t ||tjdddddd}||}tjddgddgddgg}t ||tjdddddd}||}tjddgd dgddgg}t ||tjdddd dd}||}tjddgdd gddgg}t ||tjddddd d}||}tjd dgd dgd dgg}t ||tjdddddd }||}tjdd gdd gdd gg}t ||dS)Nrrr_rr`rrc<r\PrarZ)r7rrr from_valuesr&r)pointstactualrs rtest_Affine2D_from_valuesrs X1vBxAwF ((Aq!Q::A [[ Fx!Q"a2q'233H)))((Aq!Q::A [[ Fx!Q!R1b'233H)))((Aq!Q::A [[ Fx!Q"a1a&122H)))((Aq!Q::A [[ Fx!Q!R1a&122H)))((Aq!Q::A [[ Fx!Q!Q!Q011H)))((Aq!Q::A [[ Fx!Q!Q!Q011H)))))rcddg}tj}t|||||ddt||||dS)N?)rrrr&inverted translaterP)pointrs r test_affine_inverted_invalidatedrs #JEAq{{1::<<+A+A%+H+HIIJJJKKS$$&&&q{{1::<<+A+A%+H+HIIJJJJJrcttjgd}tjtjt jdd}||}|| dd}t|\}}t||j dddS)N))皙?)皙?r)rr_)rr_rclip)rrrKrKF)rsimplifyr) r _create_closedrBlendedGenericTransformrr LogTransformr$ iter_segments get_affineziprcodes)r#transtpathresulttpointstcodess rtest_clipping_of_logrs  MMM N ND  / 22v > > @ @E  + +D 1 1E  !1!1!3!3&6*/!11F6lOGVFDJssO,,,,,rc:eZdZdZdZdZdZfdZdZdZ xZ S)NonAffineForTestz A class which looks like a non affine transform, but does whatever the given transform does (even if it is affine). This is very useful for testing NonAffine behaviour with a simple Affine transform. FrcH||_tj|i|dSrD) real_transrr)rrrrrs rrzNonAffineForTest.__init__s,$$)&)))))rc6|j|SrD)rr&)rrs rr'z%NonAffineForTest.transform_non_affines((000rc6|j|SrD)rr!r"s rr$z*NonAffineForTest.transform_path_non_affines--d333r) r(r)r*r+r.r-r,rr'r$r/r0s@rrrsq IKJ*****1114444444rrc2eZdZdZdZdZdZdZdZdS)TestBasicTransformc\tjdtjdz |_tjddd|_tjddd|_ ttjddd |_ ttjddd |_ ttjddd |_ |j|j |jzz|j z|_|j|j z|jz|j z|_|j |jz|j z|_dS) Nta1)shorthand_namerta2rrta3r`tn1tn2tn3)rrrotater7pirrrrrrrrrstack1stack2 stack2_subsetrs r setup_methodzTestBasicTransform.setup_methodsm'u===DDRUQYOO'u===GGANN'u===CCAqII#K$8$:$:$D$DQ$J$J38:::#K$8$:$:$D$DQ$J$J38:::#K$8$:$:$D$DQ$J$J38:::x48dh#6748C h)DH4tx? !X048;rcr|jjdksJ|jjdksJ|jjdksJdS)Nr\rc)rdepthrrrs rtest_transform_depthz'TestBasicTransform.test_transform_depthsK{ A%%%%{ A%%%%!'1,,,,,,rc|j|j|j|jzzz|jz}||j|j|jzz|jz|j|jz|jz|j|jz|jg}d|D}t |t |ksJt||D] \}}||ksJdS)Ncg|]\}}|SrIrI).0_rhs r zCTestBasicTransform.test_left_to_right_iteration..!s E E EEArR E E Er)rrrrr_iter_break_from_left_to_rightlenr)rstack3target_transformsr target_stackstacks rtest_left_to_right_iterationz/TestBasicTransform.test_left_to_right_iterations(dh$(TX*=>?48K$"h$(TX*=>$(J"h1TX=!X0!X  F EVBBDD E E E1vv.//////#&'8!#<#< ) ) L%5((((( ) )rcF|j|jz |jksJ|j|jz |jksJ|j|jz |jksJ|j|jz jdksJt jt5|j|jz dddn #1swxYwY|j|j |j zz}|j |j z}||z |jksJ||j z ||j zksJ|j|j z |j|j |j zzksJ|j|j z |j|j z|j zksJ|j |j z|j z |j z|j |j zksJdS)Nr`) rrrrrrpytestraises ValueErrorrrr)raff1aff2s rtest_transform_shortcutsz+TestBasicTransform.test_transform_shortcuts's{T//48;;;;{T//48;;;;!DK/483D3D3F3FFFFF"T[071<<<< ]: & & & & K$+ % % & & & & & & & & & & & & & & &x48dh./x$("d{dh&&&&dh$):):)<)<"<<<<<{TX%TX5H)IIIII{TX%DH)>>>>{**4+=>>>>>%55dkBBBBB%55dkBBBBB{**48dh+>?????{**48dh+>?????;..tx$(/BCCCCCCCrc tjddgddgtjdgddggtj}|j|}|j|}tjddgd d gtjtjgddggtj}tjd d gd dgtjtjgd dggtj}t||t||t|j||t|j |||j |j z }|j }t|||j }t||dS)Nrrr_r`r)dtyperr}g3(@g&@g@g"g8@)r7rnanfloat64rr'r&rtransform_affinerrrrPrr)rrna_ptsall_pts na_expected all_expectedexpected_resultrs rtest_affine_simplificationz-TestBasicTransform.test_affine_simplificationWsAq6B8bfa[2q'B " ,,,11&99+''//hR4+!# 02r( &(j222 "&+666!'<888 "$+">">v"F"F". 0 0 0 "$+"8"8":":"D"DV"L"L". 0 0 0  8dh.::<<''))4466?F333''))4466?F33333rN) r(r)r*rrrrrr rIrrrrsq<<<2--- ))) %%%,DDD4#4#4#4#4#4rrcJeZdZdZdZdZdZdZdZdZ dZ d Z d Z d S) TestTransformPlotInterfacec@tj}|gdgd|jt |jtjtj tj gtj tj ggdS)Nrz333333?皙???rr2) r4r5r6r;rrrr7rinfrr>s rtest_line_extent_axes_coordsz7TestTransformPlotInterface.test_line_extent_axes_coords~s XZZ BLIII2:00228bfbf%5')vgw%7%9:: ; ; ; ; ;rctj}|gdgd|jt |jtjddgddggdS)Nr rr2rzrrr) r4r5r6r9rrrr7rrs rtest_line_extent_data_coordsz7TestTransformPlotInterface.test_line_extent_data_coordssz XZZ BLIII2:00228cC[3+$>?? A A A A ArcJtj}tj|j|j}|gdgd|t|j tj tj dgtj dggdS)Nr #r2ggA@) r4r5rblended_transform_factoryr;r9r6rrrr7rrrr>rs r!test_line_extent_compound_coords1ztj}tjd|jz}|gdgd|t|j tj ddgddggdS) Nrr rr2rIru@) r4r5rrrr9r6rrrr7rrs r)test_line_extent_predata_transform_coordszDTestTransformPlotInterface.test_line_extent_predata_transform_coordss XZZ$&&,,R002<? ???2:00228b$Z#t$=>> @ @ @ @ @rctj}tj|jtjd|jz}|gdgd|t|j tj tjdgtj dggdS)Nrr rr2r!r")r4r5rrr;rrr9r6rrrr7rrrs r!test_line_extent_compound_coords2zoffsetexpected_data_lims rtest_line_extents_affinez3TestTransformPlotInterface.test_line_extents_affines XZZ%''11"b99 2&2<*?@@@@Hr2hb %:;;b@!"*"7"7"9"9;LMMMMMrctj}tjdd}t tjdd}tjtjd||z|j ztj ddgddggdz}t|j |dS)Nrr2r'r(r_)r4r5rrrrr6r7r8r9rrrr)rr>r+ na_offsetr,s rtest_line_extents_non_affinez7TestTransformPlotInterface.test_line_extents_non_affines XZZ%''11"b99$[%9%;%;%E%Eb"%M%MNN  2&9*r+r/pthpatchr,s rtest_pathc_extents_non_affinez8TestTransformPlotInterface.test_pathc_extents_non_affines XZZ%''11"b99$[%9%;%;%E%Eb"%M%MNN QFQGb"XAw788"3-3i-?",-NPPP UHr2hs %<==B!"*"7"7"9"9;LMMMMMrctj}tjdd}t ddgddgddgddgg}t j|||jz}| |tj ddgddggdz}t|j |dS)Nrrr2r'r2)r4r5rrrr r3r4r9r5r7rrrr)rr>r+r6r7r,s rtest_pathc_extents_affinez4TestTransformPlotInterface.test_pathc_extents_affines XZZ%''11"b99QFQGb"XAw788"3&2<2GHHH UHr2hs %<==B!"*"7"7"9"9;LMMMMMrcjtjd}tjdd}tjt jd||jzt j ddgddggddgz}t|j |dS)Nrrrrr2r'r(r)r*s r*test_line_extents_for_non_affine_transDatazETestTransformPlotInterface.test_line_extents_for_non_affine_transDatas X ) ) )%''11!R88 2&2<*?@@@@Hr2hb %:;;q"gE!"*"7"7"9"9;LMMMMMrN) r(r)r*rrrr#r%r-r0r8r:r<rIrrr r }s;;;AAA 7 7 7@@@HHHNNNNNN N N NNNN N N N N Nrr c:t|j|jdSrD)rbounds)bbox1bbox2s rassert_bbox_eqrAsu|U\22222rctjddddd}|}t |j|jdS)Nr'r)minpos)rr from_extentsfrozenrrC)bboxrEs rtest_bbox_frozen_copies_minposrGsJ   ( (c3C ( H HD [[]]Fv}dk22222rc tjj}tjj}|dddd}|dddd}|dddd}|dddd}|dddd}t ||||t ||||ddddt |||||||Jt ||||dddddS)Nrr`rr{?g@r)rr rD intersectionrA) bbox_from_extinterrrr3r4r5s rtest_bbox_intersectionrPs'$1M   )E q!Q " "B sCc * *B sAtT * *B sCC ( (B q!Q " "B55R=="%%%55R==--S!Q"?"?@@@55R=="%%% 5R== 55R==--1a";";<<<<??@@@ #a&&64. ) )F ""1q!}}!!!!! C &C..64. 1 1F 551F71a==#..//14444455rc`ttjdjdksJdS)NrraCompositeGenericTransform( CompositeGenericTransform( CompositeGenericTransform( TransformWrapper( BlendedAffine2D( IdentityTransform(), IdentityTransform())), CompositeAffine2D( Affine2D().scale(1.0), Affine2D().scale(1.0))), PolarTransform( PolarAxes(0.125,0.1;0.775x0.8), use_rmin=True, _apply_theta_transforms=False)), CompositeGenericTransform( CompositeGenericTransform( PolarAffine( TransformWrapper( BlendedAffine2D( IdentityTransform(), IdentityTransform())), LockableBbox( Bbox(x0=0.0, y0=0.0, x1=6.283185307179586, y1=1.0), [[-- --] [-- --]])), BboxTransformFrom( _WedgeBbox( (0.5, 0.5), TransformedBbox( Bbox(x0=0.0, y0=0.0, x1=6.283185307179586, y1=1.0), CompositeAffine2D( Affine2D().scale(1.0), Affine2D().scale(1.0))), LockableBbox( Bbox(x0=0.0, y0=0.0, x1=6.283185307179586, y1=1.0), [[-- --] [-- --]])))), BboxTransformTo( TransformedBbox( Bbox(x0=0.125, y0=0.09999999999999998, x1=0.9, y1=0.9), BboxTransformTo( TransformedBbox( Bbox(x0=0.0, y0=0.0, x1=8.0, y1=6.0), Affine2D().scale(80.0))))))))rTr4subplotr9rIrrtest_str_transformr`sB s{g...8 9 9,>8, 8, 8, 8, 8, 8, 8rcptj}|d}|jdksJdS)Nr`r`)r)rrrr)rrs rtest_transform_single_pointrc8s6A 6""A 7d??????rctj\}}|d|jddS)Nrrb)r4subplots set_yscaler9r&)rr>s rtest_log_transformrg>sAlnnGCMM%L6"""""rctjddgddgg}tjddgdtjgg}||rJdS)Nrr`)rr r7roverlaps)arYs rtest_nan_overlaprkFsZ1a&1a&)**A1a&1bf+.//Azz!}}rc,tj}tjgd}tjddgddgddgg}|||}t ||t jt5|||ddddfdddn #1swxYwYt jt5|||ddddfddddS#1swxYwYdS)N)r_-rrr`r) rrr7rtransform_anglesrrrr)ranglesr new_angless rtest_transform_anglesrqLsA Xlll # #F X1v1v1v. / /F##FF33Jfj111 z " "55 66!A#qs(#3444555555555555555 z " "33 66!A#qqq&>222333333333333333333s$#B55B9<B9#D  D D ctjddg}dtjfddg}|D]2}tjtj|}t ||3dS)NgMbPgMbP?rrr)rg)r7rrr nonsingularr)zero_expansioncasesrouts rtest_nonsingularrx^sjXvuo..N[&- 0E00h{.5663////00rctj}tjt5|ddddn #1swxYwYtjt5|dgggdddn #1swxYwYtjt 5|gdddn #1swxYwYtjt 5|dgdddn #1swxYwYtjt 5|dggdddn #1swxYwYtjt 5|gdgddddS#1swxYwYdS)Nr`)r`rrc)rrrrrr& RuntimeErrorrs rtest_invalid_argumentsr|gsA z " " A z " " qcUG | $ $ B | $ $ QC | $ $ aSE | $ $!! YYYK   !!!!!!!!!!!!!!!!!!skAAA3BBB<CC"%C"D%%D),D) E--E14E1F77F;>F;c2gd}t|d}tj}tj||}t |j|dtjdz }| tj dz t |jd||fdd|zf| |fgd dgdz|_ t |jd||fdd|zf| |fgd dS) N)rs)r`rrb)rr`T)closedr`rr\rsrgV瞯<)atol) r rrTransformedPathrget_fully_transformed_pathverticesr7sqrtrrr)rr#r trans_pathrs rtest_transformed_pathr|s4 - - -F t $ $ $D  " "E,T599JJ99;;DfMMM RWQZZB LLJ99;;Db"X1r6{bS"I>    (Q,DKJ99;;Db"X1r6{bS"I>      rctj}tjdddd|}tj|}|j}|dt|j|dz| dt|j|dS)Nrsr`rmr2rr) rrr3WedgeTransformedPatchPathrrrr set_radius)rr7tpatchrs rtest_transformed_patch_pathrs  " "E N61b# ? ? ?E  -e 4 4F  . . 0 0 9F KKNNNF5577@&1*MMM SF5577@&IIIIIrlocked_elementx0y0x1y1cgd}||tj}tj|fi|di}t ||dksJt |d|zdksJ|D]&}t ||t ||ksJ'||dzt ||dksJt |d|zdksJ|D]&}t ||t ||ksJ't|d|zdt |d|zJtj ||ksJt|d|zdt ||dksJt |d|zdksJ|D]&}t ||t ||ksJ'dS)Nrrlocked_rrc) removerr unit LockableBboxrW set_pointsrsetattrr7all)rother_elementsoriglockedelems rtest_lockable_bboxrsU---N.)))   " "D  %d B B~q.A B BF 6> * *a / / / / 69~5 6 6! ; ; ; ;<<vt$$d(;(;;;;;; OODOO%%*+++ 6> * *a / / / / 69~5 6 6! ; ; ; ;<<vt$$d(;(;;;;;; FI.555 69~5 6 6 > > > 6$//##v'8'8':':: ; ;;;; FI.222 6> * *a / / / / 69~5 6 6! ; ; ; ;<<vt$$d(;(;;;;;;< > >&& e $$%%%&&&&&&&&&&&&&&&&&&s(A66A:=A:png) extensionsc :tjdtjd}tjddd}t ||gddgD]\}}|}|||d||tj |d z d z tj d tj zz |j |d dS) Nrr)sizerraTF)rdensityrlinear)r7rrnormalrgrrehistr6exprrrr:rf)fig_testfig_refsamplesrqr log_stater>s rtest_scale_swappingrsINN8iB''G B2Ax1D%=AA  Y \\^^ Y555 26AqD'A+&&RU););;<<<  h   rc,tjtd5tjdddddn #1swxYwYtjtd5tjddddddS#1swxYwYdS)NzZ'fontsize' is not a valid value for units; supported values are 'dots', 'points', 'inches'rfontsize)unitsz3For units of inches or points a fig kwarg is neededinches)rrrr offset_copyrIrrtest_offset_copy_errorsrs, zP Q Q Q88 J7777888888888888888 zR T T T66H5555666666666666666666s!?AA%B  B B c(ttjtd}|ddsJ|ddsJ|ddrJttjtdd}|dd sJ|dd rJ|d d rJdS) Nrrgٿg333333?rg?r~r{rz)r r rr rotate_degcontainsrfully_contains)bbs rtest_transformedbbox_containsr s hjj&;&;B&?&? @ @B ;;r2   ;;sC    {{2r""""" hjj&:&:3&C&C D DB ;;tS ! !!!!  s+++++  R(((((((r)@rnumpyr7 numpy.testingrrrrr matplotlibrmatplotlib.pyplotpyplotr4matplotlib.patchespatchesr3matplotlib.transforms transformsrrr r matplotlib.pathr matplotlib.testing.decoratorsr r r?rRrurrrrrrrr3rrr rArGrPr]r`rcrgrkrqrxr|rrmark parametrizerrrrrrrrIrrrs JJJJJJJJJJJJ %%%%%%++++++AAAAAAAAAA OOOOOOOO###LCCC$'(d' I I IF;;; ; ; ; ; ; ;&"*"*"*JKKK - - -44444{,444,B4B4B4B4B4B4B4B4JZNZNZNZNZNZNZNZNz333333 ===, 5 5 5/8/8/8d ### 333$000!!!*   ,JJJ")+C+C+CDD<<ED