U md~@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 TestHistogramcCsdSNselfrrX/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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#Zarear*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%TrCr5decimalrCweightsr5r*rCrI)rr=rrrrr<)rr"r1r#wrrr test_outliersJszTestHistogram.test_outliersc CsNtdd}tdd}ttdt|ddg|dd\}}W5QRXdS) Nrr9 z same shape asr%rATrH)rr=r r-r)rr"rMr1r#rrrtest_arr_weights_mismatchbs z'TestHistogram.test_arr_weights_mismatchcCstdd}t|\}}tt|jtjt|dd\}}tt|jtjt|tdt d\}}tt|jtjt|tdt d\}}tt|jtjdS)Nrr9Tr4rI) rr=rrZ issubdtypedtypeintegerZfloatingonesintfloat)rr"r1r#rrr test_typeis zTestHistogram.test_typecCsTtjdddgtjd}tjdddgtjd}tj||dd \}}}t|d dS) Ng2VEq@gx>fQg-VT5@rRg!@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%rrXzConverting 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)NrTr4rQrIr5rrrOr*rIr%r&r(r')r*rIr5g$@rYrAr6g?r7g333333?) rrrrTrr r concatenatezerosr=rr.r) rr!rMr"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%rLr&y?rr'rdrK)Decimalrb)rr.rr rGro)rvaluesrIrirjrorrrtest_exotic_weightss$&& z!TestHistogram.test_exotic_weightscCs8tdddg}tj|ddgddt|dddgdS)Nrlrmrnrr)rCr*)rr.rr )rrprrrtest_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+rnumrbg333333@)rrrr TypeErrorrvalsrrrtest_error_binnum_types z$TestHistogram.test_error_binnum_typecCsPtjdddd}t|ddgdttt|tjdgdttt|dtjgddS)Nr8r+rrwr;?rB)rrrr r-nanr>rzrrrtest_finite_rangeszTestHistogram.test_finite_rangec Cs<tjdddd}ttdtj|ddgdW5QRXdS) Nr8r+rrwzmax must be larger thanr7g{Gz?rB)rrr r-rrzrrrtest_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*rCrrLr%)rr.rzipr) rarrr/r0maskZ left_edgesZ right_edgesrZleftrightrrrtest_bin_edge_casess z!TestHistogram.test_bin_edge_casesc CsBtdddddddddg }tj|ddd \}}t|d d dS) Nr8r+r,rY@@rbrrLr%)rr.rr)rrr/r0rrrtest_last_bin_inclusive_rangesz+TestHistogram.test_last_bin_inclusive_rangec CsNtjdddd}tddgddgg}ttdtj||d W5QRXdS) Nr8r+rrwrr9g333333?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'Zuint64rXr))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-rrCrr>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.r~rfilterr^r r-rrr)rZone_nanZall_nanr`r1r#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'rbrztimedelta64[D]r))rZ datetime64r.rRrastyperrU) 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)NrJr%r&r(rXr)r)rrRitemsizer.rr)rrRexponentrr/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.rrrrRr float_small float_largerrrCcountZx_locrrrdo_precision_lower_boundTs  z&TestHistogram.do_precision_lower_boundcCstt|j}tdg|}tdd|g|}||ddkrDdStj|d|d\}}t|dgt|j|dS)Nr+r8rLr%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,rYrrrrrrr)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&)rrfr=rrtype)rrZxbinsZybinsZzbinsr/rrrtest_big_arrayss  zTestHistogram.test_big_arraysN)'__name__ __module__ __qualname__rrr$r3r@rNrPrWr\rarkrqrsrvr|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%rt)restimator_list estimatorr"r#rrrrvsz$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(rJrr&rrrrrrrrr rA#Gr2iirrrLrbr%r'*For the {0} estimator with datasize of {1}err_msgN)itemsrrrerrr_format) r basic_testtestlenexpectedResultsx1x2rZrnumbinsr"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)rrTrrrr_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)rrTrrrr.)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()startstoprx)rZ geomspaceroundrrU)rr)rrrszATestHistogramOptimBinNums.test_scott_vs_stone..rrrr9g333333?g ףp= ?g{Gz?gQ?r&rFN)rCabsrZmeanr)rllavgrrrtest_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. rJrEr)rrrrrrr rP!ErrrrLrbr&r%rr'ri)irrBzFor the {0} estimatorz with datasize of {0}rN)rrrrrrrr_) rrrrrrZx3rZrrr"r#msgrrrtest_simple_range%s4   z+TestHistogramOptimBinNums.test_simple_ranger*rrrrrrrcCsZtjdddgtjd}tj||d\}}tj|tj|d\}}t||t||dS)NrrXr))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'rQN)r ryr)rrrrrrtest_simple_weightedNs z.TestHistogramOptimBinNums.test_simple_weightedN)rrr__doc__rvr$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:rm)r&r'r'rLr%rr'rBrr&Tr*r5g(@)r&r'r(r(rDg@cSsg|]}t|qSr)rZsqueeze)rr[rrrrqsz/TestHistogramdd.test_simple..r)r(r'r&r)rbrbrbr+rb) rr.rrrallr splitrflistrCr=)rrZHr0ZanswerZedzZrrrr$ZsN    $    &&zTestHistogramdd.test_simplecCs<d}tjdd}|D] }t||\}}t|j|kqdS)N))rbr(r6)r6r(rb)rbr6r()r(r6rb)r6rbr()r(rbr6rr'rrrrrshaperr*rr#rr0rrr test_shape_3ds zTestHistogramdd.test_shape_3dcCs<d}tjdd}|D] }t||\}}t|j|kqdS)N))rr(rbr6)r(rbrr6)rbr6r(r)rr6rbr()rbrr6r()r(r6rrb)r6rbrr()rrbr(r6)rr(r6rb)r6r(rrb)r6rrbr()r(r6rbr)r(rrbr6)rbr(r6r)rbrr(r6)r6rr(rb)r6rbr(r)r(rr6rb)r(rbr6r)rr6r(rb)rbr(rr6)rbr6rr()r6r(rbr)rrbr6r(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&Tr4rQrc)rrrrrTrrU)rr!r/r0Zn_histZw_histrrrrks   zTestHistogramdd.test_weightscCs:tdt}t|dd\}}t|dtdddgdS)N)rr&r&r)rrr8r9)rrfrUrrr.)rrZr/r0rrrtest_identical_sampless z&TestHistogramdd.test_identical_samplescCs`tgggddgddgfd\}}t|tdggtjggggdd\}}t|tddS)Nrr%r)r8r&)r&r&r&)rr rr.rfrurrrrvs zTestHistogramdd.test_emptyc Cstddd}tttj|ddddgdtttj|ddddgdtttj|dddddd d ggdttj|dddddd dggddS) NrJr&r(rLrbr)r%gGz?r')rr=reshaper r-rr)rrZrrrtest_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)rL)rZerrstater=rr.rr>r )rrZexpectedr1r2rrrtest_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)rrZr*r/_rrrtest_rightmost_binedges(    z&TestHistogramdd.test_rightmost_binedgecCsvtjd}t|ddgddgddggdttt|ddgddgdtjggdttt|ddgtjdgddggddS)N)rr'r8r+r;r}r9rB)rrrr r-r>r~rzrrrrs 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)rrZr[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 rrrrbrEr)r%N)rr.rrrR)rrZr[rrr/r0rrrtest_edge_dtypeszTestHistogramdd.test_edge_dtypecCs`d}tdgtj}tddgtj}||}||}t||f||fd\}}t|dddS)NlrrLr%r))rr)rr.Zint64rr)rbigrZrr[rr/r0rrrtest_large_integerssz#TestHistogramdd.test_large_integerscCstdddg}tdddg}tddgddgg}tdgdgddgddgd}tdgdgddgddgd}t||f||fd \}}t||t||f||fd d \}}t|d dS) Nrr&rJr6r'rAr%rr)Trg?r)rrrZrelative_areasrZr[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$rrrkrrvrrrrrrr!r"r#rrrrrXs%      r)numpyrZnumpy.lib.histogramsrrrZ numpy.testingrrrrr r r r r rZnumpy.testing._private.utilsrrrrrrrrrs0 =