qKgddlZddlmZmZmZddlmZmZmZm Z m Z m Z m Z m Z mZmZddlmZddlZGddZGddZGd d Zy) N) histogram histogramddhistogram_bin_edges) assert_ assert_equalassert_array_equalassert_almost_equalassert_array_almost_equal assert_raisesassert_allcloseassert_array_max_ulpassert_raises_regexsuppress_warnings)requires_memoryceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!e"jFjId !d"Z%d#Z&y$)% TestHistogramcyNselfs c/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numpy/lib/tests/test_histograms.py setup_methodzTestHistogram.setup_method cyrrrs rteardown_methodzTestHistogram.teardown_methodrrcd}tjj|}t|\}}t tj |d|ttj ddd\}}t|dy)Ndraxis )nprandomrandrrsumlinspacer)rnvabs r test_simplezTestHistogram.test_simplesb  IINN1 1ARVVAA&*2;;q"c23A1b!rc8tgdddg\}}t|dgt|ddgtttddgdtddgd\}}t |t j dgt|t j ddgy)Nr/r0rbins?@)rrr ValueErrorrr#arrayr )rhistedgeshes r test_one_binzTestHistogram.test_one_bin s q!f5 e4!'51a&)j)aV!<!Qa(1Q! &288RH-.rcd}tjj|}t|d\}}tj|tj |z}t |dtjd}gd}t||d\}}t|dttj|tj |zdt||d\}}t|gd tjd}d dd d tjg}t||d\}}t|gd tjgd ddtjgd\}}t|dd gy)NrTdensityr/r"rr/r1r"皙?Fr.rr1rB)rCrCrC???) r#r$r%rr&diffr arangerrinf) rr(r)r*r+arear4countsdmys r test_densityzTestHistogram.test_density*s.  IINN1 D)1vva"''!*n%D!$ IIbMD$/11b!RVVA N+Q/D%011l+ IIbM1aBFF#D$/11./ll 3RVV,d< Vc1X&rcltjddz}t|ddg\}}t|j dt|ddg\}}t|j dt|ddgd\}}t |tj |zj dd tjddz}t|ddg|d \}}t|tj |zj dt|d ddg| \}}t||ddy)Nr"rEr ranger/TrRr@decimalrRweightsr@r4rRrX)r#rIrrr&r rH)rr*r;r+ws r test_outlierszTestHistogram.test_outliersJs  IIbMB 1a&)1QUUWa 1b'*1QUUWa 1a&$71Q^002ArB IIbMB 1a&!TB1a"''!*n))+Q/1a&!<1Q!B rctjddz}tjddz}ttd5t |ddg|d\}}dddy#1swYyxYw) Nr"rE z same shape asr/rPTrW)r#rIrr7r)rr*r\r;r+s rtest_arr_weights_mismatchz'TestHistogram.test_arr_weights_mismatchbsT IIbMB  IIbMB  _ =Qq!faFDAq> = =s AA(ctjddz}t|\}}ttj|j tj t|d\}}ttj|j tjt|tjdt\}}ttj|j tj t|tjdt\}}ttj|j tjy)Nr"rETr?rX) r#rIrr issubdtypedtypeintegerfloatingonesintfloat)rr*r;r+s r test_typezTestHistogram.test_typeis IIbMB |1 aggrzz23D)1 aggr{{34BGGB$451 aggrzz23BGGB$671 aggr{{34rc tjgdtj}tjgdtj}tj||d\}}}t |j dy)N)g2VEq@gx>fQg-VT5@rd)g!@gS@g^z@rr3@)r#r8float32 histogram2drr&)rxy counts_histxedgesyedgess rtest_f32_roundingzTestHistogram.test_f32_roundingxsV HH8 K HHN9N&NO6!QIB!"bhh1v&6bhh1v>N9N&NO $/*((GAJ GAJ?@6 7CB!"wqz71:&>?6!QIB!"wqz71:&>?rctjgd}tj|ddgdt|gdy)Nrr"r)rRr4)r#r8rr )rrs rtest_no_side_effectsz"TestHistogram.test_no_side_effectss0/* VC937!&/:rctgddg\}}t|tjdgt|tjddgy)Nrr/r3rrr#r8rr*r+s r test_emptyzTestHistogram.test_emptysAAq6+11bhhsm,1bhh1v./rcxtjddd}t|dttt|dy)NrDr5rnumrg333333@)r#r'rr TypeErrorrvalss rtest_error_binnum_typez$TestHistogram.test_error_binnum_types,{{3-$iD#6rctjddd}t|ddgttt|tj dgttt|dtj gy)NrDr5rrrG?rQ)r#r'rr r7nanrJrs rtest_finite_rangezTestHistogram.test_finite_rangesN{{3-$tDk*j)T"&&Gj)T$rvvGrctjddd}ttd5tj|ddgdddy#1swYyxYw) NrDr5rrzmax must be larger thanrCg{Gz?rQ)r#r'rr7rrs rtest_invalid_rangez TestHistogram.test_invalid_ranges>{{3- -F G LLc4[ 1H G Gs A  Actjgd}tj|dd\}}|dkD}|dd|}|dd|}t|||D]"\}}} t ||k\t || k$y)N)iQiii&iiiih )r0ir4rRrr[r/)r#r8rzipr) rarrr9r:mask left_edges right_edgesrpleftrights rtest_bin_edge_casesz!TestHistogram.test_bin_edge_casesshh=>ll3TC eax3BZ% ABio !#z;?NAtU AI  AI @rctjgd}tj|dd\}}t|ddy)N rDrDrDr5r6rmrmg@g@rrr[r/)r#r8rr)rrr9r:s rtest_last_bin_inclusive_rangez+TestHistogram.test_last_bin_inclusive_ranges5hhCDll3RyA eT"Xq!rctjddd}tjddgddgg}ttd5tj || dddy#1swYyxYw) NrDr5rrrrEg333333?z must be 1dr3)r#r'r8rr7r)rrr4s rtest_bin_array_dimsz!TestHistogram.test_bin_array_dimssS{{3-xx!SC:./ \ : LLD ); : :s A%%A.ctjdg}tjgdd}tt5tj||\}}dddy#1swYyxYw)Nr0)r/r1r/uint64rlr3)r#r8r r7r)rrr4r9r:s r test_unsigned_monotonicity_checkz.TestHistogram.test_unsigned_monotonicity_checksKhhsmxx 2 : &,,s6KD%' & &s A##A,c ~ttttdDcgc]}t j dc}tj gzttttdDcgc]}t j dc}tj gzt jtdDcgc]}t j dc}dgzt jtdDcgc]}t j dc}dgzycc}wcc}wcc}wcc}w)Nr"g?rEg ?)r r7rrRr#r8rJ)ris rtest_object_array_of_0dz%TestHistogram.test_object_array_of_0d sj uRy9y! y9bffWIE Gj uRy9y! y9RVVHD F U2Y7YbhhsmY7;K:LLM U2Y7YbhhsmY72$>? :987sD+ /D0 D5D:ctjddtjg}tjtjtjg}t}|j t |5t tt|dt tt|dt|dd\}}t|jdt|dd\}}t|jdt|ddg\}}t|jdt|ddg\}}t|jddddy#1swYyxYw)Nrr/autor3rr/rr0) r#r8rrfilterryr r7rrr&)rone_nanall_nanr}r;r+s rtest_some_nan_valuesz"TestHistogram.test_some_nan_valuess((Aq"&&>*((BFFBFF+, ! >"  *iv F *iv FW6@DAq ! $W6@DAq ! $WAq62DAq ! $WAq62DAq ! $SSs ;CE!!E*ctjdd}tjgd}tjgd}||z}||z}tjd}t ||\}}t |j ||j |\} } t ||\} } t || t | | t ||z j t| t | j t| t |j|jt | j|y)Nz 2000-01-01D) rrr/r/r0r1rr")rr0rztimedelta64[D]r3)r# datetime64r8rdrastyperrh) rbeginoffsetsr4dates date_binstdd_countd_edget_countt_edgei_counti_edges r test_datetimezTestHistogram.test_datetime-s lC0((89xx &DL XX& ' $E :#GNN2$6T[[_M#G$7Wg&Wg&fun,,S16:V]]3'0V\\5;;/V\\2&rcdtj|jzdz }tjd|z dzd|zdz g|}t |d\}}t |d|z dzdd|zdz gt |ddgy)NrYr/r0r2rlr3r)r#rditemsizer8rr)rrdexponentrr9r<s rdo_signed_overflow_boundsz'TestHistogram.do_signed_overflow_boundsFsrxx///!3hhH q(!X+/:%HCa(aQ!X+)1akAo>?TAq6"rc:|jtj|jtj|jtj|jtj |jtj yr)rr#byteshortintcint_longlongrs rtest_signed_overflow_boundsz)TestHistogram.test_signed_overflow_boundsMs\ &&rww/ &&rxx0 &&rww/ &&rww/ &&r{{3rcRtj|j}tjdg|}tjd|zdg|}|j |ddk7rytj |d|\}}t |dgt |j|y)Nr5r6rr/rr#finfoepsr8rrrrdr float_small float_largerrrRcountx_locs rdo_precision_lower_boundz&TestHistogram.do_precision_lower_boundTshh{#''hhuk*#)S);7 << $Q '1 , ||Cau= uUQC U[[+.rcRtj|j}tjdg|}tjdd|z g|}|j |ddk7rytj |d|\}}t |dgt |j|y)Nr5rDr[r/rrrrs rdo_precision_upper_boundz&TestHistogram.do_precision_upper_boundcshh{#''hhuk*#sSy);7 << $R (A - ||Cau= uUQC U[[+.rcL|j|||j||yr)rr)rrrs r do_precisionzTestHistogram.do_precisionss" %%k;? %%k;?rc,|jtjtj|jtjtj|jtjtj |jtjtj|jtjtj |jtjtj yr)rr#halfsingledouble longdoublers rtest_precisionzTestHistogram.test_precisionws "''299- "''299- "''2==1 "))RYY/ "))R]]3 "))R]]3rc<tgdddg\}}tgdddg}t||tjgd}t|dd\}}t|dd}t||t|dd \}}t|dd }t||y) Nr.r/r0rrrrrr)rrrr#r8)rr9r<r:rs rtest_histogram_bin_edgesz&TestHistogram.test_histogram_bin_edgessL1a&1a#L1a&95!$hhCDCb :a#Cb B5!$CfF;a#CfFC5!$rz,Bad memory reports lead to OOM in ci testing)reasonctjddg}d}d}tjd}tj||||f}t t |t dy)Nir1ii>)sampler4)r/r0)r#rrIrrtype)rrxbinsybinszbinsr9s rtest_big_arrayszTestHistogram.test_big_arrayssU9a.) % ~~V5%2GHT$Zf.rctjtjdgddd\}}tjddg}t||y) Ng$xz>f8rlr0)gx0gel rr/r)r#rr8r)rr9r< expected_hists r test_gh_23110zTestHistogram.test_gh_23110sG,,rxx 5A$%%79a!Q( 4/rN)'__name__ __module__ __qualname__rrr,r=rNr]r`rjrurrrrrrrrrrrrrrrrrrrrrrpytestmarkskiprrrrrrr s   "/'@!0G 5,1"3:@<;0 7 H2 " *7 @%2'2#4 // @4 %  [[KL/M/0rrceZdZdZdZdZdZdZdZdZ dZ d Z d Z e jjd gd d ZdZy)TestHistogramOptimBinNumsz] Provide test coverage when using provided estimators for optimal number of bins cgd}|D]S}tg|\}}t|tjdgt|tjddgUy)Nfdscottricesturgesdoanesqrtrstoner3rr/r)restimator_list estimatorr*r+s rrz$TestHistogramOptimBinNums.test_emptysP<(IRi0DAq q"((A3- 0 q"((Aq6"2 3(rc ddddddddddddddd dd dd d d d d dd ddd}|jD]\}}tjdd|dzdz}tjdd|dzdz}tj||f}|jD]E\}}tj||\} } t t | |dj||Gy)z Straightforward testing with a mixture of linspace data (for consistency). All test values have been precomputed and the values shouldn't change r2rYrr0r r" rP#Gr2iirr[rr/r1*For the {0} estimator with datasize of {1}err_msgN)itemsr#r'rrrrzformat) r basic_testtestlenexpectedResultsx1x2rprnumbinsr*r+s rr,z%TestHistogramOptimBinNums.test_simples$%A1&'AK#$B2&("b1N#%B2&("b2O P )3(8(8(: $G_S"glQ&67BQGqL1$45BBx(A&5&;&;&=" 7||Ay11SVW744:F9g4NP'> );rc Jddddddddddddddddddddddddd}|jD]r\}}tj|}|jD]E\}}tj||\}}t t ||dj ||Gty) z Smaller datasets have the potential to cause issues with the data adaptive methods, especially the FD method. All bin numbers have been precalculated. r/)rrrrrrrr0r1r/r0r1r!r"N)r$r#rIrrrzr%) r small_datr'r(testdatrexpbinsr*r+s r test_smallz$TestHistogramOptimBinNums.test_smalls !1aA"#Q< 1aA"#Q< 1aA"#Q< = )2(9 $G_ii(G&5&;&;&=" 7||GY71SVW744:F9g4NP'>):rcJgd}|D]}tttgd|y)zS Check a Value Error is thrown when an unknown string is passed in )madfreeman histogramsIQRr-N)r r7r)r check_listrs rtest_incorrect_methodsz0TestHistogramOptimBinNums.test_incorrect_methodss!= #I *iI F$rc tjd}ddddddddd}|jD]D\}}tj||\}}t t ||dj |Fy)z Check that methods handle no variance in data Primarily for Scott and FD as the SD and IQR are both 0 in this case rr/r z{0} estimator, No Variance testr"N)r#rgr$rrrzr%)r novar_datasetnovar_resultdictrr+r*r+s rtest_novariancez)TestHistogramOptimBinNums.test_novariancesy  "#aq%&1qJ#3"8"8": Iw<< y9DAq Q3,,2F9,= ?#;rcPtjd}d|ddd|ddt|d}t|tjdddt|d }t|tj ddgt|d }t|tjdddy) zu Check when IQR is 0, but variance exists, we return the sturges value and not the fd value. irNr1rrrrr)r#rgrrr'r8)r lim_var_data edges_autoedges_fd edges_sturgess rtest_limited_variancez/TestHistogramOptimBinNums.test_limited_variances wwt}  Ra RS(v> ZQR!89&|T:XrxxC12+L)D ]BKK3$;.nbins_ratiosl))''-C q 5Ar||Aw/23Sa9QRS9T5UqAA; rr"rr2)startstoprrr rE)g333333?g ףp= ?g{Gz?gQ?r0rUN) rRr# geomspaceroundrrhabsmeanr )rrUrSrPllavgs rtest_scott_vs_stonez-TestHistogramOptimBinNums.test_scott_vs_stones  )%#$46<>SVWc: '>);rr4)rrrrrrrc$tjgdtj}tj||\}}tj|j tj |\}}t ||t ||y)N)rrlr3)r#r8int8rrint32r)rr4r*r9r:hist32edges32s rtest_signed_integer_dataz2TestHistogramOptimBinNums.test_signed_integer_dataIsa HH\ 1ll140 e,,qxx'9E4(5'*rc Rgd}|D]}tttgd|gd y)z= Check that weighted data raises a TypeError )rrrrrr-rbN)r rr)rrrs rtest_simple_weightedz.TestHistogramOptimBinNums.test_simple_weightedSs'D'I )Y #Y 8(rN)rrr__doc__rr,r1r8r<rCrKr^rgrr parametrizerorqrrrr r sj 4P4P(G ?=$*(F ;> [[V&BC+C+8rr cfeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZy)TestHistogramddc tjgdgdgdgdgdgdg}t|ddd gd d gd d gg \}}tjgd gdgdggd gdgdgg}t||gdgdgdg}t||d\}}t tj ||dz k(t|ddd gd d gd dggd\}}tjgdgdgdggdgdgdgg}t ||dz dtj|d d Dcgc]}tj|}}t|dddgd d gd dgg \}}tjd d gd d gd d ggd d gd d gd d ggd d gd d gd d ggd d gd d gd d ggg}t||tjd!}d"|ttd#ttd#ttd#f<ttjd#tjd#tjd#gd#\}}t||ycc}w)$N)rrErF)rrFr)rrrE)rErErF)rErFr)rErr)r0r1r1r[r/rr1rQ)rr/r)rrr/)r/rr)rirr0)rr/r0r1Tr4r@g(@)r0r1r2r2rS)rr/rr)rrr/r)r/rrrg@r )r2r1r0rir0r)rrrr5r) r#r8rrrallr splitsqueezerlistrRrI) rrpHr:answeredrqzZs rr,zTestHistogramdd.test_simple_s_ HHno~$oH Iq)')1g1v1v%>@5Iy)<%y)<>?1f%, 5qr485qFSL()*q)')1g1v1v%>'+-5L, E(, EGH!!Vb[!4%'HHQ$: ;$:qRZZ]$: ; Ir1g1v1v%>@5QFQFQF3FQFQF3FQFQF3FQFQF356 1f% HHY <>$uQx.$uQx.$uQx. 89 ! biilBIIaLI1M51a  ?rctgggddgddgf\}}t|tjdggtjggggd\}}t|tjdy)Nrr/r3rDr0)r0r0r0)rr r#r8rrs rrzTestHistogramdd.test_emptysgB8Aq6Aq6*:;1Q2$ 01~~r2rl31Q 34rc ttjdjdd}tttj |gdtttj |gdtttj |dddgdgt tj |dddgd gy) NrYr0r2)r[r0r2rr3)r/gGz?r/r/r/)r/r0r1r.)r#rIreshaper r7rr)rrps rtest_bins_errorsz TestHistogramdd.test_bins_errorss} IIaL A &j"..!-Hj"..!/J Aq-0H Jq1a'>?@rc |tjd5tjdjdd}tjddgddgddgg}tj |dtj ddgg \}}t||tj |dtjd dtj gg \}}t||tj |dtj dtj gg \}}t||dddy#1swYyxYw) Nignore)invalidrBr1r0r/rr"r3r[)r#errstaterIrr8rrJr )rrpexpectedr;r<s rtest_inf_edgeszTestHistogramdd.test_inf_edgess [[ * ! $$Q*Axx!Q!Q!Q 89H>>!1w2.>*?@DAq Ax (>>!1bhhArvv.G*HIDAq Ax (>>!1w266.B*CDDAq Ax (+ * *s DD22D;cdg}gdg}t||\}}t|ddk(t|ddk(dg}gdg}t||\}}t|ddk(t|ddk(dg}gdg}t||\}}t|ddk(t|ddk(d g}gdg}t||\}}t|ddk(t|ddk(y) NgG?)rDrEr5r3rrDr/r5g8?gqh?)rr)rrpr4r9_s rtest_rightmost_binedgez&TestHistogramdd.test_rightmost_binedges Nad+aQ3Q2  Ead+aQ3Q2  Nad+aQ3Q3 Had+aQ3Q3rc &tjjd}t|ddgddgddggttt|ddgddgdtj ggttt|ddgtj dgddggy)N)rr1rDr5rGrrErQ)r#r$rr r7rJrrs rrz!TestHistogramdd.test_finite_rangesyy)D#sdD\D#; GHj+t!3Z$bff~F Hj+t!3Z"&&$$E Grctjgd}tjgd}tjgd}d}t||f||f\}}tjdgdgg}t||y)z: Test that adjacent entries in an edge array can be equal )rr/r0)rr0r0r/r3r6r5Nr#r8rr)rrprqx_edgesy_edgesr9r: hist_expecteds rtest_equal_edgesz TestHistogramdd.test_equal_edgesss HHY  HHY ((9%!1a&/AB e D D"   T=)rc(tjgd}|dz }tjgd}|dz }t||f||f\}}t|dj|jt|dj|jy)z< Test that if an edge array is input, its type is preserved )rr"rr")rrrTrr3rr/N)r#r8rrrd)rrprqrrr9r:s rtest_edge_dtypezTestHistogramdd.test_edge_dtypesu HH[ ! F((>*B,!1a&/AB eU1X^^W]]3U1X^^W]]3rcd}tjdgtj}tjddgtj}||z}||z}t||f||f\}}t |ddy)Nlrr[r/r3)rr)r#r8int64rr)rbigrprrqrr9r:s rtest_large_integersz#TestHistogramdd.test_large_integerssn HHaS"(( #((B8RXX. !G-!1a&/AB eT$Z#rctjgd}tjgd}tjddgddgg}tjdgdgdzzdgdzzdgdzz}tjdgdgdzzdgdzzdgdzz}t||f||f\}}t||t||f||fd \}}t|d y) N)rr0rY)rrBrYr1rPr/rr3Trwg?r)rrrrelative_areasrprqr9r:s rtest_density_non_uniform_2dz+TestHistogramdd.test_density_non_uniform_2d s((9%((9% F F# HHaSA3q5[A3q5(A3q50 1 HHaSA3q5[A3q5(A3q50 1"1a&/AB eT>*"1a&/A4P eT9%rctjd}tjgd}t||d\}}t |f|fd\}}t ||t ||dy)Nr"rATr?r)r#rIr8rrr)rr)r4r9r:hist_ddedges_dds rtest_density_non_uniform_1dz+TestHistogramdd.test_density_non_uniform_1d&s_ IIbMxx()46 e'tgtDT7#UHQK(rN)rrrr,rrrrrrrrrrrrrrrrrruru]sQ#!J" " -@ 5 A ) .G * 4 $&6)rru)numpyr#rrr numpy.testingrrrr r r r r rrnumpy.testing._private.utilsrrrr rurrrrsK== 9 Q0Q0h y8y8xP)P)r