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&Z4dS)'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 dgdS)Nc|SNxs /Users/tedpeters/Documents/Projects/Claude Projects/Misc Tools/notename-sheets/.venv/lib/python3.11/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{{KK89999ttVt((((t$''''tt!$$$$t!t1a&!!!! 4KK#=>>>>DDf----Dt,,,,DDq))))Dq!!!!D1v&&&& {{KK())) kk23333LLL6""""D!!!!LLLAAQFrc|ddd}|d|d|}|}|ddd}||||ddg|d||ddgddS)Nzlogsymlog)yscalexscaleg8@)xlimylimb) add_subplotaxvlineaxhlineget_xlimget_ylimsetplot)fig_testfig_refax_testr?r@ax_refs rtest_log_scalesrMBs""3uX"FFG OOD OOD     D     D  U8 D DF JJDtJ$$$ KKt dC((( KKtTlC(((((rctddd}|}tjddd}|||}t ||t |t |usJtj|d<|||}t ||t |t |usJtj |}|||}t ||t |t |usJtjj |d<|||}t ||t |t |usJdS) N r&g?r() r invertednparangetransform_non_affinertypenanmaarraymasked)sltsltirouts rtest_symlog_mask_nanr`Os ""a + +C <<>>D $3A  # #C$<$ ? ??? ?  "a ' ' ' ' ' 'rctj\}}|dt|jtt dddS)Nr;rOr"r#)rirjrreprrrrs rtest_logscale_transform_reprrsTlnnGCMM%bf - - -.....rzlogscale_nonpos_values.pngg{Gz?mpl20)rbtolrcctjdtjt d}t jdd\}\\}}\}}||dd|d||dd|dd tj d dd }tj | }d d|z ztj d|zztj | z}| |||z ||z|dtj dd} | dz} | dz} || | | |d|d|gddS)Ni!N,g@@)sizer&)rQrO)rangebinsr;rr#rrerfrQrPr()yerr)rerPgY@)rUrandomseednormalintrirjhistrrVrcos fill_betweenlogspaceerrorbarrk set_yticks) rrsax1ax2ax3ax4xdataydataedatarrurs rtest_logscale_nonpos_valuesrsINN8   s3xx  ( (B$'LA$6$6!C !*3jsCHHRwRH(((NN5HHRwRH(((NN5fN--- IaT " "E FE6NNE EN26!E'?? *265&>> 9EUEEM55=999NN5 BA QA a4DLLADL!!!NN5NN5NN???#####rcBtj\}}|tddtdd|d|}t jt5| ddddn #1swxYwY||ksJt jt5| ddddn #1swxYwY||ksJ| d| }t jt5| ddddn #1swxYwY| |ksJt jt5| ddddn #1swxYwY| |ksJdS) NrrSr;)leftr)right)bottom)top) rirjr}rrkrErwarns UserWarningset_xlimrrFset_ylim)rsrt original_xlim original_ylims rtest_invalid_log_limsrslnnGCJJuQ{{E!QKK(((MM%KKMMM k " "   ;;==M ) ) ) ) k " " "  ;;==M ) ) ) )MM%KKMMM k " " 1  ;;==M ) ) ) ) k " "   ;;==M ) ) ) ) ) )sH B..B25B2,DDD6FF FG::G>G>zfunction_scales.pngcd}d}tj\}}tjdd}||||d||f|dddS)Nc |dzS)Nr&rrs rinversez$test_function_scale..inverses !t rc |dzS)NrRrrs rforwardz$test_function_scale..forwards 3xrrPifunctionr )rirjrUrVrHrkr)rrrsrtrs rtest_function_scalerslnnGC !TAGGAqMMMMM*'(:M;;;KK4rcVtj\}}tjd}||tjd}|||jdksJ|jdksJdS)Nr1r;) rirjmscaleLogScalerkrxaxis get_scaleyaxisrs rtest_pass_scalerslnnGC O & & &EMM% O & & &EMM% 8    5 ( ( ( ( 8    5 ( ( ( ( ( (rctjdd}tj|}t |t |ksJ|j|jusJdS)NrrOr3)rrcopydeepcopystr get_transform _transform)scsc2s rtest_scale_deepcopyr sy c + + +B -  C r!! " "c#*;*;*=*=&>&> > > > > = . . . . . .rc,eZdZdZdZdZdZdZdS)TestAsinhScalecd}tjddd}t|}|}|}||}||}t ||||}t ||tj||z zdS)Ng1@ird)rUrrrTrWrarcsinh) selfa0arrinvinv a_forward a_inverteda_invinvs rtest_transformszTestAsinhScale.test_transformss  KR % % $$""$$!!##0033 11)<<  A&&&..q11"rz!b&'9'9"9:::::rctj\}}tdd}|jdksJ|jdksJ|jdksJ|}t|tsJ|j|jksJdS)Ng7@r2rO)r&rQ) rirjrr'_base_subsrrr)rrsrtstxs r test_initzTestAsinhScale.test_init$s,..R Dt 4 4 4~####w"}}}}w&    __  "n-----!.000000rctj\}}tdd}|jdksJ|jdksJtddd}|jdksJ|jdksJdS)Nr(r3)r&)r&rS)r0r*r-)rirjrrr)rrsrts3s7s rtest_base_initzTestAsinhScale.test_base_init0s,..R T * * *x1}}}}x4 T 7 7 7x1}}}}x6!!!!!!rcGdd}|}t|d}||t|jdtsJt|jdt sJ|}t|d}||t|jdtsJt|jdt sJdS)Nc eZdZdZdZdZdS)-TestAsinhScale.test_fmtloc..DummyAxisci|_dSrfieldsrs r__init__z6TestAsinhScale.test_fmtloc..DummyAxis.__init__=s   rc*|jjdi|dS)Nr)rupdate)rkwargss rrGz1TestAsinhScale.test_fmtloc..DummyAxis.set?s#" ",,V,,,,,rc||jd<dS)Nmajor_formatterr)rfs rset_major_formatterzATestAsinhScale.test_fmtloc..DummyAxis.set_major_formatterAs12 -...rN)__name__ __module__ __qualname__rrGr rrr DummyAxisr<sA ! ! ! - - - 3 3 3 3 3rrrr3 major_locatorrrQ)r#set_default_locators_and_formattersrrr rr )rrax0s0ax5rs r test_fmtloczTestAsinhScale.test_fmtloc;s 3 3 3 3 3 3 3 3ikk Sq ) ) ) ..s333#*_5|DDDDD#*%67=====ikk Sq ) ) ) ..s333#*_5|DDDDD#*%67133 3 33 3 3rc|tj\}}tjt5t dddddn #1swxYwYtjt5t dddddn #1swxYwYt d}t dd}dS)Nrr2rr1g@)rirjrr ValueErrorr)rrsrtrs1s rtest_bad_scalezTestAsinhScale.test_bad_scaleQs',..R ]: & & 2 2 Dq 1 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 ]: & & 3 3 Dr 2 2 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 T $ $ $ T 4 4 4s#AAA2BBBN)r r r rrrrrrrrrrs_ ; ; ; 1 1 1 " " "333,55555rrcGddtGfddtj} tj|t j\}}|dt|j sJ tj d=tj d=dS#tj d=tj d=wxYw)z[ Test that one can register and use custom scales that don't take an *axis* param. ceZdZdS)7test_custom_scale_without_axis..CustomTransformNr r r rrrCustomTransformr` rrc*eZdZdZfdZdZdZdS)3test_custom_scale_without_axis..CustomScalecustomc&|_dSrr)rrs rrz.CustomScale.__init__g-o//DOOOrc|jSrr#rs rrzAtest_custom_scale_without_axis..CustomScale.get_transformj ? "rc|t|t|t |tdSrset_major_locatorr r rset_minor_locatorrset_minor_formatterr rr0s rrzWtest_custom_scale_without_axis..CustomScale.set_default_locators_and_formattersmj  " ";== 1 1 1  $ $_%6%6 7 7 7  " ";== 1 1 1  $ $]__ 5 5 5 5 5rNr r r namerrrrsr CustomScaler cQ 0 0 0 0 0 # # # 6 6 6 6 6rr1r!N) rr ScaleBaseregister_scalerirjrkrrr_scale_mapping_scale_has_axis_parameterr1rsrtrs @rtest_custom_scale_without_axisr8\s     +   6666666f&666 7k***,..R h"(0022ODDDDDD  !( +  ,X 6 6 6  !( +  ,X 66666s A(B55CcGddtGfddtj} tjt d5tj|dddn #1swxYwYtj\}}| dt|j sJ tj d=tjd=dS#tj d=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. ceZdZdS)4test_custom_scale_with_axis..CustomTransformNrrrrrr;rrrc*eZdZdZfdZdZdZdS)0test_custom_scale_with_axis..CustomScaler!c&|_dSrr#)rr0rs rrz9test_custom_scale_with_axis..CustomScale.__init__r$rc|jSrr#rs rrz>test_custom_scale_with_axis..CustomScale.get_transformr&rc|t|t|t |tdSrr(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      +   6666666f&666 7 \)BDDD / /  !+ . . . / / / / / / / / / / / / / / /,..R h"(0022ODDDDDD  !( +  ,X 6 6 6  !( +  ,X 66666s0C( A, C(,A00C(3A04AC((Dcddddtjdfdtjdfdddd d tjdfd tjdfd d d dddtjdfdtjdfddddtjdfdtjdfg}|D]S\}}}tj|}|d}||}||usJd|d|d|d|TdS)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_rangerOsD  265! 265!     "&% "&%   265! 265!? JD *  c8)$/ I4 $$!!! 1d 1 1# 1 1  1 1(. 1 1"!!!   rchtjd}|ddusJ|ddusJ|tjdusJ|tjdusJ|tj dusJdS)Nr1rgTgF)rr3rIrUrYrH)rs rtest_val_in_range_base_fallbackrQs d###A >>#  $ & & & & >>$  4 ' ' ' ' >>"& ! !U * * * * >>"& ! !U * * * * >>26' " "e + + + + + +rcBtjdddtjtjdg}gdgdgdgdgdd}|D]S\}}t j|d}tj| ||Tt jd d tjd dgdtjgg}tj|d d gd d ggdS) NrRrGrFg?)TTTFFT)TFFFFT)rr;r<asinhrhr1r;rgTF) rUr[rYrHitemsrr5testingassert_array_equalrI)arrcasesr/rLrr_s rtest_val_in_range_arrayrYs* (CsBFBFD9 : :C888:::888888:::   E ++--EEh  !$ 'T 2 2 2 %%annS&9&98DDDD   &D 1 1 1 > > 3+RV}-.. 0 0CJ!!#u e}'EFFFFFr)5rmatplotlib.pyplotpyplotrimatplotlib.scalerrrrrrr rrmatplotlib.tickerr r r r rrmatplotlib.testing.decoratorsrrmatplotlib.transformsrnumpyrU numpy.testingrr~rr8rMr`rwrrrrrrrrrrrrr8rCrOrQrYrrrrbs[  "!!!!!POOOOOOO333333)))))) (((V ) ) )   6%&DPPP   QP  ###&&'TQQQ / /RQ /--- ( ( (////0"G===$$==$4***0()t7KKK  LK  )))///E5E5E5E5E5E5E5E5P777D#7#7#7L, , , ^ , , ,GGGGGr