yfQg-VT5@rd )g!@gS@g^z@r r3 @)r# r8 float32 histogram2dr r& )r xy counts_histxedgesyedgess r test_f32_roundingz TestHistogram.test_f32_roundingx sy H888
K K K H<<N>N9N&NOOO 6!QIIIB!"bh1v&6&6bh1v>N>N9N&NOOO $#####///**(GGAJJ
GGAJJ?@@ 6 7CCCB!"wwqzz771::&>??? 6!QIIIB!"wwqzz771::&>?????r c t j g d }t j |ddgd t |g d d S )Nr r" r )rR r4 )r# r8 r r
)r r s r test_no_side_effectsz"TestHistogram.test_no_side_effects sL ///**
VC937777!&///:::::r c t g ddg \ }}t |t j dg t |t j ddg d S )Nr r/ r3 r r r# r8 r r* r+ s r
test_emptyzTestHistogram.test_empty sZ Aq6+++11bhsmm,,,1bh1v . ./////r c t j ddd }t |d t t t |d d S )NrD r5 r numr g333333@)r# r' r r TypeErrorr valss r test_error_binnum_typez$TestHistogram.test_error_binnum_type sB {3---$iD#66666r c t j ddd }t |ddg t t t |t j dg t t t |dt j g d S )NrD r5 r r rG ?rQ )r# r' r r r7 nanrJ r s r test_finite_rangez TestHistogram.test_finite_range so {3---$tDk****j)T"& GGGGj)T$rv GGGGGGr c t j ddd }t t d 5 t j |ddg d d d d S # 1 swxY w Y d S ) NrD r5 r r zmax must be larger thanrC g{Gz?rQ )r# r' r r7 r r s r test_invalid_rangez TestHistogram.test_invalid_range s {3---
-F
G
G 2 2 Lc4[ 1 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2s AAAc . t j g d }t j |dd \ }}|dk }|d d | }|dd | }t ||| D ],\ }}} t ||k t || k -d S )N)iQ i i i& i i i ih )r0 i r4 rR r r[ r/ )r# r8 r zipr )
r arrr9 r: mask
left_edges right_edgesrp leftrights
r test_bin_edge_casesz!TestHistogram.test_bin_edge_cases s h===>>l3TCCC eax3B3Z%
ABBio !#z; ? ? NAtU AI AI r c t j g d }t j |dd \ }}t |d d d S )N rD rD rD r5 r6 rm rm g @g @ r r r[ r/ )r# r8 r r )r r r9 r: s r test_last_bin_inclusive_rangez+TestHistogram.test_last_bin_inclusive_range sL hCCCDDl3RyAAA eT"Xq!!!!!r c t j ddd }t j ddgddgg }t t d 5 t j || d d d d S # 1 swxY w Y d S )
NrD r5 r r r rE g333333?z
must be 1dr3 )r# r' r8 r r7 r )r r r4 s r test_bin_array_dimsz!TestHistogram.test_bin_array_dims s {3---x!SC:.//
\
:
: * * LD ) ) ) ) * * * * * * * * * * * * * * * * * *s A++A/2A/c t j dg }t j g dd }t t 5 t j || \ }}d d d d S # 1 swxY w Y d S )Nr0 )r/ r1 r/ uint64rl r3 )r# r8 r r7 r )r r r4 r9 r: s r test_unsigned_monotonicity_checkz.TestHistogram.test_unsigned_monotonicity_check s hsmmx 222
:
&
& 7 7 ,s666KD% 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7s A))A-0A-c t t t d t d D t j gz t t t d t d D t j gz t j d t d D dgz t j d t d D dgz d S )Nc 6 g | ]}t j d S g?r# r8 .0is r
z9TestHistogram.test_object_array_of_0d.. 999!
999r r" c 6 g | ]}t j d S r r r s r r z9TestHistogram.test_object_array_of_0d.. r r c 6 g | ]}t j d S rE r r s r r z9TestHistogram.test_object_array_of_0d.. 777bhsmm777r g ?c 6 g | ]}t j d S r r r s r r z9TestHistogram.test_object_array_of_0d.. r r rE )r r7 r rR r# rJ r s r test_object_array_of_0dz%TestHistogram.test_object_array_of_0d s j 99uRyy999bfWIE G G Gj 99uRyy999RVHD F F F 77U2YY777;K:LLMMM
77U2YY7772$>?????r c R t j ddt j g }t j t j t j g }t }| t
|5 t
t t |d t
t t |d t |dd \ }}t |
d t |dd \ }}t |
d t |ddg \ }}t |
d t |ddg \ }}t |
d d d d d S # 1 swxY w Y d S )Nr r/ autor3 r r/ r r0 ) r# r8 r r filterry r r7 r r r& )r one_nanall_nanr} r; r+ s r test_some_nan_valuesz"TestHistogram.test_some_nan_values s (Aq"&>**(BFBF+,, !!
>"""
% % *iv F F F F *iv F F F F W6@@@DAq ! $ $ $ W6@@@DAq ! $ $ $ WAq6222DAq ! $ $ $ WAq6222DAq ! $ $ $ % % % % % % % % % % % % % % % % % %s 6DF F #F c t j dd }t j g d }t j g d }||z }||z }t j d }t || \ }}t | | | | \ } }
t || \ } } t
|| t
| | t
||z
t | t
|
t | t
|j |j t
|
j | d S )Nz
2000-01-01D) r r r/ r/ r0 r1 r r" )r r0 r ztimedelta64[D]r3 )r#
datetime64r8 rd r astyper rh )
r beginoffsetsr4 dates date_binstdd_countd_edget_countt_edgei_counti_edges
r
test_datetimezTestHistogram.test_datetime- sB
lC00(88899x
&&DL
X&
'
'
$E :::#GNN2$6$6T[[__MMM#G$777Wg&&&Wg&&&fun,,S116:::V]]3''000V\5;///V\2&&&&&r c dt j | j z dz
}t j d|z dz d|z dz
g| }t |d \ }}t |d|z dz dd|z dz
g t |ddg d S )NrY r/ r0 r2 rl r3 r )r# rd itemsizer8 r r )r rd exponentr r9 r< s r do_signed_overflow_boundsz'TestHistogram.do_signed_overflow_boundsF s rx//!3hH q(!X+/:%HHHCa(((aQ!X+ )1akAo>???TAq6"""""r c < | t j | t j | t j | t j | t j d S r )r r# byteshortintcint_longlongr s r test_signed_overflow_boundsz)TestHistogram.test_signed_overflow_boundsM sv &&rw/// &&rx000 &&rw/// &&rw/// &&r{33333r c Z t j | j }t j dg| }t j d|z dg| }| | d dk rd S t j |d| \ }}t
|dg t
|j | d S )Nr5 r6 r r/ r r# finfoepsr8 r r r rd r float_small float_larger r rR countx_locs r do_precision_lower_boundz&TestHistogram.do_precision_lower_boundT s h{##'huk**#)S);77 << $ $Q '1 , , F |Cau=== uUQC U[+.....r c Z t j | j }t j dg| }t j dd|z
g| }| | d dk rd S t j |d| \ }}t
|dg t
|j | d S )Nr5 rD r[ r/ r r r s r do_precision_upper_boundz&TestHistogram.do_precision_upper_bounde s h{##'huk**#sSy);77 << $ $R (A - - F |Cau=== uUQC U[+.....r c ^ | || | || d S r )r r )r r r s r do_precisionzTestHistogram.do_precisionv s4 %%k;??? %%k;?????r c | t j t j | t j t j | t j t j | t j t j | t j t j | t j t j d S r )r r# halfsingledouble
longdoubler s r test_precisionz TestHistogram.test_precisionz s "'29--- "'29--- "'2=111 ")RY/// ")R]333 ")R]33333r c t g dddg \ }}t g dddg }t || t j g d }t |dd \ }}t |dd }t || t |dd \ }}t |dd }t || d S )
Nr. r/ r0 r r r r r r )r r r r# r8 )r r9 r< r: r s r test_histogram_bin_edgesz&TestHistogram.test_histogram_bin_edges s LLL1a&11a#LLL1a&995!$$$hCCCDDCb :::a#Cb BBB5!$$$CfF;;;a#CfFCCC5!$$$$$r g _B)
free_bytesc t j ddg }d}d}t j d }t j ||||f }t t | t d d S )Ni r1 i i> )sampler4 )r/ r0 )r# r rI r r type)r r xbinsybinszbinsr9 s r test_big_arraysz TestHistogram.test_big_arrays sm 9a.)) % ~V5%2GHHHT$ZZf .....r N)'__name__
__module__ __qualname__r r r, r= rN r] r` rj ru r r r r r r r r r r r r r r r r r r r r r r r pytestmarkslowr r r r r r
s.
" " "/ / / ' ' '@! ! !0G G G
5
5
5 , , ,1 1 1"3 3 3: @ @ @<; ; ;0 0 0
7 7 7 H H H2 2 2 " " "
* * *7 7 7 @ @ @% % %2' ' '2# # #4 4 4/ / /"/ / /"@ @ @4 4 4 % % % _%%% [/ / &%/ / /r r c e Zd ZdZd Zd Zd Zd Zd Zd Z d Z
d Z d
Z e
j d g d d
Zd ZdS )TestHistogramOptimBinNumsz]
Provide test coverage when using provided estimators for optimal number of
bins
c g d}|D ]]}t g | \ }}t |t j dg t |t j ddg ^d S )Nfdscottricesturgesdoanesqrtr stoner3 r r/ r )r estimator_list estimatorr* r+ s r r z$TestHistogramOptimBinNums.test_empty s < < < ( 4 4I Ri000DAq q"(A3-- 0 0 0 q"(Aq6"2"2 3 3 3 3 4 4r c ddddddddddddddd dd
dd d d d
d dd ddd}| D ]\ }}t j dd|dz dz }t j dd|dz dz }t j ||f }| D ]P\ }}t j || \ } }
t t
| |d || QdS )z
Straightforward testing with a mixture of linspace data (for
consistency). All test values have been precomputed and the values
shouldn't change
r2 rY r r0 r r" rP # G r 2 i i r r[ r r/ r1 *For the {0} estimator with datasize of {1}err_msgN)itemsr# r' r r r rz format) r
basic_testtestlenexpectedResultsx1x2rp r numbinsr* r+ s r r, z%TestHistogramOptimBinNums.test_simple se $%A1&'A K K#$B2&("b1 N N#%B2&("b2 O O P P
)3(8(8(:(: P P $G_ S"glQ&677BQGqL1$455BBx((A&5&;&;&=&=
P
P" 7|Ay111 SVVW 744:F9g4N4NP P P P P
P
P Pr c
h ddddddddddddddddddddddddd}| D ]~\ }}t j | }| D ]P\ }}t j || \ }}t t | |d || QdS ) 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/ )r r r r r r r r0 r1 r/ r0 r1 r* r+ N)r- r# rI r r rz r. ) r small_datr0 r1 testdatr expbinsr* r+ s r
test_smallz$TestHistogramOptimBinNums.test_small s !1aA"#Q< < 1aA"#Q< < 1aA"#Q< < = = )2(9(9 P P $G_i((G&5&;&;&=&=
P
P" 7|GY771 SVVW 744:F9g4N4NP P P P P
P P Pr c T g d}|D ] }t t t g d| !dS )zS
Check a Value Error is thrown when an unknown string is passed in
)madfreeman
histogramsIQRr6 N)r r7 r )r
check_listr s r test_incorrect_methodsz0TestHistogramOptimBinNums.test_incorrect_methods sG =<<
# G GI *iI F F F F G Gr c t j d }ddddddddd}| D ]O\ }}t j || \ }}t t | |d | PdS )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
r r/ r z {0} estimator, No Variance testr+ N)r# rg r- r r rz r. )r
novar_datasetnovar_resultdictr r4 r* r+ s r test_novariancez)TestHistogramOptimBinNums.test_novariance s
"#aq%&1q J J #3"8"8":": ? ? Iw<
y99DAq Q 3,,2F9,=,=
?
?
?
?
? ? ?r c t j d }d|dd< d|dd< t |d }t |t j ddd t |d }t |t j ddg t |d
}t |t j ddd dS ) zu
Check when IQR is 0, but variance exists, we return the sturges value
and not the fd value.
i r Nr1 r r r" r r )r# rg r r r' r8 )r lim_var_data
edges_autoedges_fd
edges_sturgess r test_limited_variancez/TestHistogramOptimBinNums.test_limited_variance s
wt}} RaR RSS( v>>
ZQR!8!8999&|T::XrxC 1 1222+L)DD
]BK3$;$;<<<<.nbins_ratio sx )''--C
q
55Ar|Aw//233S a9Q9QRS9T5U5UqAA; r c g | ]Pfd t j ddd t D QS )c ( g | ]} | S r r )r rY r^ r\ s r r zLTestHistogramOptimBinNums.test_scott_vs_stone... s% ooo4{{4&&ooor r" r r2 )startstopr )r# geomspaceroundr rh )r r\ r^ s @r r zATestHistogramOptimBinNums.test_scott_vs_stone.. so % % % poooo2>> SVVWc:::::
; ; ;r r4 )r r r r r r r c ( t j g dt j }t j || \ }}t j | t j | \ }}t
|| t
|| d S )N)r rl r3 )r# r8 int8r r int32r )r r4 r* r9 r: hist32edges32s r test_signed_integer_dataz2TestHistogramOptimBinNums.test_signed_integer_dataD s
H\\\ 1 1 1l14000 e ,qxx'9'9EEE4(((5'*****r c \ g d}|D ]$}t t t g d|g d %dS )z=
Check that weighted data raises a TypeError
)r r r r r r6 rb N)r r r )r r r s r test_simple_weightedz.TestHistogramOptimBinNums.test_simple_weightedN sX DCC' 8 8I )Y #YYY
8
8
8
8
8 8 8r N)r
r r __doc__r r, r: rA rE rL rT ri rr r r parametrizerz r| r r r r r s
4 4 4P P P4P P P(G G G ? ? ? = = =$* * *(F F F ; ; ;> [ V &B &B &B C C+ +C C+8 8 8 8 8r r c h e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd ZdS )TestHistogramddc t j g dg dg dg dg dg dg }t |ddd gd
d gd
d gg \ }}t j g d
g dg dgg d
g dg dgg }t || g dg dg dg}t ||d \ }}t t j ||dz k t |ddd gd
d gd
dggd \ }}t j g dg dg dgg dg dg dgg }t
||dz d d t j |d d D }t |d d d gd
d gd
d gg! \ }}t j d
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 || t j d" }d#|t t d$ t t d$ t t d$ f< t t j d$ t j d$ t j d$ gd$ \ }}t || d S )%N)r rE rF )r rF r )r r rE )rE rE rF )rE rF r )rE r r )r0 r1 r1 r[ r/ r r1 rQ )r r/ r )r r r/ )r/ r r )rt r r0 )r r/ r0 r1 Tr4 r@ g (@)r0 r1 r2 r2 rS )r r/ r r )r r r/ r )r/ r r r g @c 6 g | ]}t j | S r )r# squeeze)r rq s r r z/TestHistogramdd.test_simple..q s ; ; ;qRZ]] ; ; ;r r )r2 r1 r0 rt r0 r )r r r r5 r ) r# r8 r r r allr
splitr listrR rI )r rp Hr: answeredzZs r r, zTestHistogramdd.test_simpleZ s Hnnnooo~~~$nnoooH
I
I q)')1g1v1v%>@ @ @5IIIyyy)))< %IIyyy)))<> ? ?1f%%% jj,,,
5 qr48885qFSL())*** q)')1g1v1v%>'+- - -5 LLL,,, E (LL,,, EG H H!!Vb[!444
< ;BHQ$:$:$: ; ; ;
Ir1g1v1v%>@ @ @5QFQFQF3 FQFQF3 FQFQF3 FQFQF35 6 6 1f%%% HY <>$uQxx..$uQxx..$uQxx..
89 ! billBIaLL I1MM51a r c d}t j dd }|D ]-}t || \ }}t |j |k .d S )N))r r2 rB )rB r2 r )r rB r2 )r2 rB r )rB r r2 )r2 r rB r" r1 r# r$ r% r r shaper r4 rr+ r r: s r
test_shape_3dz TestHistogramdd.test_shape_3d sa INN2q ! ! " "A"1a((HAu AGqL ! ! ! ! " "r c d}t j dd }|D ]-}t || \ }}t |j |k .d S )N))r r2 r rB )r2 r r rB )r rB r2 r )r rB r r2 )r r rB r2 )r2 rB r r )rB r r r2 )r r r2 rB )r r2 rB r )rB r2 r r )rB r r r2 )r2 rB r r )r2 r r rB )r r2 rB r )r r r2 rB )rB r r2 r )rB r r2 r )r2 r rB r )r2 r rB r )r rB r2 r )r r2 r rB )r rB r r2 )rB r2 r r )r r rB r2 r" r2 r r s r
test_shape_4dz TestHistogramdd.test_shape_4d sd H
INN2q ! ! " "A"1a((HAu AGqL ! ! ! ! " "r c t j dd }t | \ }}t |d \ }}t |t j d \ }}t || t |t j d dz d \ }}t || t |t j dt dz \ }}t |d|z d S )Nr r0 Tr? rb r )r# r$ r% r rg r rh )r r) r9 r: n_histw_hists r r z TestHistogramdd.test_weights s INN3 " "!!nn e#At444
#Arws||<<<
64(((#Arws||a/?NNN
66***#ArwsC/@/@1/DEEE
61t8,,,,,r c t j dt }t |d \ }}t |d t j g d d S )N)r" r0 r0 r3 r )r rD rE )r# r rh r r r8 )r rp r9 r: s r test_identical_samplesz&TestHistogramdd.test_identical_samples sQ HWc " "!!!,,, e58RXooo%>%>?????r c t g g gddgddgf \ }}t |t j dgg t j g g g gd \ }}t |t j d d S )Nr r/ r3 rD r0 )r0 r0 r0 )r r
r# r8 r r s r r zTestHistogramdd.test_empty s B8Aq6Aq6*:;;;1 Q2$ 0 0111~r2rl3331 Q 3 344444r c
t j d dd }t t t j |g d t t t j |g d t t t j |dddg dg t
t j |dddg d g d S )
NrY r0 r2 )r[ r0 r2 r r3 )r/ gGz?r/ r/ r/ )r/ r0 r1 r. )r# rI reshaper r7 r r )r rp s r test_bins_errorsz TestHistogramdd.test_bins_errors s
IaLL A & &j".!---HHHHj".!///JJJJ Aq---0H J J J Jq1a '>???@@@@@r c
x t j d 5 t j d dd }t j ddgddgddgg }t j |dt j ddgg \ }}t || t j |dt j d
dt j g g \ }}t || t j |dt j dt j gg \ }}t || d d d d S # 1 swxY w Y d S ) Nignore)invalidrB r1 r0 r/ r r" r3 r[ )r# errstaterI r r8 r rJ r )r rp expectedr; r< s r test_inf_edgesz TestHistogramdd.test_inf_edges si
[
*
*
* ) ) ! $$Q**Ax!Q!Q!Q 899H>!1w2.>*?@@@DAq Ax ( ( (>!1bhArv.G.G*HIIIDAq Ax ( ( (>!1w26.B*CDDDDAq Ax ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s D D//D36D3c v dg}g dg}t || \ }}t |d dk t |d dk dg}g dg}t || \ }}t |d dk t |d dk dg}g dg}t || \ }}t |d dk t |d dk d g}g dg}t || \ }}t |d dk t |d dk d S )
NgG?)rD rE r5 r3 r rD r/ r5 g8 ?gqh ?)r r )r rp r4 r9 _s r test_rightmost_binedgez&TestHistogramdd.test_rightmost_binedge s\
N ad+++aQ3 Q2
E ad+++aQ3 Q2
N ad+++aQ3 Q3
H ad+++aQ3 Q3 r c 8 t j d }t |ddgddgddgg t t t |ddgddgdt j gg t t t |ddgt j dgddgg d S )N)r r1 rD r5 rG r rE rQ )r# r$ r r r7 rJ r r s r r z!TestHistogramdd.test_finite_range s y ))D#sdD\D#; GHHHHj+t !3Z$ bf~ F H H H Hj+t !3Z"&$ $ E G G G G G Gr c t j g d }t j g d }t j g d }d}t ||f||f \ }}t j dgdgg }t || dS )z: Test that adjacent entries in an edge array can be equal )r r/ r0 )r r0 r0 r/ r3 r6 r5 Nr# r8 r r )r rp rq x_edgesy_edgesr9 r:
hist_expecteds r test_equal_edgesz TestHistogramdd.test_equal_edges s HYYY HYYY (999%%!1a&/ABBB e
D
D"
T=)))))r c " t j g d }|dz }t j g d }|dz }t ||f||f \ }}t |d j |j t |d j |j dS )z< Test that if an edge array is input, its type is preserved )r r" r r" )r r rT r r3 r r/ N)r# r8 r r rd )r rp rq r r r9 r: s r test_edge_dtypez TestHistogramdd.test_edge_dtype s H[[[ ! !
F(>>>**B,!1a&/ABBB eU1X^W]333U1X^W]33333r c d}t j dgt j }t j ddgt j }||z }||z }t ||f||f \ }}t |d d d S )Nl r r[ r/ r3 )r r )r# r8 int64r r )r bigrp r rq r r9 r: s r test_large_integersz#TestHistogramdd.test_large_integers s~ HaS"( # #(B8RX.. !G-!1a&/ABBB eT$Z#####r c t j g d }t j g d }t j ddgddgg }t j dgdgdz z dgdz z dgdz z }t j dgdgdz z dgdz z dgdz z }t ||f||f \ }}t || t ||f||fd \ }}t |d
d S ) N)r r0 rY )r rB rY r1 rP r/ r r3 Tr g ?r )r r r relative_areasrp rq r9 r: s r test_density_non_uniform_2dz+TestHistogramdd.test_density_non_uniform_2d s (999%%(999%%
F
F#
HaSA3q5[A3q5(A3q50 1 1 HaSA3q5[A3q5(A3q50 1 1 "1a&/ABBB eT>*** "1a&/A4PPP eT9%%%%%r c t j d }t j g d }t ||d \ }}t |f|fd \ }}t || t ||d d S )Nr" rA Tr? r )r# rI r8 r r r )r r) r4 r9 r: hist_ddedges_dds r test_density_non_uniform_1dz+TestHistogramdd.test_density_non_uniform_1d! s IbMMx((()) 4666 e 'tgt D D DT7###UHQK(((((r N)r
r r r, r r r r r r r r r r r r r r r r r r r X s #! #! #!J" " " " " " - - -@ @ @
5 5 5 A A A
)
)
) .G G G * * * 4 4 4
$
$
$& & &6) ) ) ) )r r )numpyr# numpy.lib.histogramsr r r
numpy.testingr r r r r
r r r
r r numpy.testing._private.utilsr r r r r r r r r sb L L L L L L L L L L
9 8 8 8 8 8
L/ L/ L/ L/ L/ L/ L/ L/^ y8 y8 y8 y8 y8 y8 y8 y8xP) P) P) P) P) P) P) P) P) P)r