
    Cwg                     8    d dl Zd dlmZmZmZmZ  G d d      Zy)    N)assert_assert_array_equalassert_allclosesuppress_warningsc                   `    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y)TestRegressionc                     t         j                  j                  g dg d      }t        t         j                  j	                  |      g dg       y )N)r            r            )r   r   r   r
   r
   r
   r   r   mask)r
   r   r      )npmamasked_arrayr   nonzero)selfxs     U/var/www/horilla/myenv/lib/python3.12/site-packages/numpy/ma/tests/test_regression.pytest_masked_array_createz'TestRegression.test_masked_array_create   s:    EE7$<  >255==+l^<    c                 H    t         j                  j                  ddg       y )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
   y )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   s:    3HHVHHQWWh'A	!r   c                     t         j                  j                  d      }t         j                  j                  |d<   t        j                  d      }||z   ||z   y )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   sA    EEKK%%,,$HHV	!	!r   c                 f    t         j                  j                  dgd      j                  d       y )Nr
   Fr   
   )r   r   r   repeatr   s    r   test_masked_array_repeatz'TestRegression.test_masked_array_repeat"   s#    
QCe$++B/r   c                 T    t        t        j                  j                  d             y )NUnicode)reprr   r   r   r   s    r   test_masked_array_repr_unicodez-TestRegression.test_masked_array_repr_unicode&   s    RUU[[#$r   c                     t         j                  j                  g dg d      }t        j                  |      }t	        |j
                  j                  dk(         t	        |j
                  j                  dk(         y )N)g        g333333?g      @)FTFr   r
   r   )r   r   r   
atleast_2dr   r   ndimr,   s      r   test_atleast_2dzTestRegression.test_atleast_2d*   sT    EE5IJMM!q !q !r   c                     t         j                  j                  g dg d      }d|_        t	        |j                  dk(         y )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   s3    EEY?#$r   c                 
   t         j                  j                  t        j                  d      d      }t         j                  j                  dt              }|j                  |       t        |j                  dk(         y )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   sS    EEKK		!4K0uu{{2U{+	$

a r   c                    t         j                  j                  g dd      }t        j                  g d      }t	               5 }|j                  t        d       t         j                  j                  ||d      }t         j                  j                  ||d      }t        |j                  |j                         d d d        y # 1 sw Y   y xY w)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    EE2HH()  	.CJJ)+IJ11-B11-BBGGRWW-	. 	. 	.s   A;CCc                 v   t         j                  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(         y )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    EEr2heU^<$%GGAv$%6!"&'r   c                 n    t         j                  j                  g dd      }t        |g    |d d        y )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   s0    UU:'J2b62bq6*r   c                     t         j                  j                  d      j                  d      }t	        |j                  d      |j                  j                                y )Nr   rX   F)order)r   r   rC   reshaper   tobytesTr]   s     r   !test_masked_array_tobytes_fortranz0TestRegression.test_masked_array_tobytes_fortranY   s=    UU\\!_$$U+2::C:0"$$,,.Ar   c                     t         j                  j                  ddt         j                  fddt         j                  fdt         j                  fgfg       y )N)r
   )r   r   r   rP   ijr@   )r   r   r   int_voidr   s    r   test_structured_arrayz$TestRegression.test_structured_array]   sH    
O>#rww#rww @AC 	 	Dr   N)__name__
__module____qualname__r   r   r)   r-   r1   r5   r9   r=   rH   rT   r[   r^   re   rk    r   r   r   r      sJ    =!0%"%!
.(+
BDr   r   )numpyr   numpy.testingr   r   r   r   r   ro   r   r   <module>rr      s      
ZD ZDr   