o 6aP @sJddlZddlZddlZddlmZmZmZmZm Z m 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/d0Zd1d2Zd3d4Zd5d6Zd7S)8TestRegressioncCs.ttdgtddgtddgdS)Nr)rnppoly1dselfrA/usr/lib/python3/dist-packages/numpy/lib/tests/test_regression.py test_poly1d s zTestRegression.test_poly1dcCs>tjd}|}tj|ddtj|ddt||dS)N)rT)rowvarF)r randomcopycovrrxyrrrtest_cov_parameterss z"TestRegression.test_cov_parameterscCs:tdD]}tgdddgtgdddgqdS)Nd)r rr r)rr rrr)ranger digitize)rirrrtest_mem_digitizes z TestRegression.test_mem_digitizecCstgttgdSN)rr uniquearrayr rrrtest_unique_zero_sized sz%TestRegression.test_unique_zero_sizedcCsVtdd}|tdtdtd|tdtdtdtddS)NcWs|Sr"r)argsrrr&sz3TestRegression.test_mem_vectorise..)r rr )rr r )r r rrr)r vectorizezeros)rvtrrrtest_mem_vectorise$s z!TestRegression.test_mem_vectorisecCs0ttjddddgttjddgdS)Nry?)rr mgridr rrrtest_mgrid_single_element+sz(TestRegression.test_mgrid_single_elementcCsdd}t|}t|dS)NcSsdS)N{r)rrrrrp2sz1TestRegression.test_refcount_vectorize..p)r r)r)rr0vrrrtest_refcount_vectorize0s  z&TestRegression.test_refcount_vectorizecCs.tjtjtjdgdd}ttjjt|ddS)Nr F)rr3)r r nanrlinalg LinAlgErrorgetattr)rr0rrrtest_poly1d_nan_roots7sz$TestRegression.test_poly1d_nan_rootscCstgdgdS)Ng?)r polymulr 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)rur1qr3rrr test_poly_divEszTestRegression.test_poly_divcCs8tgd}tddg}t||kt||kdS)Nr=rr)r r rrrrr test_poly_eqLs zTestRegression.test_poly_eqcCs4gd}gd}gd}t||d}t||dS)N)g1$ͱgH PB?g{p#gpI#@g Kr)sizer rrrandintix_nar thesizerrdpz4TestRegression.test_large_fancy_indexing..dpcs@d}t|fd}tjjd|d}|t|||||dSrrrrrrdp2rz5TestRegression.test_large_fancy_indexing..dp2)r rintpitemsizerrr)rnbitsrrrrrtest_large_fancy_indexings    z(TestRegression.test_large_fancy_indexingcCs6tddg}td|}ttj||fj|kdS)N)rf4)bi4)r )r rr*rr_)rdtrrrrtest_void_coercions z!TestRegression.test_void_coercioncCs~ddl}ddl}|j}t|jd|_z#z tdtdiWn ty*t dwW|j ||_dS|j ||_w)Nrwfoor z ticket #1243) ossysstdoutopendevnullr whor$ ExceptionAssertionErrorclose)rrr oldstdoutrrrtest_who_with_0dim_arrays    z'TestRegression.test_who_with_0dim_arraycCs2tg}|D]}tt|tt|dkqdS)N)r get_includer isinstancestr)r include_dirspathrrrtest_include_dirss z TestRegression.test_include_dirscCsxttttdgdtjtttdgdtjttttdgdtjtttdgdtjdS)Nr r)rrr polyderr ndarrayr rrrtest_polyder_return_types  z'TestRegression.test_polyder_return_typecCstddlm}tjgdtjd}gd}tdtj}tjtjtjg}z |||||WdSty9t w)Nr) append_fieldsr=r)rrcr) numpy.lib.recfunctionsrr r$int32eyeastypefloat64rr)rrbasenamesdatadlistrrrtest_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)rr>)))r )r )iorr loadtxtrr$)rrrrrrrrtest_loadtxt_fields_subarrayss z,TestRegression.test_loadtxt_fields_subarrayscCs4tjdtd}zt|WdStytw)Nrr)r r*boolnansumrr)rrrrrtest_nansum_with_booleans  z'TestRegression.test_nansum_with_booleancCs^Gddd}ttjd}zz tj||dWn ty"twW|dS|w)Nc@seZdZdZdS)z)TestRegression.test_py3_compat..Cz3Old-style class in python2, normal class in python3N)__name__ __module__ __qualname____doc__rrrrCsrr)output)rrrr infoAttributeErrorrr)rroutrrrtest_py3_compats  zTestRegression.test_py3_compatN)rrrrrr!r%r,r.r2r8r:r<rBrCrrrrrrrrrrrrrrr rrrrr s8     r) pytestrnumpyr numpy.testingrrrrrrrrrrrs