Hyp @ d dl Zd dlmZmZmZmZ G d d ZdS ) N)assert_assert_array_equalassert_allclosesuppress_warningsc b e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd ZdS )TestRegressionc t j g dg d }t t j | g dg d S )N)r r )r r r r
r
r
r r mask)r
r r )npma masked_arrayr nonzero)selfxs r/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/ma/tests/test_regression.pytest_masked_array_createz'TestRegression.test_masked_array_create sZ E 7 7 7$<$<$<
>
>25==++lll^<<<<< c J t j ddg d S )Nr
r )r r arrayr s r test_masked_arrayz TestRegression.test_masked_array s"
AQC r c ddl m}m} t j d }t j |j | } ||| }||z
d S )Nr ) masked_whereMaskType)r
r
)numpy.mar! r" r zerosshape)r r! r" abcs r test_mem_masked_wherez$TestRegression.test_mem_masked_where sZ 33333333 HV HQWh ' ' LA !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$ maskedr r& r' s r test_masked_array_multiplyz)TestRegression.test_masked_array_multiply sG EKK %,$ HV ! !r c p t j dgd 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 )NUnicode)reprr r r r s r test_masked_array_repr_unicodez-TestRegression.test_masked_array_repr_unicode& s$ RU[[
#
#$$$$$r c t j g dg d }t j | }t |j j dk t |j j dk 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 dg d }d|_ t |j dk 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 dt }| | t
|j dk d S )Nr Tr dtype)outr )r r r arangefloatvarr _data)r r& mouts r test_var_sets_maskedarray_scalarz/TestRegression.test_var_sets_maskedarray_scalar7 sc EKK ! 4K 0 0u{{2U{++ $
a r c t j g dd }t j g d }t 5 }| t d t j ||d }t j ||d }t |j |j d d d d S # 1 swxY 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 filterDeprecationWarningcorrcoefr data)r r ysupr0r1s 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:CCCc X t j ddgddg }t |j j dk t j |d }t |j j dk t |j dk t |j j dk 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 dd }t |g |d d d S )N))r
rV )r rW )r g @zi4,f4r@ r )r r rY r r r s r test_empty_list_on_structuredz,TestRegression.test_empty_list_on_structuredT sE
U
: : :'
J
J2b62bqb6*****r c t j d d }t | d |j d S )Nr
rX F)order)r r rC reshaper tobytesTr] s r !test_masked_array_tobytes_fortranz0TestRegression.test_masked_array_tobytes_fortranY sO
U\\!__
$
$U
+
+2::C:00"$,,..AAAAAr c t j ddt j fddt j fdt j fgfg d S )N)r
)r r r rP ijr@ )r r r int_voidr s r test_structured_arrayz$TestRegression.test_structured_array] sX
O > #rw #rw @AC D D D D Dr 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 BD D D D Dr r )numpyr
numpy.testingr r r r r ro r r rr s
ZD ZD ZD ZD ZD ZD ZD ZD ZD ZDr