Current File : //lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_histograms.cpython-39.opt-1.pyc
a
z[yc� � @ s� d dl Zd dlmZmZmZ d dlmZmZm Z m
Z
m Z m Z m
Z
mZmZmZ d dlmZ d dlZG dd� d�ZG dd� d�ZG d d
� d
�ZdS ) � N)� histogram� histogramdd�histogram_bin_edges)
�assert_� assert_equal�assert_array_equal�assert_almost_equal�assert_array_almost_equal�
assert_raises�assert_allclose�assert_array_max_ulp�assert_raises_regex�suppress_warnings)�requires_memoryc @ s& e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"e#dAdB�e$j%j&dCdD� ��Z'dES )F�
TestHistogramc C s d S �N� ��selfr r �E/usr/lib64/python3.9/site-packages/numpy/lib/tests/test_histograms.py� setup_method s zTestHistogram.setup_methodc C s d S r r r r r r �teardown_method s z TestHistogram.teardown_methodc C sT d}t j�|�}t|�\}}tt j|dd�|� tt �ddd��\}}t|d� d S )N�d r �Zaxis�
)�np�random�randr r �sum�linspacer )r �n�v�a�br r r � test_simple s zTestHistogram.test_simplec C s� t g d�ddg�\}}t|dg� t|ddg� ttt ddgdd� t ddgdd�\}}t|t�dg�� t|t�ddg�� d S )N�� � � � r& r' r ��bins� �?� @)r r r
�
ValueErrorr r �arrayr )r �hist�edges�h�er r r � test_one_bin s zTestHistogram.test_one_binc C s t � }|�j |�tjd�}d}tj�|�}t|dd�\}}t�|t�|� �}t |d� t
t |�d� W d � n1 sz0 Y t � }|�n |�tjd�}t� d�}g d�}t||dd�\}}t�|t�|� �}t |d� t
t |�d� W d � n1 �s0 Y d S )Nz
.*normed.*r T��normedr& r )r r& � � r )
r �recordr ZVisibleDeprecationWarningr r r r �diffr r �len�arange) r �sup�recr r! r" r# �arear+ r r r � test_normed* s$
,
zTestHistogram.test_normedc C s d}t j�|�}t|dd�\}}t �|t �|� �}t|d� t �d�}g d�}t||dd�\}}t|d� t t �|t �|� �d� t||dd�\}}t|g d �� t �d�}d
dd d t j
g}t||dd�\}}t|g d
�� t jg d �ddt j
gdd�\}}t |dd
g� d S )Nr T��densityr& r �r r& r( � r 皙�����?Fr% r r( rD )rE rE rE � � �?� �?� �?) r r r r r r: r r< r r �inf) r r r! r" r# r? r+ ZcountsZdmyr r r � test_densityB s(
�
zTestHistogram.test_densityc C s� t �d�d }t|ddgd�\}}t|�� d� t|ddgd�\}}t|�� d� t|ddgdd�\}}t|t �|� �� dd d
� t �d�d }t|ddg|dd �\}}t|t �|� �� d� t|d ddg|d
�\}}t||dd� � d S )Nr rG r r8 ��ranger& T�rM rB � ��decimal�rM �weightsrB � �r+ rM rS ���)r r< r r r r r: )r r" r2 r# �wr r r �
test_outliersb s zTestHistogram.test_outliersc C sb t �d�d }t �d�d }ttd��( t|ddg|dd�\}}W d � n1 sT0 Y d S ) Nr rG � z
same shape asr& r8 TrR )r r< r
r. r )r r" rW r2 r# r r r �test_arr_weights_mismatchz s z'TestHistogram.test_arr_weights_mismatchc C s� t �d�d }t|�\}}tt �|jt j�� t|dd�\}}tt �|jt j�� t|t �dt �d�\}}tt �|jt j�� t|t �dt
�d�\}}tt �|jt j�� d S )Nr rG TrA �rS ) r r< r r Z
issubdtype�dtypeZintegerZfloating�ones�int�float)r r" r2 r# r r r � test_type� s zTestHistogram.test_typec C sP t jg d�t jd�}t jg d�t jd�}t j||dd�\}}}t|�� d� d S )N)g2V��Eq@g��x>fQ�g-��VT5@�r\ )g!����@g���S��@g���^z�@r r* � @)r r/ Zfloat32Z histogram2dr r )r �x�yZ counts_histZxedgesZyedgesr r r �test_f32_rounding� s z TestHistogram.test_f32_roundingc C s� t jg d�t jd�}t �|�\}}t� �P}|�td�}t �g d��\}}tt|�d� t ||� t ||� W d � n1 s~0 Y d S )N)r& r& r ra zConverting input from .*)TTFr& )
r r/ Zuint8r r r9 �RuntimeWarningr r; r )r r" Zint_histZ int_edgesr= r> r0 r1 r r r �test_bool_conversion� s
z"TestHistogram.test_bool_conversionc C sR t j�d�}t �d�d }t|�\}}t|dd�\}}t||d�\}}t||dd�\} }
t|d |� t|| � t �ddd�}t �t �d�t �d�f�}t|t � d �|d
�\}}t||� tg d �d g d
�d
�\}}t
|g d�� tg d �d g d
�dd�\}}t|t � g d��d d d � tt � d�g d�g d�dd�\}}t |g d�� d S )Nr r7 TrA r[ �rS rB r r rY �r+ rS )r& r' r' r) r) )r) r( r' r& )r) r7 r r& )r+ rS rB g $@rb r8 rC ) r' r& r& r& r& r& r& r& r& )g�������?rE rE g333333�?)
r r r r] r r r � concatenate�zerosr< r r/ r ) r r! rW r" r# Zna�nb�wa�wbZnwaZnwbr r r � test_weights� s.
�
�
zTestHistogram.test_weightsc C s< t �g d��}t �g d��dt �g d�� }t|g d�|d�\}}t|t �ddg�dt �dd g� � t|ddd g|d
�\}}t|t �ddg�dt �dd g� � d d lm} t �g d��}t �|d�|d�|d �g�}t|g d�|d�\}}t||d�|d
�g� t|ddd g|d
�\}}t||d�|d
�g� d S )N�g�������?� @gffffff@)r& rV r' y �?)r' r& r' )r r' r( ri r& r' r( rU r )�Decimalr7 )r r/ r r rQ rr )r �valuesrS rm rn rr r r r �test_exotic_weights� s && z!TestHistogram.test_exotic_weightsc C s4 t �g d��}t j|ddgdd� t|g d�� d S )Nrp ���r r )rM r+ )r r/ r r )r rs r r r �test_no_side_effects� s z"TestHistogram.test_no_side_effectsc C s> t g ddgd�\}}t|t�dg�� t|t�ddg�� d S )Nr r& r* �r r r r/ �r r" r# r r r �
test_empty� s zTestHistogram.test_emptyc C s, t jdddd�}t|d� ttt|d� d S )NrF r, r ��numr7 g333333@)r r r r
� TypeError�r �valsr r r �test_error_binnum_type� s
z$TestHistogram.test_error_binnum_typec C sP t jdddd�}t|ddgd� ttt|t jdgd� ttt|dt jgd� d S )NrF r, r rz rI � �?rL )r r r r
r. �nanrJ r} r r r �test_finite_range� s z TestHistogram.test_finite_rangec C sP t jdddd�}ttd��" t j|ddgd� W d � n1 sB0 Y d S ) NrF r, r rz zmax must be larger thanrE g{�G�z�?rL )r r r
r. r r} r r r �test_invalid_range� s z TestHistogram.test_invalid_rangec
C s~ t �g d��}t j|ddd�\}}|dk}|d d� | }|dd � | }t|||�D ]"\}}} t||k� t|| k � qVd S )N)iQ i� i� i& i� i i� ih )r' i� �r+ rM r rV r& )r r/ r �zipr )
r �arrr0 r1 �maskZ
left_edgesZ right_edgesrc �left�rightr r r �test_bin_edge_cases s z!TestHistogram.test_bin_edge_casesc C s4 t �g d��}t j|ddd�\}}t|d d� d S )N� rF rF rF r, r- rb rb g @g @� �� �r7 r� rV r& )r r/ r r )r r� r0 r1 r r r � test_last_bin_inclusive_range
s z+TestHistogram.test_last_bin_inclusive_rangec C sb t jdddd�}t �ddgddgg�}ttd�� t j||d � W d � n1 sT0 Y d S )
NrF r, r rz r rG g333333�?z
must be 1dr* )r r r/ r
r. r )r r~ r+ r r r �test_bin_array_dims s z!TestHistogram.test_bin_array_dimsc C s\ t �dg�}t jg d�dd�}tt��" t j||d�\}}W d � n1 sN0 Y d S )Nr' )r& r( r& Zuint64ra r* )r r/ r
r. r )r r� r+ r0 r1 r r r � test_unsigned_monotonicity_check s
z.TestHistogram.test_unsigned_monotonicity_checkc C s� t ttdd� td�D �tj g � t ttdd� td�D �tjg � t�dd� td�D �dg � t�dd� td�D �dg � d S ) Nc S s g | ]}t �d ��qS �g�������?�r r/ ��.0�ir r r �
<listcomp>$ � z9TestHistogram.test_object_array_of_0d.<locals>.<listcomp>r c S s g | ]}t �d ��qS r� r� r� r r r r� & r� c S s g | ]}t �d ��qS �rG r� r� r r r r� ) r� g �?c S s g | ]}t �d ��qS r� r� r� r r r r� * r� rG )r
r. r rM r rJ r r r r �test_object_array_of_0d! s �� z%TestHistogram.test_object_array_of_0dc C s t �ddt jg�}t �t jt jg�}t� }|�t� |�� ttt|dd� ttt|dd� t|ddd�\}}t |�
� d� t|ddd�\}}t |�
� d� t|ddgd�\}}t |�
� d� t|ddgd�\}}t |�
� d� W d � n1 s�0 Y d S )Nr r& �autor* �r r& r� r' ) r r/ r� r �filterrf r
r. r r r )r Zone_nanZall_nanr= r2 r# r r r �test_some_nan_values, s
z"TestHistogram.test_some_nan_valuesc
C s� t �dd�}t �g d��}t �g d��}|| }|| }t �d�}t||d�\}}t|�|�|�|�d�\} }
t||d�\} } t|| � t| | � t|| �t�| � t|
�t�| � t|j|j� t|
j|� d S )Nz
2000-01-01�D) r r r& r& r' r( r7 r � )r r' � r� ztimedelta64[D]r* )r Z
datetime64r/ r\ r �astyper r^ )
r ZbeginZoffsetsr+ ZdatesZ date_binsZtdZd_countZd_edgeZt_countZt_edgeZi_countZi_edger r r �
test_datetimeE s
zTestHistogram.test_datetimec C s| dt �|�j d }t jd| d d| d g|d�}t|dd�\}}t|d| d dd| d g� t|ddg� d S )NrT r& r' r) ra r* r )r r\ �itemsizer/ r r )r r\ Zexponentr� r0 r3 r r r �do_signed_overflow_bounds^ s
$"z'TestHistogram.do_signed_overflow_boundsc C s@ | � tj� | � tj� | � tj� | � tj� | � tj� d S r )r� r ZbyteZshortZintcZint_Zlonglongr r r r �test_signed_overflow_boundse s
z)TestHistogram.test_signed_overflow_boundsc C st t �|�j}t �dg|�}t �d| dg|�}|�|�d dkrDd S t j|d|d�\}}t|dg� t|j|� d S )Nr, r- r r& r� �r Zfinfo�epsr/ r� r r r\ �r � float_small� float_larger� r� rM �countZx_locr r r �do_precision_lower_boundl s z&TestHistogram.do_precision_lower_boundc C st t �|�j}t �dg|�}t �dd| g|�}|�|�d dkrDd S t j|d|d�\}}t|dg� t|j|� d S )Nr, rF rV r& r� r� r� r r r �do_precision_upper_bound} s z&TestHistogram.do_precision_upper_boundc C s | � ||� | �||� d S r )r� r� )r r� r� r r r � do_precision� s zTestHistogram.do_precisionc C sd | � tjtj� | � tjtj� | � tjtj� | � tjtj� | � tjtj� | � tjtj� d S r )r� r ZhalfZsingleZdoubleZ
longdoubler r r r �test_precision� s z TestHistogram.test_precisionc C s� t g d�ddg�\}}tg d�ddg�}t||� t�g d��}t |ddd�\}}t|ddd�}t||� t |dd d�\}}t|dd d�}t||� d S )
Nr% r& r' r� r� r� r� r� r� )r r r r r/ )r r0 r3 r1 r� r r r �test_histogram_bin_edges� s
z&TestHistogram.test_histogram_bin_edgesg _�B)Z
free_bytesc C sJ t �ddg�}d}d}t �d�}t j||||fd�}tt|�td�� d S )Ni �r( i� i�> )�sampler+ )r& r' )r rk r< r r �type)r r� ZxbinsZybinsZzbinsr0 r r r �test_big_arrays� s
z TestHistogram.test_big_arraysN)(�__name__�
__module__� __qualname__r r r$ r4 r@ rK rX rZ r` re rg ro rt rv ry r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r �pytest�markZslowr� r r r r r
sF
r c @ sz e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z e
j�dg d��dd� �Zdd� ZdS )�TestHistogramOptimBinNumsz]
Provide test coverage when using provided estimators for optimal number of
bins
c C sL g d�}|D ]:}t g |d�\}}t|t�dg�� t|t�ddg�� q d S )N��fd�scott�rice�sturges�doane�sqrtr� �stoner* r r& rw )r �estimator_list� estimatorr" r# r r r ry � s
z$TestHistogramOptimBinNums.test_emptyc C s� ddddddddd�dddddd dd
d�d d d d
d dd dd�d�}|� � D ]~\}}t�dd|d d �}t�dd|d d �}t�||f�}|� � D ]2\}}t�||�\} }
tt| �|d�||�d� q�qJdS )z�
Straightforward testing with a mixture of linspace data (for
consistency). All test values have been precomputed and the values
shouldn't change
r) rT r� r' r� � r � � r8 � �# � �G r� ��2 i� i� ru rV r7 r& r( �*For the {0} estimator with datasize of {1}��err_msgN)�itemsr r rj r r r; �format) r �
basic_test�testlen�expectedResults�x1�x2rc r� �numbinsr" r# r r r r$ � s& �����z%TestHistogramOptimBinNums.test_simplec
C s� dddddddd�dddddddd�dddddddd�d�}|� � D ]N\}}t�|�}|� � D ]2\}}t�||�\}}tt|�|d�||�d� q^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& )r� r� r� r� r� r� r� r' r( �r& r' r( r� r� N)r� r r< r r r; r� ) r Z small_datr� r� Ztestdatr� Zexpbinsr" r# r r r �
test_small� s" ����
�z$TestHistogramOptimBinNums.test_smallc C s( g d�}|D ]}t ttg d�|� q dS )zS
Check a Value Error is thrown when an unknown string is passed in
)ZmadZfreemanZ
histogramsZIQRr� N)r
r. r )r Z
check_listr� r r r �test_incorrect_methods� s z0TestHistogramOptimBinNums.test_incorrect_methodsc C s^ t �d�}ddddddddd�}|�� D ]0\}}t �||�\}}tt|�|d�|�d� q(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
r r& r� z {0} estimator, No Variance testr� N)r r] r� r r r; r� )r Z
novar_datasetZnovar_resultdictr� r� r" r# r r r �test_novariance� s
��z)TestHistogramOptimBinNums.test_novariancec C s� t �d�}d|dd�< d|dd�< t|d�}t|t �ddd�� t|d �}t|t �ddg�� t|d
�}t|t �ddd�� dS ) zu
Check when IQR is 0, but variance exists, we return the sturges value
and not the fd value.
i� r Nr( r ���r� r� r� r� )r r] r r r r/ )r Z lim_var_dataZ
edges_autoZedges_fdZ
edges_sturgesr r r �test_limited_variance s
z/TestHistogramOptimBinNums.test_limited_variancec C sh t �ddd�}t �t �ddd�|f�}dddd d
�}|�� D ]&\}}t �||�\}}tt|�|� q<d 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.
ru r r� i�������r7 � rY rD )r� r� r� r� N)r r �hstackr� r r r; )r ZxcenterZoutlier_datasetZoutlier_resultdictr� r� r" r# r r r � test_outlier s z&TestHistogramOptimBinNums.test_outlierc sJ dd� � � fdd�t d�D �}ttj|dd�d �}t|g d �d
d � d S )
zQVerify that Scott's rule and Stone's rule converges for normally distributed datac S sR t j�| �}|jdd|d�}tt �|d�d �tt �|d�d � }}||| S )Nr r' )�locZscale�sizer� r� )r r Z RandomStateZnormalr; r )�seedr� �rngrc r" r# r r r � nbins_ratio0 s *zBTestHistogramOptimBinNums.test_scott_vs_stone.<locals>.nbins_ratioc s6 g | ].� �� fd d�t jdddd��� �t�D ��qS )c s g | ]}� �|��qS r r )r� r� )r� r� r r r� 6 r� zLTestHistogramOptimBinNums.test_scott_vs_stone.<locals>.<listcomp>.<listcomp>r r r) )�start�stopr{ )r Z geomspace�roundr� r^ )r� �r� )r� r r� 6 s �zATestHistogramOptimBinNums.test_scott_vs_stone.<locals>.<listcomp>r r r rG )g333333�?g
ףp=
�?g{�G�z�?g� ��Q��?r' rP N)rM �absr Zmeanr )r ZllZavgr r� r �test_scott_vs_stone- s
�z-TestHistogramOptimBinNums.test_scott_vs_stonec
C s� ddddddd�ddddddd�d d d
d d dd�d �}|� � D ]�\}}t�d
d|d d �}t�dd|d d �}t�dd|�}t�|||f�}|� � D ]F\}} tj||dd�\}
} d�|�} | d�|�7 } tt|
�| | d� q�q>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.
rT rO r� )r� r� r� r� r� r� r� � r� �P �! �E � r� ru rV r7 r' r& r r( r� i���)i��r� rL zFor the {0} estimatorz with datasize of {0}r� N)r� r r r� r r� r r; )
r r� r� r� r� r� Zx3rc r� r� r" r# �msgr r r �test_simple_range= s(
����
z+TestHistogramOptimBinNums.test_simple_ranger+ )r� r� r� r� r� r� r� c C sX t jg d�t jd�}t j||d�\}}t j|�t j�|d�\}}t||� t||� d S )N)���r � ra r* )r r/ Zint8r r� Zint32r )r r+ r" r0 r1 Zhist32Zedges32r r r �test_signed_integer_data\ s
z2TestHistogramOptimBinNums.test_signed_integer_datac C s0 g d�}|D ] }t ttg d�|g d�d� q dS )z=
Check that weighted data raises a TypeError
)r� r� r� r� r� r� r[ N)r
r| r )r r� r� r r r �test_simple_weightedf s
�z.TestHistogramOptimBinNums.test_simple_weightedN)r� r� r� �__doc__ry r$ r� r� r� r� r� r� r� r� r� Z parametrizer� r� r r r r r� � s
r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$�TestHistogramddc C s: t �g d�g d�g d�g d�g d�g d�g�}t|ddd gd
d gd
d ggd �\}}t �g d
�g d�g d�gg d
�g d�g d�gg�}t||� g d�g d�g d�g}t||dd�\}}tt �||d k�� t|ddd gd
d gd
dggdd�\}}t �g d�g d�g d�gg d�g d�g d�gg�}t||d d� d d � t j|d d d �D �}t|d d d!gd
d gd
d!ggd"�\}}t �d
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||� t �d#�}d$|t t
d%��t t
d%��t t
d%��f< tt � d%�t � d%�t � d%�gd%�\}}t||� d S )&N)r� rG rH )r� rH rq )r� rq rG )rG rG rH )rG rH rq )rG rq rq )r' r( r( rV r& r r( rL )r r& r )r r r& )r& r r )r� r r' )r r& r' r( T�r+ rB g (@)r' r( r) r) rN )r r& r r )r r r& r )r& r r r g @c S s g | ]}t �|��qS r )r Zsqueeze)r� rd r r r r� � r� z/TestHistogramdd.test_simple.<locals>.<listcomp>r )r) r( r' r� r' r� )r7 r7 r7 r, r7 ) r r/ r r r �allr �splitrk �listrM r< )r rc �Hr1 ZanswerZed�z�Zr r r r$ r sF ��
�
�
��
�
&&zTestHistogramdd.test_simplec C s< d}t j�dd�}|D ] }t||�\}}t|j|k� qd S )N))r7 r) rD )rD r) r7 )r7 rD r) )r) rD r7 )rD r7 r) )r) r7 rD r r( �r r r r r �shape�r r+ �rr# r� r1 r r r �
test_shape_3d� s
z TestHistogramdd.test_shape_3dc C s< d}t j�dd�}|D ] }t||�\}}t|j|k� qd S )N))r� r) r7 rD )r) r7 r� rD )r7 rD r) r� )r� rD r7 r) )r7 r� rD r) )r) rD r� r7 )rD r7 r� r) )r� r7 r) rD )r� r) rD r7 )rD r) r� r7 )rD r� r7 r) )r) rD r7 r� )r) r� r7 rD )r7 r) rD r� )r7 r� r) rD )rD r� r) r7 )rD r7 r) r� )r) r� rD r7 )r) r7 rD r� )r� rD r) r7 )r7 r) r� rD )r7 rD r� r) )rD r) r7 r� )r� r7 rD r) r r) r r r r r �
test_shape_4d� s
z TestHistogramdd.test_shape_4dc C s� t j�dd�}t|�\}}t|dd�\}}t|t �d�d�\}}t||� t|t �d�d dd�\}}t||� t|t �dt�d d�\}}t|d| � d S )Nr r' TrA r[ rh )r r r r r] r r^ )r r! r0 r1 Zn_histZw_histr r r ro � s
z TestHistogramdd.test_weightsc C s8 t �dt�}t|dd�\}}t|d t �g d��� d S )N)r r' r' r* r )r� rF rG )r rk r^ r r r/ )r rc r0 r1 r r r �test_identical_samples� s z&TestHistogramdd.test_identical_samplesc C s` t g g gddgddgfd�\}}t|t�dgg�� tj g g g gdd�\}}t|t�d�� d S )Nr r& r* rF r' )r' r' r' )r r r r/ rk rx r r r ry � s zTestHistogramdd.test_emptyc C s~ t �d��dd�}ttt j|g d�d� ttt j|g d�d� ttt j|dddg d�gd� tt j|dddg d �gd�� d S )
NrT r' r) )rV r' r) r7 r* )r& g�G�z��?r& r& r& )r& r' r( ���r% )r r< �reshaper
r. r r )r rc r r r �test_bins_errors� s �z TestHistogramdd.test_bins_errorsc C s� t jdd��� t �d��dd�}t �ddgddgddgg�}t j|dt j ddggd �\}}t||� t j|dt �d
dt jg�gd �\}}t||� t j|dt j dt jggd �\}}t||� W d � n1 s�0 Y d S ) N�ignore)ZinvalidrD r( r' r& r r r* rV )r Zerrstater< r
r/ r rJ r )r rc Zexpectedr2 r3 r r r �test_inf_edges� s
$
"z TestHistogramdd.test_inf_edgesc C s dg}g d�g}t ||d�\}}t|d dk� t|d dk� dg}g d�g}t ||d�\}}t|d dk� t|d dk� dg}g d�g}t ||d�\}}t|d dk� t|d dk� d g}g d�g}t ||d�\}}t|d dk� t|d dk� d S )
Ng�G�����?)rF rG r, r* r rF r& r, g8� �?gq���h �?)r r )r rc r+ r0 �_r r r �test_rightmost_binedge� s(
z&TestHistogramdd.test_rightmost_binedgec C sv t j�d�}t|ddgddgddggd� ttt|ddgddgdt jggd� ttt|ddgt jdgddggd� d S )N)r r( rF r, rI r� rG rL )r r r r
r. rJ r� r} r r r r� � s ��z!TestHistogramdd.test_finite_rangec C sf t �g d��}t �g d��}t �g d��}d}t||f||fd�\}}t �dgdgg�}t||� dS )z: Test that adjacent entries in an edge array can be equal )r r& r' )r r' r' r& r* r- r, N�r r/ r r )r rc rd �x_edges�y_edgesr0 r1 Z
hist_expectedr r r �test_equal_edges� s �z TestHistogramdd.test_equal_edgesc C sl t �g d��}|d }t �g d��}|d }t||f||fd�\}}t|d j|j� t|d j|j� dS )z< Test that if an edge array is input, its type is preserved )r r r� r )r r7 rO r� r* r r& N)r r/ r r r\ )r rc rd r r r0 r1 r r r �test_edge_dtype s z TestHistogramdd.test_edge_dtypec C s` d}t �dgt j�}t �ddgt j�}|| }|| }t||f||fd�\}}t|d d� d S )Nl r rV r& r* )r r )r r/ Zint64r r )r �bigrc r rd r r0 r1 r r r �test_large_integers s z#TestHistogramdd.test_large_integersc C s� t �g d��}t �g d��}t �ddgddgg�}t �dgdgd dgd dgd �}t �dgdgd dgd dgd �}t||f||fd�\}}t||� t||f||fdd �\}}t|d
� d S ) N)r r' rT )r rD rT r( r8 r& r� r* Tr� g �?r )r r r Zrelative_areasrc rd r0 r1 r r r �test_density_non_uniform_2d s �**
z+TestHistogramdd.test_density_non_uniform_2dc C s\ t �d�}t �g d��}t||dd�\}}t|f|fdd�\}}t||� t||d � d S )Nr rC TrA r �r r< r/ r r r �r r! r+ r0 r1 �hist_dd�edges_ddr r r �test_density_non_uniform_1d9 s
z+TestHistogramdd.test_density_non_uniform_1dc C s\ t �d�}t �g d��}t||dd�\}}t|f|fdd�\}}t||� t||d � d S )Nr rC TrA r5 r r r r r r �test_density_via_normedB s
z'TestHistogramdd.test_density_via_normedc C s^ t �d�}t �g d��}ttd��( t|f|fddd�\}}W d � n1 sP0 Y d S )Nr rC zCannot specify bothT)rB r6 )r r< r/ r
r| r )r r! r+ r r r r r � test_density_normed_redundancyK s
�z.TestHistogramdd.test_density_normed_redundancyN)r� r� r� r$ r r ro r ry r r
r r� r r r r r r r r r r r r� p s" %
r� )Znumpyr Znumpy.lib.histogramsr r r Z
numpy.testingr r r r r r
r r r
r Z numpy.testing._private.utilsr r� r r� r� r r r r �<module> s 0 * =
Mini Shell