U b@sddlZddlmZmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZddlZGdddZGdddZGdd 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_warningsc@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"dAS)B TestHistogramcCsdSNselfrr_/home/fireinfo/NEWAFireInfo/venv/lib/python3.8/site-packages/numpy/lib/tests/test_histograms.pysetupszTestHistogram.setupcCsdSrrrrrrteardownszTestHistogram.teardowncCsTd}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_bins 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) rsuprecrr 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'???) rrrrrr8rr:rrinf) rrr r!r"r=r)countsZdmyrrr test_densityAs,       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)NrrDrr6ranger$TrKr@decimalrKweightsr@r)rKrQ)rr:rrrrr8)rr!r0r"wrrr test_outliersaszTestHistogram.test_outliersc CsNtdd}tdd}ttdt|ddg|dd\}}W5QRXdS) NrrD z same shape asr$r6TrP)rr:r r,r)rr!rUr0r"rrrtest_arr_weights_mismatchys z'TestHistogram.test_arr_weights_mismatchcCstdd}t|\}}tt|jtjt|dd\}}tt|jtjt|tdt d\}}tt|jtjt|tdt d\}}tt|jtjdS)NrrDTr?rQ) rr:rrZ issubdtypedtypeintegerZfloatingonesintfloat)rr!r0r"rrr test_types zTestHistogram.test_typecCsTtjdddgtjd}tjdddgtjd}tj||dd \}}}t|d dS) Ng2VEq@gx>fQg-VT5@rZg!@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$rr`zConverting input from .*TF) rr-Zuint8rrr7RuntimeWarningrr9r)rr!Zint_histZ int_edgesr;r<r.r/rrrtest_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)Nrr5Tr?rYrQr@rrrWr)rQr$r%r'r&)r)rQr@g$@rar6rAg?rBg333333?) rrrr\rr r concatenatezerosr:rr-r) rr rUr!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$rTr%y?rr&rhrS)Decimalr5)rr-rr rOrs)rvaluesrQrmrnrsrrrtest_exotic_weightss$&& z!TestHistogram.test_exotic_weightscCs8tdddg}tj|ddgddt|dddgdS)Nrprqrrrr)rKr))rr-rr )rrtrrrtest_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)NrCr*rnumr5g333333@)rrrr TypeErrorrvalsrrrtest_error_binnum_types z$TestHistogram.test_error_binnum_typecCsPtjdddd}t|ddgdttt|tjdgdttt|dtjgddS)NrCr*rr{rF?rJ)rrrr r,nanrGr~rrrtest_finite_rangeszTestHistogram.test_finite_rangec Cs<tjdddd}ttdtj|ddgdW5QRXdS) NrCr*rr{zmax must be larger thanrBg{Gz?rJ)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)rKrrTr$)rr-rzipr) rarrr.r/maskZ left_edgesZ right_edgesrbleftrightrrrtest_bin_edge_casess z!TestHistogram.test_bin_edge_casesc CsBtdddddddddg }tj|ddd \}}t|d d dS) NrCr*r+ra@@r5rrTr$)rr-rr)rrr.r/rrrtest_last_bin_inclusive_range sz+TestHistogram.test_last_bin_inclusive_rangec CsNtjdddd}tddgddgg}ttdtj||d W5QRXdS) NrCr*rr{rrDg333333?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&Zuint64r`r()rr-r r,r)rrr)r.r/rrr test_unsigned_monotonicity_checks  z.TestHistogram.test_unsigned_monotonicity_checkcCstttddtdDtj gtttddtdDtjgtddtdDdgtddtdDdgdS) NcSsg|]}tdqSg?rr-.0irrr #sz9TestHistogram.test_object_array_of_0d..rcSsg|]}tdqSrrrrrrr%scSsg|]}tdqSrDrrrrrr(sg ?cSsg|]}tdqSrrrrrrr)srD)r r,rrKrrGrrrrtest_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-rrfilterrer r,rrr)rZone_nanZall_nanr;r0r"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&r5rztimedelta64[D]r()rZ datetime64r-rZrastyperr]) rbeginoffsetsr)datesZ date_binstdZd_countZd_edgeZt_countZt_edgeZi_countZi_edgerrr test_datetimeDs    zTestHistogram.test_datetimecCs|dt|jd}tjd| dd|dg|d}t|dd\}}t|d| ddd|dgt|ddgdS)NrRr$r%r'r`r(r)rrZitemsizer-rr)rrZexponentrr.r1rrrdo_signed_overflow_bounds]s $"z'TestHistogram.do_signed_overflow_boundscCs@|tj|tj|tj|tj|tjdSr)rrbyteshortZintcint_Zlonglongrrrrtest_signed_overflow_boundsds     z)TestHistogram.test_signed_overflow_boundscCstt|j}tdg|}td|dg|}||ddkrDdStj|d|d\}}t|dgt|j|dS)Nr*r+rr$rrZfinfoepsr-rrrrZr float_small float_largerrrKcountZx_locrrrdo_precision_lower_boundks  z&TestHistogram.do_precision_lower_boundcCstt|j}tdg|}tdd|g|}||ddkrDdStj|d|d\}}t|dgt|j|dS)Nr*rCrTr$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'rCr*r+rarrrrrrr)rrrrr-)rr.r1r/rrrrtest_histogram_bin_edgess  z&TestHistogram.test_histogram_bin_edgesN)#__name__ __module__ __qualname__rrr#r2r>rIrVrXr_rdrfrorurwrzrrrrrrrrrrrrrrrrrrrrrr s@      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$rx)restimator_list estimatorr!r"rrrrzsz$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'rRrr%rrrrrrrrr r6#Gr2iirvrTr5r$r&*For the {0} estimator with datasize of {1}err_msgN)itemsrrrirrr9format) r basic_testtestlenexpectedResultsx1x2rbrnumbinsr!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:rrr9r) 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\rrrr9r)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)rrrrr+szLTestHistogramOptimBinNums.test_scott_vs_stone...rrr')startstopr|)rZ geomspaceroundrr])rr)rrr+szATestHistogramOptimBinNums.test_scott_vs_stone..rrrrDg333333?g ףp= ?g{Gz?gQ?r%rNN)rKabsrZmeanr)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. rRrMr)rrrrrrr rP!ErrvrTr5r%r$rr&ri)irrJzFor the {0} estimatorz with datasize of {0}rN)rrrrrrrr9) rrrrrrZx3rbrrr!r"msgrrrtest_simple_range2s4   z+TestHistogramOptimBinNums.test_simple_ranger)rrrrrrrcCsZtjdddgtjd}tj||d\}}tj|tj|d\}}t||t||dS)Nrr`r()rr-Zint8rrZint32r)rr)r!r.r/Zhist32Zedges32rrrtest_signed_integer_dataQs  z2TestHistogramOptimBinNums.test_signed_integer_datac Cs:dddddg}|D]"}tttdddg|dddgd qd S) z= Check that weighted data raises a TypeError rrrrrr$r%r&rYN)r r}r)rrrrrrtest_simple_weighted[s z.TestHistogramOptimBinNums.test_simple_weightedN)rrr__doc__rzr#rrrrrrrpytestmarkZ 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)NrrDrErq)r%r&r&rTr$rr&rJrr%Tr)r@g(@)r%r&r'r'rLg@cSsg|]}t|qSr)rZsqueeze)rrcrrrr~sz/TestHistogramdd.test_simple..r)r'r&r%r)r5r5r5r*r5) rr-rrrallr splitrjlistrKr:)rrbHr/ZanswerZedzZrrrr#gsN    $    &&zTestHistogramdd.test_simplecCs<d}tjdd}|D] }t||\}}t|j|kqdS)N))r5r'rA)rAr'r5)r5rAr')r'rAr5)rAr5r')r'r5rArr&rrrrrshaperr)rr"rr/rrr test_shape_3ds zTestHistogramdd.test_shape_3dcCs<d}tjdd}|D] }t||\}}t|j|kqdS)N))rr'r5rA)r'r5rrA)r5rAr'r)rrAr5r')r5rrAr')r'rArr5)rAr5rr')rr5r'rA)rr'rAr5)rAr'rr5)rArr5r')r'rAr5r)r'rr5rA)r5r'rAr)r5rr'rA)rArr'r5)rAr5r'r)r'rrAr5)r'r5rAr)rrAr'r5)r5r'rrA)r5rArr')rAr'r5r)rr5rAr'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%Tr?rYrg)rrrrr\rr])rr r.r/Zn_histZw_histrrrros   zTestHistogramdd.test_weightscCs:tdt}t|dd\}}t|dtdddgdS)N)rr%r%r(rrrCrD)rrjr]rrr-)rrbr.r/rrrtest_identical_sampless z&TestHistogramdd.test_identical_samplescCs`tgggddgddgfd\}}t|tdggtjggggdd\}}t|tddS)Nrr$r(rCr%)r%r%r%)rr rr-rjryrrrrzs zTestHistogramdd.test_emptyc Cstddd}tttj|ddddgdtttj|ddddgdtttj|dddddd d ggdttj|dddddd dggddS) NrRr%r'rTr5r(r$gGz?r&)rr:reshaper r,rr)rrbrrrtest_bins_errorssz TestHistogramdd.test_bins_errorsc Cstjddtddd}tddgddgddgg}tj|dtj ddggd \}}t||tj|dtd dtjggd \}}t||tj|dtj dtjggd \}}t||W5QRXdS) Nignore)invalidrAr&r%r$rrr(rT)rZerrstater:rr-rrGr )rrbexpectedr0r1rrrtest_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?rCrDr*r(rr$g8?gqh?)rr)rrbr)r._rrrtest_rightmost_binedges(    z&TestHistogramdd.test_rightmost_binedgecCsvtjd}t|ddgddgddggdttt|ddgddgdtjggdttt|ddgtjdgddggddS)N)rr&rCr*rFrrDrJ)rrrr r,rGrr~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)rrbrcx_edgesy_edgesr.r/Z 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 rrrr5rMr(r$N)rr-rrrZ)rrbrcrrr.r/rrrtest_edge_dtypeszTestHistogramdd.test_edge_dtypecCs`d}tdgtj}tddgtj}||}||}t||f||fd\}}t|dddS)NlrrTr$r()rr)rr-Zint64rr)rbigrbrrcrr.r/rrrtest_large_integerssz#TestHistogramdd.test_large_integerscCstdddg}tdddg}tddgddgg}tdgdgddgddgd}tdgdgddgddgd}t||f||fd \}}t||t||f||fd d \}}t|d dS) Nrr%rRrAr&r6r$rr(Trg?r)rrrZrelative_areasrbrcr.r/rrrtest_density_non_uniform_2ds ** z+TestHistogramdd.test_density_non_uniform_2dcCsbtd}tdddddg}t||dd\}}t|f|fdd\}}t||t||ddS)Nrrr$r&rATr?rr:r-rrrrr r)r.r/hist_ddedges_ddrrrtest_density_non_uniform_1d.s   z+TestHistogramdd.test_density_non_uniform_1dcCsbtd}tdddddg}t||dd\}}t|f|fdd\}}t||t||ddS) Nrrr$r&rATr?r3r"r#rrrtest_density_via_normed7s   z'TestHistogramdd.test_density_via_normedc CsPtd}tdddddg}ttdt|f|fddd\}}W5QRXdS) Nrrr$r&rAzCannot specify bothT)r@r4)rr:r-r r}r)rr r)r$r%rrrtest_density_normed_redundancy@s   z.TestHistogramdd.test_density_normed_redundancyN)rrrr#rrrorrzrrrrrrr r!r&r'r(rrrrres"%        r)ZnumpyrZnumpy.lib.histogramsrrrZ numpy.testingrrrrr r r r r rrrrrrrrrs0 =