o V=ig@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 TestHistogramcCdSNselfrreC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\numpy/lib/tests/test_histograms.py setup_methodzTestHistogram.setup_methodcCrrrrrrrteardown_methodrzTestHistogram.teardown_methodcCsTd}tj|}t|\}}ttj|dd|ttddd\}}t|ddS)NdrZaxis )nprandomrandrrsumlinspacer)rnvabrrr test_simples   zTestHistogram.test_simplecCstgdddg\}}t|dgt|ddgtttddgddtddgdd\}}t|tdgt|tddgdS)Nr(r)rbins?@)rrr ValueErrorrrarrayr )rhistedgesherrr test_one_bin s zTestHistogram.test_one_binc Cst}|5|tjd}d}tj|}t|dd\}}t|t|}t |dt t |dWdn1s=wYt}|8|tjd}t d}gd}t||dd\}}t|t|}t |dt t |dWddS1swYdS)Nz .*normed.*rTnormedr(r)rr( r) rrecordrZVisibleDeprecationWarningrrrr diffrrlenarange) rsuprecr"r#r$r%arear-rrr test_normed*s(     "zTestHistogram.test_normedc Csd}tj|}t|dd\}}t|t|}t|dtd}gd}t||dd\}}t|dt t|t|dt||dd\}}t|gd td}d dd d tj g}t||dd\}}t|gd tjgd ddtj gdd\}}t |dd gdS)NrTdensityr(rrr(r*r皙?Fr'rr*rF)rGrGrG???) rrrrr r<rr>rrinf) rr"r#r$r%rAr-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)NrrIrr:ranger(TrPrDdecimalrPweightsrDr-rPrV)rr>rrr rr<)rr$r4r%wrrr test_outliersbszTestHistogram.test_outlierscCsdtdd}tdd}ttdt|ddg|dd\}}WddS1s+wYdS) NrrI z same shape asr(r:TrU)rr>r r0r)rr$rZr4r%rrrtest_arr_weights_mismatchzs  "z'TestHistogram.test_arr_weights_mismatchcCstdd}t|\}}tt|jtjt|dd\}}tt|jtjt|tdt d\}}tt|jtjt|tdt d\}}tt|jtjdS)NrrITrCrV) rr>rrZ issubdtypedtypeintegerZfloatingonesintfloat)rr$r4r%rrr test_types zTestHistogram.test_typecCsPtjgdtjd}tjgdtjd}tj||dd\}}}t|ddS)N)g2VEq@gx>fQg-VT5@r_)g!@gS@g^z@rr,@)rr1Zfloat32Z histogram2drr )rxyZ counts_histZxedgesZyedgesrrrtest_f32_roundingszTestHistogram.test_f32_roundingcCstjgdtjd}t|\}}t)}|td}tgd\}}tt|dt ||t ||WddS1s@wYdS)N)r(r(rrezConverting input from .*)TTFr() rr1Zuint8rrr;RuntimeWarningrr=r)rr$Zint_histZ int_edgesr?r@r2r3rrrtest_bool_conversions   "z"TestHistogram.test_bool_conversionc CsRtjd}tdd}t|\}}t|dd\}}t||d\}}t||dd\} } t|d|t|| tddd}ttdtdf}t|t d |d \}}t||tgd d gd d \}}t |gdtgd d gd dd\}}t|t gdddd tt dgdgddd\}}t |gddS)Nrr9TrCr^rVrDrrr\r-rV)r(r)r)r+r+)r+r*r)r()r+r9rr()r-rVrDg$@rfr:rE) r)r(r(r(r(r(r(r(r()g?rGrGg333333?) rrrrarr r! concatenatezerosr>rr1r) rr#rZr$r%nanbwawbZnwaZnwbrrr test_weightss.      zTestHistogram.test_weightscCs<tgd}tgddtgd}t|gd|d\}}t|tddgdtdd gt|ddd g|d \}}t|tddgdtdd gd d lm}tgd}t|d|d|d g}t|gd|d\}}t||d|d gt|ddd g|d \}}t||d|d gdS)Ng?@gffffff@)r(rYr)y?)r)r(r))rr)r*rmr(r)r*rXr)Decimalr9)rr1rr rTrw)rvaluesrVrrrsrwrrrtest_exotic_weightss && z!TestHistogram.test_exotic_weightscCs4tgd}tj|ddgddt|gddS)Nrurr)rPr-)rr1rr )rrxrrrtest_no_side_effectssz"TestHistogram.test_no_side_effectscCs>tgddgd\}}t|tdgt|tddgdS)Nrr(r,rrrr1rr$r%rrr test_emptyszTestHistogram.test_emptycCs,tjdddd}t|dttt|ddS)NrHr.rnumr9g333333@)rr!rr TypeErrorrvalsrrrtest_error_binnum_types z$TestHistogram.test_error_binnum_typecCsPtjdddd}t|ddgdttt|tjdgdttt|dtjgddS)NrHr.rrrK?rO)rr!rr r0nanrLrrrrtest_finite_rangeszTestHistogram.test_finite_rangecCsRtjdddd}ttdtj|ddgdWddS1s"wYdS) NrHr.rrzmax must be larger thanrGg{Gz?rO)rr!r r0rrrrrtest_invalid_ranges "z TestHistogram.test_invalid_rangec Cs~tgd}tj|ddd\}}|dk}|dd|}|dd|}t|||D]\}}} t||kt|| kq+dS)N)iQiii&iiiih )r)ir-rPrrYr()rr1rzipr) rarrr2r3maskZ left_edgesZ right_edgesrgleftrightrrrtest_bin_edge_casess z!TestHistogram.test_bin_edge_casescCs4tgd}tj|ddd\}}t|dddS)N rHrHrHr.r/rfrfg@g@r9rrYr()rr1rr)rrr2r3rrrtest_last_bin_inclusive_range sz+TestHistogram.test_last_bin_inclusive_rangecCsdtjdddd}tddgddgg}ttdtj||d WddS1s+wYdS) NrHr.rrrrIg333333?z must be 1dr,)rr!r1r r0r)rrr-rrrtest_bin_array_dimss  "z!TestHistogram.test_bin_array_dimscCs^tdg}tjgddd}tttj||d\}}WddS1s(wYdS)Nr))r(r*r(Zuint64rer,)rr1r r0r)rrr-r2r3rrr test_unsigned_monotonicity_checks  "z.TestHistogram.test_unsigned_monotonicity_checkcCstttddtdDtj gtttddtdDtjgtddtdDdgtddtdDdgdS) NcSg|]}tdqSg?rr1.0irrr $z9TestHistogram.test_object_array_of_0d..rcSrrrrrrrr&rcSrrIrrrrrr)rg ?cSrrrrrrrr*rrI)r r0rrPrrLrrrrtest_object_array_of_0d!s"z%TestHistogram.test_object_array_of_0dcCstddtjg}ttjtjg}t}|t|[ttt|ddttt|ddt|ddd\}}t | dt|ddd\}}t | dt|ddgd\}}t | dt|ddgd\}}t | dWddS1szwYdS)Nrr(autor,rr(rr)) rr1rrfilterrjr r0rrr )rZone_nanZall_nanr?r4r%rrrtest_some_nan_values,s  "z"TestHistogram.test_some_nan_valuesc Cstdd}tgd}tgd}||}||}td}t||d\}}t||||d\} } t||d\} } t|| t| | t||t| t| t| t|j|jt| j|dS)Nz 2000-01-01D) rrr(r(r)r*r9r)rr)rztimedelta64[D]r,)rZ datetime64r1r_rastyperrb) 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)NrWr(r)r+rer,r)rr_itemsizer1rr)rr_exponentrr2r5rrrdo_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|}||ddkr"dStj|d|d\}}t|dgt|j|dS)Nr.r/rr(rrZfinfoepsr1rrrr_r float_small float_largerrrPcountZx_locrrrdo_precision_lower_boundl  z&TestHistogram.do_precision_lower_boundcCstt|j}tdg|}tdd|g|}||ddkr"dStj|d|d\}}t|dgt|j|dS)Nr.rHrYr(rrrrrrdo_precision_upper_bound}rz&TestHistogram.do_precision_upper_boundcCs||||||dSr)rr)rrrrrr do_precisions zTestHistogram.do_precisioncCsd|tjtj|tjtj|tjtj|tjtj|tjtj|tjtjdSr)rrZhalfsingledoubleZ longdoublerrrrtest_precisions zTestHistogram.test_precisioncCstgdddg\}}tgdddg}t||tgd}t|ddd\}}t|ddd}t||t|dd d\}}t|dd d}t||dS) Nr'r(r)rrrrrr)rrrrr1)rr2r5r3rrrrtest_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))rror>rrtype)rrZxbinsZybinsZzbinsr2rrrtest_big_arrayss  zTestHistogram.test_big_arraysN)(__name__ __module__ __qualname__rrr&r6rBrNr[r]rdrirkrtryr{r~rrrrrrrrrrrrrrrrrrpytestmarkZslowrrrrrr sH      rc@szeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ e jdgdddZddZdS)TestHistogramOptimBinNumsz] Provide test coverage when using provided estimators for optimal number of bins cCsLgd}|D]}tg|d\}}t|tdgt|tddgqdS)Nfdscottricesturgesdoanesqrtrstoner,rr(r|)restimator_list estimatorr$r%rrrr~s z$TestHistogramOptimBinNums.test_emptyc Csddddddddddddddd dd dd d d d d dd ddd}|D]?\}}tdd|dd}tdd|dd}t||f}|D]\}}t||\} } tt| |d||dqJq%dS)z Straightforward testing with a mixture of linspace data (for consistency). All test values have been precomputed and the values shouldn't change r+rWrr)rr r:#Gr2iirzrYr9r(r**For the {0} estimator with datasize of {1}err_msgN)itemsrr!rnrrr=format) r basic_testtestlenexpectedResultsx1x2rgrnumbinsr$r%rrrr&s* z%TestHistogramOptimBinNums.test_simplec Csddddddddddddddddddddddddd}|D]'\}}t|}|D]\}}t||\}}tt||d||dq/q"dS) 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_smalls&  z$TestHistogramOptimBinNums.test_smallcCs(gd}|D] }tttgd|qdS)zS Check a Value Error is thrown when an unknown string is passed in )ZmadZfreemanZ histogramsZIQRrN)r r0r)rZ check_listrrrrtest_incorrect_methodssz0TestHistogramOptimBinNums.test_incorrect_methodsc Cs^td}ddddddddd}|D]\}}t||\}}tt||d|dqdS)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)rrarrrr=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)rrarrr!r1)rZ lim_var_dataZ edges_autoZedges_fdZ edges_sturgesrrrtest_limited_variances      z/TestHistogramOptimBinNums.test_limited_variancecCshtddd}ttddd|f}dddd d }|D]\}}t||\}}tt||qd S) a Check the FD, Scott and Doane with outliers. The FD estimates a smaller binwidth since it's less affected by outliers. Since the range is so (artificially) large, this means more bins, most of which will be empty, but the data of interest usually is unaffected. The Scott estimator is more affected and returns fewer bins, despite most of the variance being in one area of the data. The Doane estimator lies somewhere between the other two. rzrrir9r\rF)rrrrN)rr!hstackrrrr=)rZxcenterZoutlier_datasetZoutlier_resultdictrrr$r%rrr test_outliers z&TestHistogramOptimBinNums.test_outliercsJddfddtdD}ttj|ddd}t|gd d d d S) zQVerify that Scott's rule and Stone's rule converges for normally distributed datacSsRtj|}|jdd|d}tt|ddtt|dd}}|||S)Nrr))locZscalesizerr)rrZ RandomStatenormalr=r)seedrrngrgr$r%rrr nbins_ratio0s * zBTestHistogramOptimBinNums.test_scott_vs_stone..nbins_ratiocs6g|]fddtjddddtDqS)csg|]}|qSrr)rr)rrrrr6rzLTestHistogramOptimBinNums.test_scott_vs_stone...rrr+)startstopr)rZ geomspaceroundrrb)rr)rrr6s.zATestHistogramOptimBinNums.test_scott_vs_stone..rrrrI)g333333?g ףp= ?g{Gz?gQ?r)rSN)rPabsrmeanr)rZllavgrrrtest_scott_vs_stone-s  z-TestHistogramOptimBinNums.test_scott_vs_stonec Csddddddddddddddd d d d d ddd }|D]Q\}}td d|dd}tdd|dd}tdd|}t|||f}|D]#\}} tj||dd\} } d|} | d|7} tt| | | dqLqdS)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. rWrRr)rrrrrrr rP!ErrzrYr9r)r(rr*ri)irrOzFor the {0} estimatorz with datasize of {0}rN)rrr!rrrrr=) rrrrrrZx3rgrrr$r%msgrrrtest_simple_range=s,   z+TestHistogramOptimBinNums.test_simple_ranger-)rrrrrrrcCsXtjgdtjd}tj||d\}}tj|tj|d\}}t||t||dS)N)rrer,)rr1Zint8rrZint32r)rr-r$r2r3Zhist32Zedges32rrrtest_signed_integer_data\s  z2TestHistogramOptimBinNums.test_signed_integer_datacCs0gd}|D]}tttgd|gddqdS)z= Check that weighted data raises a TypeError )rrrrrrr^N)r rr)rrrrrrtest_simple_weightedfs  z.TestHistogramOptimBinNums.test_simple_weightedN)rrr__doc__r~r&rrrrrrrrrZ parametrizer r rrrrrs   rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#S)$TestHistogramddc Cs:tgdgdgdgdgdgdg}t|ddd gd d gd d ggd \}}tgd gdgdggd gdgdgg}t||gdgdgdg}t||dd\}}tt||dkt|ddd gd d gd dggdd\}}tgdgdgdggdgdgdgg}t||ddddtj|d d dD}t|dd d!gd d gd d!ggd"\}}td 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||td#}d$|t t d%t t d%t t d%f<tt d%t d%t d%gd%\}}t||dS)&N)rrIrJ)rrJrv)rrvrI)rIrIrJ)rIrJrv)rIrvrv)r)r*r*rYr(rr*rO)rr(r)rrr()r(rr)r rr))rr(r)r*Tr-rDg(@)r)r*r+r+rQ)rr(rr)rrr(r)r(rrrg@cSsg|]}t|qSr)rZsqueeze)rrhrrrrrz/TestHistogramdd.test_simple..r)r+r*r)r r)r)r9r9r9r.r9) rr1rrrallr splitrolistrPr>)rrgHr3ZanswerZedzZrrrr&rsF      &&zTestHistogramdd.test_simplecC<d}tjdd}|D]}t||\}}t|j|kq dS)N))r9r+rF)rFr+r9)r9rFr+)r+rFr9)rFr9r+)r+r9rFrr*rrrrrshaperr-rr%rr3rrr test_shape_3ds zTestHistogramdd.test_shape_3dcCr)N))rr+r9rF)r+r9rrF)r9rFr+r)rrFr9r+)r9rrFr+)r+rFrr9)rFr9rr+)rr9r+rF)rr+rFr9)rFr+rr9)rFrr9r+)r+rFr9r)r+rr9rF)r9r+rFr)r9rr+rF)rFrr+r9)rFr9r+r)r+rrFr9)r+r9rFr)rrFr+r9)r9r+rrF)r9rFrr+)rFr+r9r)rr9rFr+rr+rrrrr test_shape_4ds zTestHistogramdd.test_shape_4dcCstjdd}t|\}}t|dd\}}t|tdd\}}t||t|tdddd\}}t||t|tdtdd\}}t|d|dS)Nrr)TrCr^rl)rrrrrarrb)rr#r2r3Zn_histZw_histrrrrts   zTestHistogramdd.test_weightscCs8tdt}t|dd\}}t|dtgddS)N)rr)r)r,r)rrHrI)rrorbrrr1)rrgr2r3rrrtest_identical_sampless z&TestHistogramdd.test_identical_samplescCs`tgggddgddgfd\}}t|tdggtjggggdd\}}t|tddS)Nrr(r,rHr))r)r)r))rr rr1ror}rrrr~s zTestHistogramdd.test_emptyc Cs~tddd}tttj|gddtttj|gddtttj|dddgdgdttj|dddgd gddS) NrWr)r+)rYr)r+r9r,)r(gGz?r(r(r()r(r)r*r')rr>reshaper r0rr)rrgrrrtest_bins_errorss"z TestHistogramdd.test_bins_errorsc Cstjddbtddd}tddgddgddgg}tj|dtj ddggd \}}t||tj|dtd dtjggd \}}t||tj|dtj dtjggd \}}t||WddS1skwYdS) Nignore)invalidrFr*r)r(rrr,rY)rZerrstater>rr1rrLr )rrgexpectedr4r5rrrtest_inf_edgess  $ " "zTestHistogramdd.test_inf_edgescCsdg}gdg}t||d\}}t|ddkt|ddkdg}gdg}t||d\}}t|ddkt|ddkdg}gdg}t||d\}}t|ddkt|ddkd g}gdg}t||d\}}t|ddkt|ddkdS) NgG?)rHrIr.r,rrHr(r.g8?gqh?)rr)rrgr-r2_rrrtest_rightmost_binedges(    z&TestHistogramdd.test_rightmost_binedgecCsvtjd}t|ddgddgddggdttt|ddgddgdtjggdttt|ddgtjdgddggddS)N)rr*rHr.rKrrIrO)rrrr r0rLrrrrrrs  z!TestHistogramdd.test_finite_rangecCsftgd}tgd}tgd}d}t||f||fd\}}tdgdgg}t||dS)z: Test that adjacent entries in an edge array can be equal )rr(r))rr)r)r(r,r/r.Nrr1rr)rrgrhx_edgesy_edgesr2r3Z hist_expectedrrrtest_equal_edgessz TestHistogramdd.test_equal_edgescCsltgd}|d}tgd}|d}t||f||fd\}}t|dj|jt|dj|jdS)z< Test that if an edge array is input, its type is preserved )rrrr)rr9rRrr,rr(N)rr1rrr_)rrgrhr(r)r2r3rrrtest_edge_dtypeszTestHistogramdd.test_edge_dtypecCs`d}tdgtj}tddgtj}||}||}t||f||fd\}}t|dddS)NlrrYr(r,)rr)rr1Zint64rr)rbigrgr(rhr)r2r3rrrtest_large_integerssz#TestHistogramdd.test_large_integerscCstgd}tgd}tddgddgg}tdgdgddgddgd}tdgdgddgddgd}t||f||fd\}}t||t||f||fdd \}}t|d dS) N)rr)rW)rrFrWr*r:r(rr,Trg?r')rr(r)Zrelative_areasrgrhr2r3rrrtest_density_non_uniform_2ds ** z+TestHistogramdd.test_density_non_uniform_2dcCs\td}tgd}t||dd\}}t|f|fdd\}}t||t||ddS)NrrETrCrrr>r1rrrrr#r-r2r3hist_ddedges_ddrrrtest_density_non_uniform_1d9  z+TestHistogramdd.test_density_non_uniform_1dcCs\td}tgd}t||dd\}}t|f|fdd\}}t||t||ddS)NrrETrCr7rr/r0rrrtest_density_via_normedBr4z'TestHistogramdd.test_density_via_normedcCs`td}tgd}ttdt|f|fddd\}}WddS1s)wYdS)NrrEzCannot specify bothT)rDr8)rr>r1r rr)rr#r-r1r2rrrtest_density_normed_redundancyKs    "z.TestHistogramdd.test_density_normed_redundancyN)rrrr&rrrtrr~r r$r&rr*r+r-r.r3r5r6rrrrrps$%       r)ZnumpyrZnumpy.lib.histogramsrrrZ numpy.testingrrrrr r r r r rZnumpy.testing._private.utilsrrrrrrrrrs0 *=