U O8c~@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@seZdZddZddZddZddZd 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!e"d?d@e#j$j%dAdBZ&dCS)D TestHistogramcCsdSNselfrrC/tmp/pip-unpacked-wheel-fd_gsd75/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 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(???) rrrrrdiffrarangerrinf) rr r!r"r#arear*countsZdmyrrr test_density*s,       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)Nrr9r ranger%TrDr5decimalrDweightsr5r*rDrJ)rr=rrrrr<)rr"r1r#wrrr test_outliersJszTestHistogram.test_outliersc CsNtdd}tdd}ttdt|ddg|dd\}}W5QRXdS) Nrr9 z same shape asr%rBTrI)rr=r r-r)rr"rNr1r#rrrtest_arr_weights_mismatchbs z'TestHistogram.test_arr_weights_mismatchcCstdd}t|\}}tt|jtjt|dd\}}tt|jtjt|tdt d\}}tt|jtjt|tdt d\}}tt|jtjdS)Nrr9Tr4rJ) rr=rrZ issubdtypedtypeintegerZfloatingonesintfloat)rr"r1r#rrr test_typeis zTestHistogram.test_typecCsTtjdddgtjd}tjdddgtjd}tj||dd \}}}t|d dS) Ng2VEq@gx>fQg-VT5@rSg!@gS@g^z@rr)@)rr.Zfloat32Z histogram2drr)rxyZ counts_histZxedgesZyedgesrrrtest_f32_roundingxszTestHistogram.test_f32_roundingc Cs|tjdddgtjd}t|\}}tH}|td}tdddg\}}tt|dt ||t ||W5QRXdS)Nr%rrYzConverting input from .*TF) rr.Zuint8rrrecordRuntimeWarningrlenr)rr"Zint_histZ int_edgessupZrecr/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)NrTr4rRrJr5rrrPr*rJr%r&r(r')r*rJr5g$@rZrBr6g?r7g333333?) rrrrUrr r concatenatezerosr=rr.r) rr!rNr"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%rMr&y?rr'rerL)Decimalrc)rr.rr rHrp)rvaluesrJrjrkrprrrtest_exotic_weightss$&& z!TestHistogram.test_exotic_weightscCs8tdddg}tj|ddgddt|dddgdS)Nrmrnrorr)rDr*)rr.rr )rrqrrrtest_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)Nr8r+rnumrcg333333@)rrrr TypeErrorrvalsrrrtest_error_binnum_types z$TestHistogram.test_error_binnum_typecCsPtjdddd}t|ddgdttt|tjdgdttt|dtjgddS)Nr8r+rrxr;?rC)rrrr r-nanr>r{rrrtest_finite_rangeszTestHistogram.test_finite_rangec Cs<tjdddd}ttdtj|ddgdW5QRXdS) Nr8r+rrxzmax must be larger thanr7g{Gz?rC)rrr r-rr{rrrtest_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*rDrrMr%)rr.rzipr) rarrr/r0maskZ left_edgesZ right_edgesr[leftrightrrrtest_bin_edge_casess z!TestHistogram.test_bin_edge_casesc CsBtdddddddddg }tj|ddd \}}t|d d dS) Nr8r+r,rZ@@rcrrMr%)rr.rr)rrr/r0rrrtest_last_bin_inclusive_rangesz+TestHistogram.test_last_bin_inclusive_rangec CsNtjdddd}tddgddgg}ttdtj||d W5QRXdS) Nr8r+rrxrr9g333333?z must be 1dr))rrr.r r-r)rr|r*rrrtest_bin_array_dimss z!TestHistogram.test_bin_array_dimsc CsJtdg}tjdddgdd}tttj||d\}}W5QRXdS)Nr&r%r'Zuint64rYr))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|]}tdqSrrrrrrrscSsg|]}tdqSr9rrrrrrsg ?cSsg|]}tdqSrrrrrrrsr9)r r-rrDrr>rrrrtest_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.rrfilterr_r r-rrr)rZone_nanZall_nanrar1r#rrrtest_some_nan_valuess 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'rcrztimedelta64[D]r))rZ datetime64r.rSrastyperrV) rbeginoffsetsr*datesZ date_binstdZd_countZd_edgeZt_countZt_edgeZi_countZi_edgerrr test_datetime-s    zTestHistogram.test_datetimecCs|dt|jd}tjd| dd|dg|d}t|dd\}}t|d| ddd|dgt|ddgdS)NrKr%r&r(rYr)r)rrSitemsizer.rr)rrSexponentrr/r2rrrdo_signed_overflow_boundsFs $"z'TestHistogram.do_signed_overflow_boundscCs@|tj|tj|tj|tj|tjdSr)rrbyteshortZintcint_Zlonglongrrrrtest_signed_overflow_boundsMs     z)TestHistogram.test_signed_overflow_boundscCstt|j}tdg|}td|dg|}||ddkrDdStj|d|d\}}t|dgt|j|dS)Nr+r,rr%rrZfinfoepsr.rrrrSr float_small float_largerrrDcountZx_locrrrdo_precision_lower_boundTs  z&TestHistogram.do_precision_lower_boundcCstt|j}tdg|}tdd|g|}||ddkrDdStj|d|d\}}t|dgt|j|dS)Nr+r8rMr%rrrrrrdo_precision_upper_boundes  z&TestHistogram.do_precision_upper_boundcCs||||||dSr)rr)rrrrrr do_precisionvs zTestHistogram.do_precisioncCsd|tjtj|tjtj|tjtj|tjtj|tjtj|tjtjdSr)rrZhalfZsingledoubleZ longdoublerrrrtest_precisionzs 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(r8r+r,rZrrrrrrr)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&)rrgr=rrtype)rrZxbinsZybinsZzbinsr/rrrtest_big_arrayss  zTestHistogram.test_big_arraysN)'__name__ __module__ __qualname__rrr$r3rArOrQrXr]rbrlrrrtrwr}rrrrrrrrrrrrrrrrrpytestmarkZslowrrrrrr sD      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%ru)restimator_list estimatorr"r#rrrrwsz$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(rKrr&rrrrrrrrr rB#Gr2iirsrMrcr%r'*For the {0} estimator with datasize of {1}err_msgN)itemsrrrfrrr`format) r basic_testtestlenexpectedResultsx1x2r[rnumbinsr"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)rrUrrrr`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)rrUrrrr.)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)rrrrrszLTestHistogramOptimBinNums.test_scott_vs_stone...rrr()startstopry)rZ geomspaceroundrrV)rr)rrrszATestHistogramOptimBinNums.test_scott_vs_stone..rrrr9g333333?g ףp= ?g{Gz?gQ?r&rGN)rDabsrZmeanr)rZllavgrrrtest_scott_vs_stones  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. rKrFr)rrrrrrr rP!ErrsrMrcr&r%rr'ri)irrCzFor the {0} estimatorz with datasize of {0}rN)rrrrrrrr`) rrrrrrZx3r[rrr"r#msgrrrtest_simple_range%s4   z+TestHistogramOptimBinNums.test_simple_ranger*rrrrrrrcCsZtjdddgtjd}tj||d\}}tj|tj|d\}}t||t||dS)NrrYr))rr.Zint8rrZint32r)rr*r"r/r0Zhist32Zedges32rrrtest_signed_integer_dataDs  z2TestHistogramOptimBinNums.test_signed_integer_datac Cs:dddddg}|D]"}tttdddg|dddgd qd S) z= Check that weighted data raises a TypeError rrrrrr%r&r'rRN)r rzr)rrrrrrtest_simple_weightedNs z.TestHistogramOptimBinNums.test_simple_weightedN)rrr__doc__rwr$rrrrrrrrrZ parametrizerrrrrrrs"  rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdS) 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)Nrr9r:rn)r&r'r'rMr%rr'rCrr&Tr*r5g(@)r&r'r(r(rEg@cSsg|]}t|qSr)rZsqueeze)rr\rrrrqsz/TestHistogramdd.test_simple..r)r(r'r&r)rcrcrcr+rc) rr.rrrallr splitrglistrDr=)rr[Hr0ZanswerZedzZrrrr$ZsN    $    &&zTestHistogramdd.test_simplecCs<d}tjdd}|D] }t||\}}t|j|kqdS)N))rcr(r6)r6r(rc)rcr6r()r(r6rc)r6rcr()r(rcr6rr'rrrrrshaperr*rr#rr0rrr test_shape_3ds zTestHistogramdd.test_shape_3dcCs<d}tjdd}|D] }t||\}}t|j|kqdS)N))rr(rcr6)r(rcrr6)rcr6r(r)rr6rcr()rcrr6r()r(r6rrc)r6rcrr()rrcr(r6)rr(r6rc)r6r(rrc)r6rrcr()r(r6rcr)r(rrcr6)rcr(r6r)rcrr(r6)r6rr(rc)r6rcr(r)r(rr6rc)r(rcr6r)rr6r(rc)rcr(rr6)rcr6rr()r6r(rcr)rrcr6r(rr(r r rrr test_shape_4ds zTestHistogramdd.test_shape_4dcCstjdd}t|\}}t|dd\}}t|tdd\}}t||t|tdddd\}}t||t|tdtdd\}}t|d|dS)Nrr&Tr4rRrd)rrrrrUrrV)rr!r/r0Zn_histZw_histrrrrls   zTestHistogramdd.test_weightscCs:tdt}t|dd\}}t|dtdddgdS)N)rr&r&r)rrr8r9)rrgrVrrr.)rr[r/r0rrrtest_identical_sampless z&TestHistogramdd.test_identical_samplescCs`tgggddgddgfd\}}t|tdggtjggggdd\}}t|tddS)Nrr%r)r8r&)r&r&r&)rr rr.rgrvrrrrws zTestHistogramdd.test_emptyc Cstddd}tttj|ddddgdtttj|ddddgdtttj|dddddd d ggdttj|dddddd dggddS) NrKr&r(rMrcr)r%gGz?r')rr=reshaper r-rr)rr[rrrtest_bins_errorssz TestHistogramdd.test_bins_errorsc Cstjddtddd}tddgddgddgg}tj|dtj ddggd \}}t||tj|dtd dtjggd \}}t||tj|dtj dtjggd \}}t||W5QRXdS) Nignore)invalidr6r'r&r%rrr)rM)rZerrstater=rr.rr>r )rr[expectedr1r2rrrtest_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?r8r9r+r)rr%g8?gqh?)rr)rr[r*r/_rrrtest_rightmost_binedges(    z&TestHistogramdd.test_rightmost_binedgecCsvtjd}t|ddgddgddggdttt|ddgddgdtjggdttt|ddgtjdgddggddS)N)rr'r8r+r;r~r9rC)rrrr r-r>rr{rrrrs 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)rr[r\x_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 rrrrcrFr)r%N)rr.rrrS)rr[r\rrr/r0rrrtest_edge_dtypeszTestHistogramdd.test_edge_dtypecCs`d}tdgtj}tddgtj}||}||}t||f||fd\}}t|dddS)NlrrMr%r))rr)rr.Zint64rr)rbigr[rr\rr/r0rrrtest_large_integerssz#TestHistogramdd.test_large_integerscCstdddg}tdddg}tddgddgg}tdgdgddgddgd}tdgdgddgddgd}t||f||fd \}}t||t||f||fd d \}}t|d dS) Nrr&rKr6r'rBr%rr)Trg?r)rrrZrelative_areasr[r\r/r0rrrtest_density_non_uniform_2ds ** z+TestHistogramdd.test_density_non_uniform_2dcCsbtd}tdddddg}t||dd\}}t|f|fdd\}}t||t||ddS)Nrrr%r'r6Tr4)rr=r.rrr)rr!r*r/r0Zhist_ddZedges_ddrrrtest_density_non_uniform_1d!s   z+TestHistogramdd.test_density_non_uniform_1dN)rrrr$rrrlrrwrrrrrrr r!r"rrrrrXs%      r)ZnumpyrZnumpy.lib.histogramsrrrZ numpy.testingrrrrr r r r r rZnumpy.testing._private.utilsrrrrrrrrrs0 =