id @ddlZddlmZmZmZmZGddZdS)N)assert_assert_array_equalassert_allclosesuppress_warningscbeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdS)TestRegressionctjgdgd}ttj|gdgdS)N)rr)rrrr r r rrmask)r r r)npma masked_arrayrnonzero)selfxs S/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/ma/tests/test_regression.pytest_masked_array_createz'TestRegression.test_masked_array_createsZ E  777$<$<$<  > >25==++lll^<<<<<cJtjddgdS)Nr r)rrarrayrs rtest_masked_arrayz TestRegression.test_masked_arrays"  AQC      rcddlm}m}tjd}tj|j|}|||}||z dS)Nr) masked_whereMaskType)r r )numpy.mar!r"rzerosshape)rr!r"abcs rtest_mem_masked_wherez$TestRegression.test_mem_masked_wheresZ33333333 HV   HQWh ' ' LA   !rctjd}tjj|d<tjd}||z||zdS)N)r r )r r)r r )rrr$maskedrr&r's rtest_masked_array_multiplyz)TestRegression.test_masked_array_multiplysG EKK  %,$ HV   ! !rcptjdgdddS)Nr Fr )rrrrepeatrs rtest_masked_array_repeatz'TestRegression.test_masked_array_repeat"s0  QCe $$++B/////rc^ttjddS)NUnicode)reprrrrrs rtest_masked_array_repr_unicodez-TestRegression.test_masked_array_repr_unicode&s$ RU[[ # #$$$$$rctjgdgd}tj|}t |jjdkt |jjdkdS)N)gg333333?g @)FTFrr r )rrr atleast_2drrndimr,s rtest_atleast_2dzTestRegression.test_atleast_2d*sj E  5I5I5I  J J M!   q !!! q !!!!!rctjgdgd}d|_t |jdkdS)N)r&r'r()r rrrX)rrr fill_valuer)rr&s rtest_set_fill_value_unicode_py3z.TestRegression.test_set_fill_value_unicode_py31sF E  YYY  ? ?  #$$$$$rctjtjdd}tjdt}||t |jdkdS)NrTrdtype)outr)rrrarangefloatvarr_data)rr&mouts r test_var_sets_maskedarray_scalarz/TestRegression.test_var_sets_maskedarray_scalar7sc EKK ! 4K 0 0u{{2U{++ $ a     rctjgdd}tjgd}t 5}|t dtj||d}tj||d}t|j |j ddddS#1swxYwYdS)N)r r r r rr )r g@g@r rzbias and ddof have no effectr)ddofr ) rr masked_equalrrfilterDeprecationWarningcorrcoefrdata)rrysupr0r1s rtest_ddof_corrcoefz!TestRegression.test_ddof_corrcoef>s E   2 2 H((( ) )   .C JJ)+I J J J11--B11--B BGRW - - -  . . . . . . . . . . . . . . . . . .sA:CCCcXtjddgddg}t|jjdktj|d}t|jjdkt|jdkt|jjdkdS)N?@Fr)r )r r r r )rr MaskedArrayrrr%tiler,s rtest_mask_not_backmangledz(TestRegression.test_mask_not_backmangledJs E  r2heU^  < < $%%% GAv   $%%%6!""" &'''''rctjgdd}t|g|dddS)N))r rV)r rW)r g@zi4,f4r@r)rrrYrrrs rtest_empty_list_on_structuredz,TestRegression.test_empty_list_on_structuredTsE U  :::'  J J2b62bqb6*****rctjdd}t |d|jdS)Nr rXF)order)rrrCreshapertobytesTr]s r!test_masked_array_tobytes_fortranz0TestRegression.test_masked_array_tobytes_fortranYsO U\\!__ $ $U + +2::C:00"$,,..AAAAArctjddtjfddtjfdtjfgfgdS)N)r )rrrrPijr@)rrrint_voidrs rtest_structured_arrayz$TestRegression.test_structured_array]sX  O>#rw#rw @AC  D D D D DrN)__name__ __module__ __qualname__rrr)r-r1r5r9r=rHrTr[r^rerkrrrrs=== !!!      000%%%"""%%% !!! . . .(((+++ BBBDDDDDrr)numpyr numpy.testingrrrrrrorrrrs ZDZDZDZDZDZDZDZDZDZDr