
    H/Ph                     @    d dl Zd dlmZmZmZmZ  G d d          ZdS )    N)assert_assert_array_equalassert_allclosesuppress_warningsc                   b    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd Zd Zd Zd ZdS )TestRegressionc                     t           j                            g dg d          }t          t           j                            |          g dg           d S )N)r            r            )r   r   r   r
   r
   r
   r   r   mask)r
   r   r      )npmamasked_arrayr   nonzero)selfxs     ^/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numpy/ma/tests/test_regression.pytest_masked_array_createz'TestRegression.test_masked_array_create   sZ    E777$<$<$<  > >25==++lll^<<<<<    c                 J    t           j                            ddg           d S )Nr
   r   )r   r   arrayr   s    r   test_masked_arrayz TestRegression.test_masked_array   s"    
AQC     r   c                     ddl m}m} t          j        d          }t          j        |j        |          } |||          }||z
   d S )Nr   )masked_whereMaskType)r
   r
   )numpy.mar!   r"   r   zerosshape)r   r!   r"   abcs         r   test_mem_masked_wherez$TestRegression.test_mem_masked_where   sZ    33333333HVHQWh''LA	!r   c                     t           j                            d          }t           j        j        |d<   t          j        d          }||z   ||z   d S )N)r   r
   )r   r   )r   r   )r   r   r$   maskedr   r&   r'   s      r   test_masked_array_multiplyz)TestRegression.test_masked_array_multiply   sG    EKK%,$HV	!	!r   c                 p    t           j                            dgd                              d           d S )Nr
   Fr   
   )r   r   r   repeatr   s    r   test_masked_array_repeatz'TestRegression.test_masked_array_repeat"   s0    
QCe$$++B/////r   c                 ^    t          t          j                            d                     d S )NUnicode)reprr   r   r   r   s    r   test_masked_array_repr_unicodez-TestRegression.test_masked_array_repr_unicode&   s$    RU[[##$$$$$r   c                     t           j                            g dg d          }t          j        |          }t	          |j        j        dk               t	          |j        j        dk               d S )N)g        g333333?g      @)FTFr   r
   r   )r   r   r   
atleast_2dr   r   ndimr,   s      r   test_atleast_2dzTestRegression.test_atleast_2d*   sj    E5I5I5IJJM!q !!!q !!!!!r   c                     t           j                            g dg d          }d|_        t	          |j        dk               d S )N)r&   r'   r(   )r
   r   r   r   X)r   r   r   
fill_valuer   )r   r&   s     r   test_set_fill_value_unicode_py3z.TestRegression.test_set_fill_value_unicode_py31   sF    EYYY??#$$$$$r   c                    t           j                            t          j        d          d          }t           j                            dt                    }|                    |           t          |j        dk               d S )Nr   Tr   dtype)outr   )r   r   r   arangefloatvarr   _data)r   r&   mouts      r    test_var_sets_maskedarray_scalarz/TestRegression.test_var_sets_maskedarray_scalar7   sc    EKK	!4K00u{{2U{++	$
a     r   c                    t           j                            g dd          }t          j        g d          }t	                      5 }|                    t          d           t           j                            ||d          }t           j                            ||d          }t          |j	        |j	                   d d d            d S # 1 swxY w Y   d S )N)r
   r   r   r   r   r   )r   g      @g@r   r   zbias and ddof have no effectr   )ddofr
   )
r   r   masked_equalr   r   filterDeprecationWarningcorrcoefr   data)r   r   ysupr0r1s         r   test_ddof_corrcoefz!TestRegression.test_ddof_corrcoef>   s    E22H((())   	.CJJ)+IJJJ11--B11--BBGRW---	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	.s   A:CCCc                 X   t           j                            ddgddg          }t          |j        j        dk               t          j        |d          }t          |j        j        dk               t          |j        dk               t          |j        j        dk               d S )N      ?       @Fr   )r   )r   r
   r   r   )r   r   MaskedArrayr   r   r%   tiler,   s      r   test_mask_not_backmangledz(TestRegression.test_mask_not_backmangledJ   s    Er2heU^<<$%%%GAv$%%%6!"""&'''''r   c                     t           j                            g dd          }t          |g          |d d                    d S )N))r
   rV   )r   rW   )r   g      @zi4,f4r@   r   )r   r   rY   r   r   r   s     r   test_empty_list_on_structuredz,TestRegression.test_empty_list_on_structuredT   sE    U:::'JJ2b62bqb6*****r   c                     t           j                            d                              d          }t	          |                    d          |j                                                   d S )Nr   rX   F)order)r   r   rC   reshaper   tobytesTr]   s     r   !test_masked_array_tobytes_fortranz0TestRegression.test_masked_array_tobytes_fortranY   sO    U\\!__$$U++2::C:00"$,,..AAAAAr   c                     t           j                            ddt           j        fddt           j        fdt           j        fgfg           d S )N)r
   )r   r   r   rP   ijr@   )r   r   r   int_voidr   s    r   test_structured_arrayz$TestRegression.test_structured_array]   sX    
O>#rw#rw @AC 	 	D 	D 	D 	D 	Dr   N)__name__
__module____qualname__r   r   r)   r-   r1   r5   r9   r=   rH   rT   r[   r^   re   rk    r   r   r   r      s        = = =! ! !    0 0 0% % %" " "% % %! ! !
. 
. 
.( ( (+ + +
B B BD D D D Dr   r   )numpyr   numpy.testingr   r   r   r   r   ro   r   r   <module>rr      s                  
ZD ZD ZD ZD ZD ZD ZD ZD ZD ZDr   