a Oc@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 TestHistogramcCsdSNselfrrV/var/www/html/gsl.tools/lib/python3.9/site-packages/numpy/lib/tests/test_histograms.pysetupszTestHistogram.setupcCsdSrrrrrrteardownszTestHistogram.teardowncCsTd}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_bins zTestHistogram.test_one_binc Cst}|j|tjd}d}tj|}t|dd\}}t|t|}t |dt t |dWdn1sz0Yt}|n|tjd}t d}gd}t||dd\}}t|t|}t |dt t |dWdn1s0YdS)Nz .*normed.*rTnormedr%r)rr% r) rrecordrZVisibleDeprecationWarningrrrrdiffrrlenarange) rsuprecrr 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'rC)rDrDrD???) rrrrrr9rr;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)NrrFrr7ranger%TrMrAdecimalrMweightsrAr*rMrS)rr;rrrrr9)rr!r1r"wrrr test_outliersaszTestHistogram.test_outlierscCsbtdd}tdd}ttd(t|ddg|dd\}}Wdn1sT0YdS) NrrF z same shape asr%r7TrR)rr;r r-r)rr!rWr1r"rrrtest_arr_weights_mismatchys z'TestHistogram.test_arr_weights_mismatchcCstdd}t|\}}tt|jtjt|dd\}}tt|jtjt|tdt d\}}tt|jtjt|tdt d\}}tt|jtjdS)NrrFTr@rS) rr;rrZ issubdtypedtypeintegerZfloatingonesintfloat)rr!r1r"rrr test_types zTestHistogram.test_typecCsPtjgdtjd}tjgdtjd}tj||dd\}}}t|ddS)N)g2VEq@gx>fQg-VT5@r\)g!@gS@g^z@rr)@)rr.float32Z histogram2drr)rxyZ counts_histZxedgesZyedgesrrrtest_f32_roundingszTestHistogram.test_f32_roundingcCstjgdtjd}t|\}}tP}|td}tgd\}}tt|dt ||t ||Wdn1s~0YdS)N)r%r%rrbzConverting input from .*)TTFr%) rr.Zuint8rrr8RuntimeWarningrr:r)rr!Zint_histZ int_edgesr<r=r/r0rrrtest_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)Nrr6Tr@r[rSrArrrYr*rS)r%r&r&r(r()r(r'r&r%)r(r6rr%)r*rSrAg$@rcr7rB) r&r%r%r%r%r%r%r%r%)g?rDrDg333333?) rrrr^rr r concatenatezerosr;rr.r) rr rWr!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%rVr&y?)r&r%r&)rr&r'rkr%r&r'rUr)Decimalr6)rr.rr rQru)rvaluesrSrprqrurrrtest_exotic_weightss && z!TestHistogram.test_exotic_weightscCs4tgd}tj|ddgddt|gddS)Nrsrr)rMr*)rr.rr )rrvrrrtest_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)NrEr+rnumr6g333333@)rrrr TypeErrorrvalsrrrtest_error_binnum_types z$TestHistogram.test_error_binnum_typecCsPtjdddd}t|ddgdttt|tjdgdttt|dtjgddS)NrEr+rr}rH?rL)rrrr r-nanrIrrrrtest_finite_rangeszTestHistogram.test_finite_rangecCsPtjdddd}ttd"tj|ddgdWdn1sB0YdS) NrEr+rr}zmax must be larger thanrDg{Gz?rL)rrr r-rrrrrtest_invalid_ranges z TestHistogram.test_invalid_rangec Cs~tgd}tj|ddd\}}|dk}|dd|}|dd|}t|||D]"\}}} t||kt|| kqVdS)N)iQiii&iiiih )r&ir*rMrrVr%)rr.rzipr) rarrr/r0maskZ left_edgesZ right_edgesreleftrightrrrtest_bin_edge_casess z!TestHistogram.test_bin_edge_casescCs4tgd}tj|ddd\}}t|dddS)N rErErEr+r,rcrcg@g@r6rrVr%)rr.rr)rrr/r0rrrtest_last_bin_inclusive_range sz+TestHistogram.test_last_bin_inclusive_rangecCsbtjdddd}tddgddgg}ttdtj||d Wdn1sT0YdS) NrEr+rr}rrFg333333?z must be 1dr))rrr.r r-r)rrr*rrrtest_bin_array_dimss z!TestHistogram.test_bin_array_dimscCs\tdg}tjgddd}tt"tj||d\}}Wdn1sN0YdS)Nr&)r%r'r%Zuint64rbr))rr.r r-r)rrr*r/r0rrr test_unsigned_monotonicity_checks  z.TestHistogram.test_unsigned_monotonicity_checkcCstttddtdDtj gtttddtdDtjgtddtdDdgtddtdDdgdS) NcSsg|]}tdqSg?rr..0irrr #z9TestHistogram.test_object_array_of_0d..rcSsg|]}tdqSrrrrrrr%rcSsg|]}tdqSrFrrrrrr(rg ?cSsg|]}tdqSrrrrrrr)rrF)r r-rrMrrIrrrrtest_object_array_of_0d sz%TestHistogram.test_object_array_of_0dcCstddtjg}ttjtjg}t}|t|ttt|ddttt|ddt|ddd\}}t | dt|ddd\}}t | dt|ddgd\}}t | dt|ddgd\}}t | dWdn1s0YdS)Nrr%autor)rr%rr&) rr.rrfilterrhr r-rrr)rZone_nanZall_nanr<r1r"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'r6r)rr&rztimedelta64[D]r))rZ datetime64r.r\rastyperr_) 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)NrTr%r&r(rbr)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_boundsds     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_largerrrMcountZx_locrrrdo_precision_lower_boundks  z&TestHistogram.do_precision_lower_boundcCstt|j}tdg|}tdd|g|}||ddkrDdStj|d|d\}}t|dgt|j|dS)Nr+rErVr%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)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)rrrrr.)rr/r2r0rrrrtest_histogram_bin_edgess  z&TestHistogram.test_histogram_bin_edgesN)#__name__ __module__ __qualname__rrr#r3r?rKrXrZrargrirrrwryr|rrrrrrrrrrrrrrrrrrrrrr s@      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|tddgq dS)Nfdscottricesturgesdoanesqrtrstoner)rr%rz)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]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(rTrr&rr r7#Gr2iirxrVr6r%r'*For the {0} estimator with datasize of {1}err_msgN)itemsrrrlrrr:format) r basic_testtestlenexpectedResultsx1x2rernumbinsr!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_smalls" z$TestHistogramOptimBinNums.test_smallcCs(gd}|D]}tttgd|q dS)zS Check a Value Error is thrown when an unknown string is passed in )ZmadZfreemanZ histogramsZIQRrN)r r-r)rZ check_listrrrrtest_incorrect_methodssz0TestHistogramOptimBinNums.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)rrrrr+rzLTestHistogramOptimBinNums.test_scott_vs_stone...rrr()startstopr~)rZ geomspaceroundrr_)rr)rrr+szATestHistogramOptimBinNums.test_scott_vs_stone..rrrrF)g333333?g ףp= ?g{Gz?gQ?r&rPN)rMabsrmeanr)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. rTrOr)rrrrrrr rP!ErrxrVr6r&r%rr'ri)irrLzFor the {0} estimatorz with datasize of {0}rN)rrrrrrrr:) rrrrrrZx3rerrr!r"msgrrrtest_simple_range2s(   z+TestHistogramOptimBinNums.test_simple_ranger*)rrrrrrrcCsXtjgdtjd}tj||d\}}tj|tj|d\}}t||t||dS)N)rrbr))rr.Zint8rrint32r)rr*r!r/r0Zhist32Zedges32rrrtest_signed_integer_dataQs  z2TestHistogramOptimBinNums.test_signed_integer_datacCs0gd}|D]}tttgd|gddq dS)z= Check that weighted data raises a TypeError )rrrrrrr[N)r rr)rrrrrrtest_simple_weighted[s  z.TestHistogramOptimBinNums.test_simple_weightedN)rrr__doc__r|r#rrrrrrrpytestmarkZ parametrizerrrrrrrs  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)rrFrG)rrGrt)rrtrF)rFrFrG)rFrGrt)rFrtrt)r&r'r'rVr%rr'rL)rr%r)rrr%)r%rr)rrr&)rr%r&r'Tr*rAg(@)r&r'r(r(rN)rr%rr)rrr%r)r%rrrg@cSsg|]}t|qSr)rZsqueeze)rrfrrrr~rz/TestHistogramdd.test_simple..r)r(r'r&rr&r)r6r6r6r+r6) rr.rrrallr splitrmlistrMr;)rreHr0ZanswerZedzZrrrr#gsF      &&zTestHistogramdd.test_simplecCs<d}tjdd}|D] }t||\}}t|j|kqdS)N))r6r(rC)rCr(r6)r6rCr()r(rCr6)rCr6r()r(r6rCrr'rrrrrshaperr*rr"r r0rrr test_shape_3ds zTestHistogramdd.test_shape_3dcCs<d}tjdd}|D] }t||\}}t|j|kqdS)N))rr(r6rC)r(r6rrC)r6rCr(r)rrCr6r()r6rrCr()r(rCrr6)rCr6rr()rr6r(rC)rr(rCr6)rCr(rr6)rCrr6r()r(rCr6r)r(rr6rC)r6r(rCr)r6rr(rC)rCrr(r6)rCr6r(r)r(rrCr6)r(r6rCr)rrCr(r6)r6r(rrC)r6rCrr()rCr(r6r)rr6rCr(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@r[rj)rrrrr^rr_)rr r/r0Zn_histZw_histrrrrrs   zTestHistogramdd.test_weightscCs8tdt}t|dd\}}t|dtgddS)N)rr&r&r)r)rrErF)rrmr_rrr.)rrer/r0rrrtest_identical_sampless z&TestHistogramdd.test_identical_samplescCs`tgggddgddgfd\}}t|tdggtjggggdd\}}t|tddS)Nrr%r)rEr&)r&r&r&)rr rr.rmr{rrrr|s zTestHistogramdd.test_emptyc Cs~tddd}tttj|gddtttj|gddtttj|dddgdgdttj|dddgd gddS) NrTr&r()rVr&r(r6r))r%gGz?r%r%r%)r%r&r'r$)rr;reshaper r-rr)rrerrrtest_bins_errorssz TestHistogramdd.test_bins_errorsc Cstjddtddd}tddgddgddgg}tj|dtj ddggd \}}t||tj|dtd dtjggd \}}t||tj|dtj dtjggd \}}t||Wdn1s0YdS) Nignore)invalidrCr'r&r%rrr)rV)rZerrstater;rr.rrIr )rreexpectedr1r2rrrtest_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?)rErFr+r)rrEr%r+g8?gqh?)rr)rrer*r/_rrrtest_rightmost_binedges(    z&TestHistogramdd.test_rightmost_binedgecCsvtjd}t|ddgddgddggdttt|ddgddgdtjggdttt|ddgtjdgddggddS)N)rr'rEr+rHrrFrL)rrrr r-rIrrrrrrs 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+Nrr.rr)rrerfx_edgesy_edgesr/r0Z 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)rr6rOrr)rr%N)rr.rrr\)rrerfr!r"r/r0rrrtest_edge_dtypeszTestHistogramdd.test_edge_dtypecCs`d}tdgtj}tddgtj}||}||}t||f||fd\}}t|dddS)NlrrVr%r))rr)rr.int64rr)rbigrer!rfr"r/r0rrrtest_large_integerssz#TestHistogramdd.test_large_integerscCstgd}tgd}tddgddgg}tdgdgddgddgd}tdgdgddgddgd}t||f||fd\}}t||t||f||fdd \}}t|d dS) N)rr&rT)rrCrTr'r7r%rr)Tr g?r )rr!r"Zrelative_areasrerfr/r0rrrtest_density_non_uniform_2ds ** z+TestHistogramdd.test_density_non_uniform_2dcCs\td}tgd}t||dd\}}t|f|fdd\}}t||t||ddS)NrrBTr@rrr;r.rrrrr r*r/r0hist_ddedges_ddrrrtest_density_non_uniform_1d.s   z+TestHistogramdd.test_density_non_uniform_1dcCs\td}tgd}t||dd\}}t|f|fdd\}}t||t||ddS)NrrBTr@r4rr)r*rrrtest_density_via_normed7s   z'TestHistogramdd.test_density_via_normedcCs^td}tgd}ttd(t|f|fddd\}}Wdn1sP0YdS)NrrBzCannot specify bothT)rAr5)rr;r.r rr)rr r*r+r,rrrtest_density_normed_redundancy@s   z.TestHistogramdd.test_density_normed_redundancyN)rrrr#rrrrrr|rrrrr#r$r'r(r-r.r/rrrrres"%        r)numpyrZnumpy.lib.histogramsrrrZ numpy.testingrrrrr r r r r rrrrrrrrrs0 =