K i$HddlZddlZddlmZmZmZmZmZm Z GddZ y)N)_assert_valid_refcountassert_assert_array_almost_equalassert_array_equal assert_equal assert_raisesceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZy)TestRegressioncttjdgtjddgz tjddgy)Nr)rnppoly1dselfs e/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/numpy/lib/tests/test_regression.py test_poly1dzTestRegression.test_poly1ds8RYYs^biiA&77YYAw' )ctjjd}|j}tj|dtj|dt ||y)N)rT)rowvarF)rrandomcopycovrrxys rtest_cov_parametersz"TestRegression.test_cov_parameterssG II  V $ FFH q q1a rctdD]6}tjgdddgtjgdddg8y)Nd)r rr r)rr r!rr")rangerdigitize)ris rtest_mem_digitizez TestRegression.test_mem_digitizes9s 1A KK q!f - KK!Q 0 1rchtgtjtjgyN)rruniquearrayrs rtest_unique_zero_sizedz%TestRegression.test_unique_zero_sized"s2ryy"67rcdtjd}|tjdtjdtjd|tjdtjdtjdtjdy)Nc|Sr()argss rz3TestRegression.test_mem_vectorise..(sr)r r!r )r!r r )r r r!r!r!)r vectorizezeros)rvts rtest_mem_vectorisez!TestRegression.test_mem_vectorise&sr \\, - 288I  3RXXi5HI 288I  3RXX?688F# %rcttjddddgttjddgy)Nry?)rrmgridrs rtest_mgrid_single_elementz(TestRegression.test_mgrid_single_element-s1288AaF+aS1288Aa="-rcJd}tj|}t|y)Ncy)N{r.)rrs rpz1TestRegression.test_refcount_vectorize..p4sr)rr2r)rr<vs rtest_refcount_vectorizez&TestRegression.test_refcount_vectorize2s  LLOq!rctjtjtjdgd}ttjj t |dy)Nr F)rr@)rrnanrlinalg LinAlgErrorgetattr)rr<s rtest_poly1d_nan_rootsz$TestRegression.test_poly1d_nan_roots9s9 IIrvvrvvq)U 3bii++Wa=rc2tjgdgy)Ng?)rpolymulrs rtest_mem_polymulzTestRegression.test_mem_polymul>s 2trcZtjg}tj|dy)Nz asdasd asdasd)rr*append)rrs rtest_mem_string_concatz%TestRegression.test_mem_string_concatBs HHRL !%&rctjgd}tjgd}tj||\}}t||z|z|y)Nr r!r)r r!rr")rrpolydivr)rur=qr@s r test_poly_divzTestRegression.test_poly_divGsD IIi  IIo &zz!Q1QUQY"rctjgd}tjddg}t||k7t||k(y)NrMrr")rrrrs r test_poly_eqzTestRegression.test_poly_eqNs8 IIi  IIq!f QQrcbgd}gd}gd}tj||d}t||y)N)g1$ͱgH PB?g{p#gpI#@g K.dpsRAq!A !!!QW!5A'(AbffQ1a# $rcd}tj|fdz}tjjd|}|tj|||||yrrrs rdp2z5TestRegression.test_large_fancy_indexing..dp2sPAq!A !!!QW!5A bffQ1a# $r)rrintpitemsizerrr)rnbitsrrrs @rtest_large_fancy_indexingz(TestRegression.test_large_fancy_indexingsX!**Q.q%x9-12 )  % j"%j#&rctjddg}tjd|}ttj||fj|k(y)N)rf4)bi4)r )rrr3rr_)rdtrs rtest_void_coercionz!TestRegression.test_void_coercionsD XX{K0 1 HHT2 ad !!R'(rctjg}|D])}tt|tt|dk7+y)N)r get_includer isinstancestr)r include_dirspaths rtest_include_dirsz TestRegression.test_include_dirss=()   D JtS) * DBJ  rc tttjtjdgdtjtttjdgdtj tttjtjdgdtjtttjdgdtj y)Nr r)rrrpolyderrndarrayrs rtest_polyder_return_typez'TestRegression.test_polyder_return_types 2::biina8"))DE 2::qc1-rzz:; 2::biina8"))DE 2::qc1-rzz:;rcfddlm}tjgdtj}gd}tj dj tj}tjtjtjg} |||||y#t$rtwxYw)Nr) append_fieldsrMr)rrcr) numpy.lib.recfunctionsrrr*int32eyeastypefloat64 ExceptionAssertionError)rrbasenamesdatadlists rtest_append_fields_dtype_listz,TestRegression.test_append_fields_dtype_listsz8xx 2vvay)RXXrxx0 ! $tU 3 !  !s  B B0c.ddlm}ddg}tj|d|}t |tj dg|dd d gfg}tj|d|}t |tj d g|d g}tj|d|}t |tj d g|dg}tj|d|}dg}t |tj ||y)Nr)StringIO)ru1r!)rrr!z0 1 2 3rrr r!rr)rr)r r)rr)))rr r!r)rrr1)r)rr)r!rr!z0 1 2 3 4 5 6 7 8 9 10 11)))rr)r"rN)))r )r )iorrloadtxtrr*)rrrrrs rtest_loadtxt_fields_subarraysz,TestRegression.test_loadtxt_fields_subarrayssn - JJx *" 5Q"2!32>?(+67 8 JJx *" 5Q"3!4B?@! " JJx *" 5Q"5!6bAB$ % JJx ;.Cz3Old-style class in python2, normal class in python3N)__name__ __module__ __qualname____doc__r.rrCrs E rrw)output)openosdevnullrinfoAttributeErrorrclose)rrouts rtest_py3_compatzTestRegression.test_py3_compatsa  2::s#  GGAC $ IIK !  ! IIKsAA$$A''A9N)rrrrrr&r+r5r8r>rErHrKrRrTrrrrrrrrrrr rr!r.rrr r s) !1 8%. "> ' #/* 'P3- '()  < !2*! rr ) rnumpyr numpy.testingrrrrrrr r.rrr$s! YYr