a :jg$@sJddlZddlZddlmZmZmZmZmZm Z ddl Z GdddZ dS)N)assert_ assert_equalassert_array_equalassert_array_almost_equal assert_raises_assert_valid_refcountc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3S)4TestRegressioncCs.ttdgtddgtddgdS)Nr)rnppoly1dselfrI/usr/local/lib/python3.9/site-packages/numpy/lib/tests/test_regression.py test_poly1d s zTestRegression.test_poly1dcCs>tjd}|}tj|ddtj|ddt||dS)N)rT)ZrowvarF)r randomcopyZcovrrxyrrrtest_cov_parameterss  z"TestRegression.test_cov_parameterscCs:tdD],}tgdddgtgdddgqdS)Nd)r rr r)rr rrr)ranger Zdigitize)rirrrtest_mem_digitizes z TestRegression.test_mem_digitizecCstgttgdSN)rr uniquearrayr rrrtest_unique_zero_sizedsz%TestRegression.test_unique_zero_sizedcCsVtdd}|tdtdtd|tdtdtdtddS)NcWs|Srr)argsrrr%z3TestRegression.test_mem_vectorise..)r rr )rr r )r r rrr)r vectorizezeros)rvtrrrtest_mem_vectorise#s z!TestRegression.test_mem_vectorisecCs0ttjddddgttjddgdS)Nry?)rr Zmgridr rrrtest_mgrid_single_element*sz(TestRegression.test_mgrid_single_elementcCsdd}t|}t|dS)NcSsdS)N{r)rrrrrp1sz1TestRegression.test_refcount_vectorize..p)r r'r)rr-vrrrtest_refcount_vectorize/s z&TestRegression.test_refcount_vectorizecCs.tjtjtjdgdd}ttjjt|ddS)Nr F)rr0)r r nanrZlinalgZ LinAlgErrorgetattr)rr-rrrtest_poly1d_nan_roots6sz$TestRegression.test_poly1d_nan_rootscCstgdgdS)Ng?)r Zpolymulr rrrtest_mem_polymul;szTestRegression.test_mem_polymulcCstg}t|ddS)Nz asdasd asdasd)r r!append)rrrrrtest_mem_string_concat?s z%TestRegression.test_mem_string_concatcCsBtgd}tgd}t||\}}t||||dS)Nr rr)r rrr)r r polydivr)rur.qr0rrr test_poly_divDszTestRegression.test_poly_divcCs8tgd}tddg}t||kt||kdS)Nr7rr)r r rrrrr test_poly_eqKs zTestRegression.test_poly_eqcCs4gd}gd}gd}t||d}t||dS)N)g1$ͱgH PB?g{p#gpI#@g K.dpcs@d}t|fd}tjjd|d}|t|||||dSrrrrrrdp2sz5TestRegression.test_large_fancy_indexing..dp2)r rZintpitemsizerrr)rnbitsrrrrrtest_large_fancy_indexings    z(TestRegression.test_large_fancy_indexingcCs6tddg}td|}ttj||fj|kdS)N)rZf4)bi4)r )r rr(rZr_)rdtrrrrtest_void_coercions z!TestRegression.test_void_coercioncCs2tg}|D]}tt|tt|dkqdS)N)r Z get_includer isinstancestr)r include_dirspathrrrtest_include_dirss z TestRegression.test_include_dirscCsxttttdgdtjtttdgdtjttttdgdtjtttdgdtjdS)Nr r)rrr Zpolyderr Zndarrayr rrrtest_polyder_return_types  z'TestRegression.test_polyder_return_typecCszddlm}tjgdtjd}gd}tdtj}tjtjtjg}z|||||Wntytt Yn0dS)Nr) append_fieldsr7r)rrcr) Znumpy.lib.recfunctionsrr r!Zint32eyeZastypeZfloat64 ExceptionAssertionError)rrbasenamesdataZdlistrrrtest_append_fields_dtype_lists  z,TestRegression.test_append_fields_dtype_listcCsddlm}ddg}tj|d|d}t|tjdg|ddd d gfg}tj|d|d}t|tjd g|dd g}tj|d|d}t|tjd g|ddg}tj|d|d}dg}t|tj||ddS)Nr)StringIO)ru1r)rrrz0 1 2 3rrr rrr)rr)r r)rr)))rr rr)rrr&)r)rr)rrrz0 1 2 3 4 5 6 7 8 9 10 11)))rr)rr8)))r )r )iorr Zloadtxtrr!)rrrrrrrrtest_loadtxt_fields_subarrayss z,TestRegression.test_loadtxt_fields_subarrayscCs:tjdtd}zt|Wnty4tYn0dS)Nrr)r r(boolZnansumrr)rrrrrtest_nansum_with_booleans  z'TestRegression.test_nansum_with_booleancCsdGddd}ttjd}z:ztj||dWntyHtYn0W|n |0dS)Nc@seZdZdZdS)z)TestRegression.test_py3_compat..Cz3Old-style class in python2, normal class in python3N)__name__ __module__ __qualname____doc__rrrrCsrw)output)openosdevnullr infoAttributeErrorrclose)rroutrrrtest_py3_compats  zTestRegression.test_py3_compatN)rrrrrrr"r*r+r/r3r4r6r<r=rrrrrrrrrrrrrrrrrr s2    r) rnumpyr Z numpy.testingrrrrrrZpytestrrrrrs