U j@sddlZddlmZmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZddlmZddlZGdddZGdddZGd d d ZdS) 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_memoryc@s&eZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9d:Zd;d<Z d=d>Z!d?d@Z"e#dAdBe$j%j&dCdDZ'dES)F TestHistogramcCsdSNselfrrC/tmp/pip-unpacked-wheel-16f7874t/numpy/lib/tests/test_histograms.py setup_methodszTestHistogram.setup_methodcCsdSrrrrrrteardown_methodszTestHistogram.teardown_methodcCsTd}tj|}t|\}}ttj|dd|ttddd\}}t|ddS)NdrZaxis )nprandomrandrrsumlinspacer)rnvabrrr test_simples   zTestHistogram.test_simplecCstddddgddg\}}t|dgt|ddgtttddgddtddgdd\}}t|tdgt|tddgdS) Nrbins?@)rrr ValueErrorrrarrayr )rhistedgesherrr test_one_bin s zTestHistogram.test_one_binc Cst}|`|tjd}d}tj|}t|dd\}}t|t|}t |dt t |dW5QRXt}|j|tjd}t d}dddd dg}t||dd\}}t|t|}t |dt t |dW5QRXdS) Nz .*normed.*rTnormedr%rr ) rrecordrZVisibleDeprecationWarningrrrrdiffrrlenarange) rsuprecr r!r"r#arear*rrr test_normed*s$    zTestHistogram.test_normedc Cs$d}tj|}t|dd\}}t|t|}t|dtd}dddddg}t||dd\}}t|d t t|t|dt||d d\}}t|dd dd gtd}ddddtj g}t||dd\}}t|d d d d gtjdd dd gddtj gdd\}}t |ddgdS)NrTdensityr%rrr'皙?Fr&r(???) rrrrrr9rr;rrinf) rr r!r"r#r>r*countsZdmyrrr test_densityBs,       zTestHistogram.test_densitycCstdd}t|ddgd\}}t|dt|ddgd\}}t|dt|ddgdd\}}t|t|dd d tdd}t|ddg|dd \}}t|t|dt|d ddg|d \}}t||dddS)NrrErr7ranger%TrLrAdecimalrLweightsrAr*rLrR)rr;rrrrr9)rr"r1r#wrrr test_outliersbszTestHistogram.test_outliersc CsNtdd}tdd}ttdt|ddg|dd\}}W5QRXdS) NrrE z same shape asr%r7TrQ)rr;r r-r)rr"rVr1r#rrrtest_arr_weights_mismatchzs z'TestHistogram.test_arr_weights_mismatchcCstdd}t|\}}tt|jtjt|dd\}}tt|jtjt|tdt d\}}tt|jtjt|tdt d\}}tt|jtjdS)NrrETr@rR) rr;rrZ issubdtypedtypeintegerZfloatingonesintfloat)rr"r1r#rrr test_types zTestHistogram.test_typecCsTtjdddgtjd}tjdddgtjd}tj||dd \}}}t|d dS) Ng2VEq@gx>fQg-VT5@r[g!@gS@g^z@rr)@)rr.Zfloat32Z histogram2drr)rxyZ counts_histZxedgesZyedgesrrrtest_f32_roundingszTestHistogram.test_f32_roundingc Cs|tjdddgtjd}t|\}}tH}|td}tdddg\}}tt|dt ||t ||W5QRXdS)Nr%rrazConverting input from .*TF) rr.Zuint8rrr8RuntimeWarningrr:r)rr"Zint_histZ int_edgesr<r=r/r0rrrtest_bool_conversions  z"TestHistogram.test_bool_conversionc Cstjd}tdd}t|\}}t|dd\}}t||d\}}t||dd\} } t|d|t|| tddd}ttdtdf}t|t d |d \}}t||td d d d gd d dd d gd \}}t |d ddd gtd d d d gd d dd d gdd\}}t|t d ddd gddd tt ddd dddgd d d d d d d d d g dd\}}t |ddddgdS)Nrr6Tr@rZrRrArrrXr*rRr%r&r(r')r*rRrAg$@rbr7rBg?rCg333333?) rrrr]rr r concatenatezerosr;rr.r) rr!rVr"r#nanbwawbZnwaZnwbrrr test_weightss8    "   $  zTestHistogram.test_weightscCsHtdddg}tdddgdtdddg}t|ddd g|d \}}t|tddgdtdd gt|ddd g|d \}}t|tddgdtdd gdd lm}tdddg}t|d|d|d g}t|ddd g|d \}}t||d|d gt|ddd g|d \}}t||d|d gdS)N?@ffffff@r%rUr&y?rr'rirT)Decimalr6)rr.rr rPrt)rvaluesrRrnrortrrrtest_exotic_weightss$&& z!TestHistogram.test_exotic_weightscCs8tdddg}tj|ddgddt|dddgdS)Nrqrrrsrr)rLr*)rr.rr )rrurrrtest_no_side_effectssz"TestHistogram.test_no_side_effectscCs>tgddgd\}}t|tdgt|tddgdS)Nrr%r)rrrr.rr"r#rrr test_emptyszTestHistogram.test_emptycCs,tjdddd}t|dttt|ddS)NrDr+rnumr6g333333@)rrrr TypeErrorrvalsrrrtest_error_binnum_types z$TestHistogram.test_error_binnum_typecCsPtjdddd}t|ddgdttt|tjdgdttt|dtjgddS)NrDr+rr|rG?rK)rrrr r-nanrHrrrrtest_finite_rangeszTestHistogram.test_finite_rangec Cs<tjdddd}ttdtj|ddgdW5QRXdS) NrDr+rr|zmax must be larger thanrCg{Gz?rK)rrr r-rrrrrtest_invalid_ranges z TestHistogram.test_invalid_rangec Cstdddddddg}tj|dd d \}}|d k}|dd |}|d d|}t|||D]"\}}} t||kt|| kq`dS)NiQiii&iiiih )r&ir*rLrrUr%)rr.rzipr) rarrr/r0maskZ left_edgesZ right_edgesrcleftrightrrrtest_bin_edge_casess z!TestHistogram.test_bin_edge_casesc CsBtdddddddddg }tj|ddd \}}t|d d dS) NrDr+r,rb@@r6rrUr%)rr.rr)rrr/r0rrrtest_last_bin_inclusive_range sz+TestHistogram.test_last_bin_inclusive_rangec CsNtjdddd}tddgddgg}ttdtj||d W5QRXdS) NrDr+rr|rrEg333333?z must be 1dr))rrr.r r-r)rrr*rrrtest_bin_array_dimss z!TestHistogram.test_bin_array_dimsc CsJtdg}tjdddgdd}tttj||d\}}W5QRXdS)Nr&r%r'Zuint64rar))rr.r r-r)rrr*r/r0rrr test_unsigned_monotonicity_checks  z.TestHistogram.test_unsigned_monotonicity_checkcCstttddtdDtj gtttddtdDtjgtddtdDdgtddtdDdgdS) NcSsg|]}tdqSg?rr..0irrr $sz9TestHistogram.test_object_array_of_0d..rcSsg|]}tdqSrrrrrrr&scSsg|]}tdqSrErrrrrr)sg ?cSsg|]}tdqSrrrrrrr*srE)r r-rrLrrHrrrrtest_object_array_of_0d!sz%TestHistogram.test_object_array_of_0dc Cstddtjg}ttjtjg}t}|t|ttt|ddttt|ddt|ddd\}}t | dt|ddd\}}t | dt|ddgd\}}t | dt|ddgd\}}t | dW5QRXdS)Nrr%autor)rr%rr&) rr.rrfilterrfr r-rrr)rZone_nanZall_nanr<r1r#rrrtest_some_nan_values,s z"TestHistogram.test_some_nan_valuesc Cstdd}tddddddddd g }tddd d g}||}||}td }t||d \}}t||||d \} } t||d \} } t|| t| | t||t| t| t| t|j|jt| j|dS) Nz 2000-01-01Drr%r&r'r6rztimedelta64[D]r))rZ datetime64r.r[rastyperr^) rbeginoffsetsr*datesZ date_binstdZd_countZd_edgeZt_countZt_edgeZi_countZi_edgerrr test_datetimeEs    zTestHistogram.test_datetimecCs|dt|jd}tjd| dd|dg|d}t|dd\}}t|d| ddd|dgt|ddgdS)NrSr%r&r(rar)r)rr[itemsizer.rr)rr[exponentrr/r2rrrdo_signed_overflow_bounds^s $"z'TestHistogram.do_signed_overflow_boundscCs@|tj|tj|tj|tj|tjdSr)rrbyteshortZintcint_Zlonglongrrrrtest_signed_overflow_boundses     z)TestHistogram.test_signed_overflow_boundscCstt|j}tdg|}td|dg|}||ddkrDdStj|d|d\}}t|dgt|j|dS)Nr+r,rr%rrZfinfoepsr.rrrr[r float_small float_largerrrLcountZx_locrrrdo_precision_lower_boundls  z&TestHistogram.do_precision_lower_boundcCstt|j}tdg|}tdd|g|}||ddkrDdStj|d|d\}}t|dgt|j|dS)Nr+rDrUr%rrrrrrdo_precision_upper_bound}s  z&TestHistogram.do_precision_upper_boundcCs||||||dSr)rr)rrrrrr do_precisions zTestHistogram.do_precisioncCsd|tjtj|tjtj|tjtj|tjtj|tjtj|tjtjdSr)rrZhalfZsingledoubleZ longdoublerrrrtest_precisions zTestHistogram.test_precisionc Cstddddgddg\}}tddddgddg}t||tdddddddd d g }t|d d d \}}t|d d d }t||t|ddd \}}t|ddd }t||dS)Nr%r&r'r(rDr+r,rbrrrrrrr)rrrrr.)rr/r2r0rrrrtest_histogram_bin_edgess  z&TestHistogram.test_histogram_bin_edgesg _B)Z free_bytescCsJtddg}d}d}td}tj||||fd}tt|tddS)Nir'ii>)sampler*)r%r&)rrkr;rrtype)rrZxbinsZybinsZzbinsr/rrrtest_big_arrayss  zTestHistogram.test_big_arraysN)(__name__ __module__ __qualname__rrr$r3r?rJrWrYr`rergrprvrxr{rrrrrrrrrrrrrrrrrrpytestmarkZslowrrrrrr sF      rc @seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ e jddddddddgddZddZd S)!TestHistogramOptimBinNumsz] Provide test coverage when using provided estimators for optimal number of bins cCsXddddddddg}|D]:}tg|d \}}t|td gt|td d gqdS) Nfdscottricesturgesdoanesqrtrstoner)rr%ry)restimator_list estimatorr"r#rrrr{sz$TestHistogramOptimBinNums.test_emptyc Csddddddddddddddd dd dd d d d d dd ddd}|D]~\}}tdd|dd}tdd|dd}t||f}|D]2\}}t||\} } tt| |d||dqqJdS)z Straightforward testing with a mixture of linspace data (for consistency). All test values have been precomputed and the values shouldn't change r(rSrr&rrrrrrrrr r7#Gr2iirwrUr6r%r'*For the {0} estimator with datasize of {1}err_msgN)itemsrrrjrrr:format) r basic_testtestlenexpectedResultsx1x2rcrnumbinsr"r#rrrr$s:z%TestHistogramOptimBinNums.test_simplec Csddddddddddddddddddddddddd}|D]N\}}t|}|D]2\}}t||\}}tt||d||dq^qDdS) 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%)rrrrrrrr&r')r%r&r'rrN)rrr;rrr:r) rZ small_datrrZtestdatrZexpbinsr"r#rrr test_smalls0 z$TestHistogramOptimBinNums.test_smallcCs.ddddg}|D]}tttdddg|qdS) zS Check a Value Error is thrown when an unknown string is passed in ZmadZfreemanZ histogramsZIQRr%r&r'N)r r-r)rZ check_listrrrrtest_incorrect_methodss z0TestHistogramOptimBinNums.test_incorrect_methodsc Cs^td}ddddddddd}|D]0\}}t||\}}tt||d|dq(dS)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%rz{0} estimator, No Variance testrN)rr]rrrr:r)rZ novar_datasetZnovar_resultdictrrr"r#rrrtest_novariances z)TestHistogramOptimBinNums.test_novariancecCstd}d|dd<d|dd<t|d}t|tdddt|d }t|tddgt|d }t|tddddS) zu Check when IQR is 0, but variance exists, we return the sturges value and not the fd value. irNr'rrrrr)rr]rrrr.)rZ lim_var_dataZ edges_autoZedges_fdZ edges_sturgesrrrtest_limited_variances      z/TestHistogramOptimBinNums.test_limited_variancecCshtddd}ttddd|f}dddd d }|D]&\}}t||\}}tt||q.nbins_ratiocs6g|].fddtjddddtDqS)csg|]}|qSrr)rr)rrrrr6szLTestHistogramOptimBinNums.test_scott_vs_stone...rrr()startstopr})rZ geomspaceroundrr^)rr)rrr6szATestHistogramOptimBinNums.test_scott_vs_stone..rrrrEg333333?g ףp= ?g{Gz?gQ?r&rON)rLabsrZmeanr)rZllavgrrrtest_scott_vs_stone-s  z-TestHistogramOptimBinNums.test_scott_vs_stonec Csddddddddddddddd d d d d ddd }|D]\}}td d|dd}tdd|dd}tdd|}t|||f}|D]F\}} tj||dd\} } d|} | d|7} tt| | | dqq>dS)z Straightforward testing with a mixture of linspace data (for consistency). Adding in a 3rd mixture that will then be completely ignored. All test values have been precomputed and the shouldn't change. rSrNr)rrrrrrr rP!ErrwrUr6r&r%rr'ri)irrKzFor the {0} estimatorz with datasize of {0}rN)rrrrrrrr:) rrrrrrZx3rcrrr"r#msgrrrtest_simple_range=s4   z+TestHistogramOptimBinNums.test_simple_ranger*rrrrrrrcCsZtjdddgtjd}tj||d\}}tj|tj|d\}}t||t||dS)Nrrar))rr.Zint8rrZint32r)rr*r"r/r0Zhist32Zedges32rrrtest_signed_integer_data\s  z2TestHistogramOptimBinNums.test_signed_integer_datac Cs:dddddg}|D]"}tttdddg|dddgd qd S) z= Check that weighted data raises a TypeError rrrrrr%r&r'rZN)r r~r)rrrrrrtest_simple_weightedfs z.TestHistogramOptimBinNums.test_simple_weightedN)rrr__doc__r{r$rrrrrrrrrZ parametrizerrrrrrrs"  rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#S)$TestHistogramddc Csttdddgdddgdddgdddgdddgdddgg}t|dddgdd gdd ggd \}}tdddgdddgdddggdddgdddgdddggg}t||d dd gddd d gddd d gg}t||d d\}}tt||dkt|dddgdd gddggd d\}}tddddgddddgddddggddddgddddgddddggg}t||ddddtj|d ddD}t|dd d gdd gdd ggd\}}tddgddgddggddgddgddggddgddgddggddgddgddggg}t||td}d|t t dt t dt t df<tt dt dt dgd\}}t||dS)NrrErFrr)r&r'r'rUr%rr'rKrr&Tr*rAg(@)r&r'r(r(rMg@cSsg|]}t|qSr)rZsqueeze)rrdrrrrsz/TestHistogramdd.test_simple..r)r(r'r&r)r6r6r6r+r6) rr.rrrallr splitrklistrLr;)rrcHr0ZanswerZedzZrrrr$rsN    $    &&zTestHistogramdd.test_simplecCs<d}tjdd}|D] }t||\}}t|j|kqdS)N))r6r(rB)rBr(r6)r6rBr()r(rBr6)rBr6r()r(r6rBrr'rrrrrshaperr*rr#r r0rrr test_shape_3ds zTestHistogramdd.test_shape_3dcCs<d}tjdd}|D] }t||\}}t|j|kqdS)N))rr(r6rB)r(r6rrB)r6rBr(r)rrBr6r()r6rrBr()r(rBrr6)rBr6rr()rr6r(rB)rr(rBr6)rBr(rr6)rBrr6r()r(rBr6r)r(rr6rB)r6r(rBr)r6rr(rB)rBrr(r6)rBr6r(r)r(rrBr6)r(r6rBr)rrBr(r6)r6r(rrB)r6rBrr()rBr(r6r)rr6rBr(rr(rrrrr test_shape_4ds zTestHistogramdd.test_shape_4dcCstjdd}t|\}}t|dd\}}t|tdd\}}t||t|tdddd\}}t||t|tdtdd\}}t|d|dS)Nrr&Tr@rZrh)rrrrr]rr^)rr!r/r0Zn_histZw_histrrrrps   zTestHistogramdd.test_weightscCs:tdt}t|dd\}}t|dtdddgdS)N)rr&r&r)rrrDrE)rrkr^rrr.)rrcr/r0rrrtest_identical_sampless z&TestHistogramdd.test_identical_samplescCs`tgggddgddgfd\}}t|tdggtjggggdd\}}t|tddS)Nrr%r)rDr&)r&r&r&)rr rr.rkrzrrrr{s zTestHistogramdd.test_emptyc Cstddd}tttj|ddddgdtttj|ddddgdtttj|dddddd d ggdttj|dddddd dggddS) NrSr&r(rUr6r)r%gGz?r')rr;reshaper r-rr)rrcrrrtest_bins_errorssz TestHistogramdd.test_bins_errorsc Cstjddtddd}tddgddgddgg}tj|dtj ddggd \}}t||tj|dtd dtjggd \}}t||tj|dtj dtjggd \}}t||W5QRXdS) Nignore)invalidrBr'r&r%rrr)rU)rZerrstater;rr.rrHr )rrcexpectedr1r2rrrtest_inf_edgess  $ "zTestHistogramdd.test_inf_edgescCs dg}dddgg}t||d\}}t|ddkt|ddkdg}dddgg}t||d\}}t|ddkt|ddkdg}dddgg}t||d\}}t|ddkt|ddkd g}dddgg}t||d\}}t|ddkt|ddkdS) NgG?rDrEr+r)rr%g8?gqh?)rr)rrcr*r/_rrrtest_rightmost_binedges(    z&TestHistogramdd.test_rightmost_binedgecCsvtjd}t|ddgddgddggdttt|ddgddgdtjggdttt|ddgtjdgddggddS)N)rr'rDr+rGrrErK)rrrr r-rHrrrrrrs z!TestHistogramdd.test_finite_rangecCsltdddg}tdddg}tdddg}d}t||f||fd\}}tdgdgg}t||dS)z: Test that adjacent entries in an edge array can be equal rr%r&r)r,r+Nrr.rr)rrcrdx_edgesy_edgesr/r0Z hist_expectedrrrtest_equal_edgessz TestHistogramdd.test_equal_edgescCsrtdddg}|d}tddddg}|d}t||f||fd\}}t|dj|jt|dj|jdS) z< Test that if an edge array is input, its type is preserved rrrr6rNr)r%N)rr.rrr[)rrcrdrr r/r0rrrtest_edge_dtypeszTestHistogramdd.test_edge_dtypecCs`d}tdgtj}tddgtj}||}||}t||f||fd\}}t|dddS)NlrrUr%r))rr)rr.Zint64rr)rbigrcrrdr r/r0rrrtest_large_integerssz#TestHistogramdd.test_large_integerscCstdddg}tdddg}tddgddgg}tdgdgddgddgd}tdgdgddgddgd}t||f||fd \}}t||t||f||fd d \}}t|d dS) Nrr&rSrBr'r7r%rr)Trg?r)rrr Zrelative_areasrcrdr/r0rrrtest_density_non_uniform_2ds ** z+TestHistogramdd.test_density_non_uniform_2dcCsbtd}tdddddg}t||dd\}}t|f|fdd\}}t||t||ddS)Nrrr%r'rBTr@rr;r.rrrrr!r*r/r0hist_ddedges_ddrrrtest_density_non_uniform_1d9s   z+TestHistogramdd.test_density_non_uniform_1dcCsbtd}tdddddg}t||dd\}}t|f|fdd\}}t||t||ddS) Nrrr%r'rBTr@r4r&r'rrrtest_density_via_normedBs   z'TestHistogramdd.test_density_via_normedc CsPtd}tdddddg}ttdt|f|fddd\}}W5QRXdS) Nrrr%r'rBzCannot specify bothT)rAr5)rr;r.r r~r)rr!r*r(r)rrrtest_density_normed_redundancyKs   z.TestHistogramdd.test_density_normed_redundancyN)rrrr$rrrprr{rrrrr!r"r$r%r*r+r,rrrrrps"%        r)ZnumpyrZnumpy.lib.histogramsrrrZ numpy.testingrrrrr r r r r rZnumpy.testing._private.utilsrrrrrrrrrs0 *=