:j<ddlZddlmZddlmZmZmZmZm Z m Z m Z ddlm Z ddlmZmZmZmZmZmZddlmZmZddlmZddlZddlmZddlZddlZdZ edZ!d Z"ed gd d dZ#dZ$dZ%edgd d dZ&dZ'dZ(dZ)edgd dddZ*dZ+edgd d dZ,dZ-dZ.Gd d!Z/d"Z0d#Z1d$Z2d%Z3d&Z4y)'N) AsinhScaleAsinhTransform FuncScale LogitScale LogTransformInvertedLogTransformSymmetricalLogTransform) AsinhLocator AutoLocatorLogFormatterSciNotation NullFormatter NullLocatorScalarFormatter)check_figures_equalimage_comparison)IdentityTransform)assert_allclosecd}t|ddft|ddft|t|dt|dt|d t|t|d t|d t|d dgt|ddft|t|dt|dt|t|d t|d t|d dgtddftddfttdtd ttd td td dgy)Nc|SNxs p/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/matplotlib/tests/test_scale.pyz.test_optional_axis_signature..#sqc|Srrrs rrz.test_optional_axis_signature..#sArc|Srrrs rrz.test_optional_axis_signature..$src|Srrrs rrz.test_optional_axis_signature..$sar functionsclip nonpositiveT) use_overline) linear_widthbasesubsc|Srrrs rrz.test_optional_axis_signature...sarc|Srrrs rrz.test_optional_axis_signature...s1r)axisr!r0)r0r$)r0r%r0r'r0r*)r0r-c|Srrrs rrz.test_optional_axis_signature..7src|Srrrs rrz.test_optional_axis_signature..7sarc|Srrrs rrz.test_optional_axis_signature..8s1rc|Srrrs rrz.test_optional_axis_signature..8sr)rrrr1s rtest_optional_axis_signaturer8s Dd[+./ d{K89ttVt(t$'tt!$t!t1a&! 4K#=>DDf-Dt,DDq)Dq!D1v& {K() k23L6"D!LAAQFrcR|jddd}|jd|jd|j}|j }|jddd}|j |||j ddg|d|j |ddgdy)Nzlogsymlog)yscalexscaleg8@)xlimylimb) add_subplotaxvlineaxhlineget_xlimget_ylimsetplot)fig_testfig_refax_testr?r@ax_refs rtest_log_scalesrMBs""3uX"FG OOD OOD    D    D  U8 DF JJDtJ$ KKt dC( KKtTlC(rc*tddd}|j}tjddd}|j |j |}t ||t |t |usJtj|d<|j |j |}t ||t |t |usJtjj|}|j |j |}t ||t |t |usJtjj|d<|j |j |}t ||t |t |usJy) N r&g?r() r invertednparangetransform_non_affinertypenanmaarraymasked)sltsltirouts rtest_symlog_mask_nanr`OsJ ""a +C <<>D $3A  # #C$<$ ?? ?  " "a '' 'rctj\}}|jdt|jtt ddy)Nr;rOr"r#)rirjrreprrrrs rtest_logscale_transform_reprrs8llnGCMM%bf -.rzlogscale_nonpos_values.pngg{Gz?mpl20)rbtolrcc`tjjdtjjt d}t j dd\}\\}}\}}|j|dd|jd|j|dd|jdd tjd dd }tj| }d d|z ztjd|zztj| z}|j|||z ||z|jdtjdd} | dz} | dz} |j| | | |jd|jd|j!gdy)Ni!N,g@@)sizer&)rQrO)rangebinsr;rr#rrerfrQrPr()yerr)rerPgY@)rUrandomseednormalintrirjhistrrVrcos fill_betweenlogspaceerrorbarrk set_yticks) rrsax1ax2ax3ax4xdataydataedatarrurs rtest_logscale_nonpos_valuesrs]IINN8   s3x  (B$'LLA$6!C !*3jsCHHRwRH(NN5HHRwRH(NN5fN- IIaT "E FFE6NE EN266!E'? *2665&> 9EUEEM55=9NN5 BA QA a4DLLADL!NN5NN5NN?#rctj\}}|jtddtdd|j d|j }t jt5|jdddd|j |k(sJt jt5|jdddd|j |k(sJ|jd|j}t jt5|jdddd|j|k(sJt jt5|jdddd|j|k(sJy#1swYxYw#1swYxYw#1swYxYw#1swYDxYw) NrrSr;)leftr)right)bottom)top) rirjr}rrkrErwarns UserWarningset_xlimrrFset_ylim)rsrt original_xlim original_ylims rtest_invalid_log_limsrsellnGCJJuQ{E!QK(MM%KKMM k "   ;;=M )) ) k " "  ;;=M )) )MM%KKMM k " 1  ;;=M )) ) k "   ;;=M )) ) s08F%F2+F>4G %F/2F;>G Gzfunction_scales.pngcd}d}tj\}}tjdd}|j |||j d||f|j ddy)Nc |dzS)Nr&rrs rinversez$test_function_scale..inverses !t rc |dzS)NrRrrs rforwardz$test_function_scale..forwards 3xrrPifunctionr )rirjrUrVrHrkr)rrrsrtrs rtest_function_scalers]llnGC !TAGGAqMMM*'(:M;KK4rcJtj\}}tjd}|j |tjd}|j ||j jdk(sJ|jjdk(sJy)Nr1r;) rirjmscaleLogScalerkrxaxis get_scaleyaxisrs rtest_pass_scalers{llnGC OO &EMM% OO &EMM% 88   5 (( ( 88   5 (( (rctjdd}tj|}t |j t |j k(sJ|j |j usJy)NrrOr3)rrcopydeepcopystr get_transform _transform)scsc2s rtest_scale_deepcopyr s` c +B -- C r! "c#*;*;*=&> >> > == .. .rc*eZdZdZdZdZdZdZy)TestAsinhScalecTd}tjddd}t|}|j}|j}|j |}|j |}t |||j |}t ||tj ||z zy)Ng1@ird)rUrrrTrWrarcsinh) selfa0arrinvinv a_forward a_inverteda_invinvs rtest_transformszTestAsinhScale.test_transformss  KKR % $""$!!#003 11)<  A&..q1"rzz!b&'9"9:rc,tj\}}tdd}|jdk(sJ|jdk(sJ|j dk(sJ|j }t|tsJ|j|jk(sJy)Ng7@r2rO)r&rQ) rirjrr'_base_subsrrr)rrsrtstxs r test_initzTestAsinhScale.test_init$s,,.R Dt 4~~###ww"}}ww&   __ "n---!..000rctj\}}tdd}|jdk(sJ|jdk(sJtddd}|jdk(sJ|jdk(sJy)Nr(r3)r&)r&rS)r0r*r-)rirjrrr)rrsrts3s7s rtest_base_initzTestAsinhScale.test_base_init0so,,.R T *xx1}}xx4 T 7xx1}}xx6!!!rcGdd}|}t|d}|j|t|jdtsJt|jdt sJ|}t|d}|j|t|jdtsJt|jdt sJy)NceZdZdZdZdZy)-TestAsinhScale.test_fmtloc..DummyAxisci|_yrfieldsrs r__init__z6TestAsinhScale.test_fmtloc..DummyAxis.__init__=s   rc<|jjdi|y)Nr)rupdate)rkwargss rrGz1TestAsinhScale.test_fmtloc..DummyAxis.set?s" "",V,rc"||jd<y)Nmajor_formatterr)rfs rset_major_formatterzATestAsinhScale.test_fmtloc..DummyAxis.set_major_formatterAs12 -.rN)__name__ __module__ __qualname__rrGr rrr DummyAxisr<s ! - 3rrrr3 major_locatorrrQ)r#set_default_locators_and_formattersrrr rr )rrax0s0ax5rs r test_fmtloczTestAsinhScale.test_fmtloc;s 3 3k Sq ) ..s3#**_5|DDD#**%67===k Sq ) ..s3#**_5|DDD#**%6713 33rcPtj\}}tjt5t dddddtjt5t dddddt d}t dd}y#1swYRxYw#1swY/xYw)Nrr2rr1g@)rirjrr ValueErrorr)rrsrtrs1s rtest_bad_scalezTestAsinhScale.test_bad_scaleQs},,.R ]]: & 2 Dq 1 2 ]]: & 3 Dr 2 3 T $ T 4  2 2 3 3sB BBB%N)r r r rrrrrrrrrrs ; 1 "3,5rrcGddtGfddtj} tj|t j \}}|j dt|jjsJ tjd=tjd=y#tjd=tjd=wxYw)z[ Test that one can register and use custom scales that don't take an *axis* param. c eZdZy)7test_custom_scale_without_axis..CustomTransformNr r r rrrCustomTransformr` rrc(eZdZdZfdZdZdZy)3test_custom_scale_without_axis..CustomScalecustomc|_yrr)rrs rrz.CustomScale.__init__g -/DOrc|jSrr#rs rrzAtest_custom_scale_without_axis..CustomScale.get_transformj ?? "rc|jt|jt|j t |j tyrset_major_locatorr r rset_minor_locatorrset_minor_formatterr rr0s rrzWtest_custom_scale_without_axis..CustomScale.set_default_locators_and_formattersmC  " ";= 1  $ $_%6 7  " ";= 1  $ $]_ 5rNr r r namerrrrsr CustomScaler c 0 # 6rr1r!N) rr ScaleBaseregister_scalerirjrkrrr_scale_mapping_scale_has_axis_parameterr1rsrtrs @rtest_custom_scale_without_axisr8\s + 6f&&6 7k*,,.R h"((002ODDD  ! !( +  , ,X 6  ! !( +  , ,X 6s A#B44$CcGddtGfddtj} tjt d5tj |dddtj\}}|jdt|jjsJ tjd=tjd=y#1swY{xYw#tjd=tjd=wxYw) z Test that one can still register and use custom scales with an *axis* parameter, but that registering issues a pending-deprecation warning. c eZdZy)4test_custom_scale_with_axis..CustomTransformNrrrrrr;rrrc(eZdZdZfdZdZdZy)0test_custom_scale_with_axis..CustomScaler!c|_yrr#)rr0rs rrz9test_custom_scale_with_axis..CustomScale.__init__r$rc|jSrr#rs rrz>test_custom_scale_with_axis..CustomScale.get_transformr&rc|jt|jt|j t |j tyrr(r,s rrzTtest_custom_scale_with_axis..CustomScale.set_default_locators_and_formattersr-rNr.r0srr1r=r2rr1z)'axis' parameter .* is pending-deprecated)matchNr!)rrr3rrPendingDeprecationWarningr4rirjrkrrrr5r6r7s @rtest_custom_scale_with_axisrC~s  + 6f&&6 7 \\)BD /  ! !+ . /,,.R h"((002ODDD  ! !( +  , ,X 6 / /  ! !( +  , ,X 6s$C$CAC$C!C$$$Dcddddtjdfdtjdfdddd d tjdfd tjdfd d d dddtjdfdtjdfddddtjdfdtjdfg}|D]J\}}}tj|}|d}|j |}||ur9Jd|d|d|d|y)N)rrT)r$T)rTrF)r;rgT)r;gYnT)r;rFF)r;Fr;)rhrRT)rhrFF)rhrgF)rhgF)rhg?Frh)r<rT)r<rET)r<rFTr<r1zFailed z.val_in_range(z ).Expected z, got )rUinfrYrr5 val_in_range) test_casesr/valexpected scale_clsrresults rtest_val_in_rangerOs&  2665! 2665!     "&&% "&&%   2665! 2665!? JD * c8))$/ 4 $! dV>#/ zx 1 !  rcXtjd}|jddusJ|jddusJ|jtjdusJ|jtj dusJ|jtj dusJy)Nr1rgTgF)rr3rIrUrYrH)rs rtest_val_in_range_base_fallbackrQs d#A >># $ && & >>$ 4 '' ' >>"&& !U ** * >>"&& !U ** * >>266' "e ++ +rc@tjdddtjtjdg}gdgdgdgdgdd}|j D]M\}}t j |d}tjj|j||Ot j d djtjd dgdtjgg}tjj|d d gd d ggy) NrRrGrFg?)TTTFFT)TFFFFT)rr;r<asinhrhr1r;rgTF) rUr[rYrHitemsrr5testingassert_array_equalrI)arrcasesr/rLrr_s rtest_val_in_range_arrayrYs ((CsBFFBFFD9 :C8:88:  E ++-Eh  ! !$ 'T 2 %%annS&98DE    &D 1 > > 3+RVV}-. 0CJJ!!#u e}'EFr)5rmatplotlib.pyplotpyplotrimatplotlib.scalerrrrrrr rrmatplotlib.tickerr r r r rrmatplotlib.testing.decoratorsrrmatplotlib.transformsrnumpyrU numpy.testingrr~rr8rMr`rwrrrrrrrrrrrrr8rCrOrQrYrrrrbs>  "P3) (V ) ) 6%&DP  Q  #&&'TQ /R /- (//0"G=$=$4*0()t7K L  )/E5E5P7D#7L, ^ ,Gr