B <ÓFd}"ã@sŽddlmZmZmZddlZddlZddlZddlm Z m Z mZmZm Z mZmZddlmZddlmZdZGdd„de ƒZed krŠe ƒdS) é)ÚdivisionÚabsolute_importÚprint_functionN)Úrun_module_suiteÚTestCaseÚassert_Úassert_equalÚassert_array_equalÚassert_array_almost_equalÚ assert_raises)Ú_assert_valid_refcount)Úunicodeéc@s$eZdZefdd„Zefdd„Zefdd„Zefdd„Zefd d „Zefdd„Z efd d„Z efdd„Zefdd„Zefdd„Z efdd„Zefdd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zefd#d$„Zefd%d&„Zefd'd(„Zefd)d*„Zd+d,„Zd-d.„Zd/d0„Zd1d2„Zd3d4„Zd5d6„Zd7S)8ÚTestRegressioncCs.tt dg¡t ddg¡t ddg¡ƒdS)Nrréÿÿÿÿ)rÚnpÚpoly1d)ÚselfÚlevel©rúR/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test_regression.pyÚtest_poly1dszTestRegression.test_poly1dcCs>tj d¡}| ¡}tj|ddtj|ddt||ƒdS)N)érr)Zrowvarr)rÚrandomÚcopyZcovr )rrÚxÚyrrrÚtest_cov_parameterss z"TestRegression.test_cov_parameterscCsHxBtdƒD]6}t ddddgddg¡t dddddgddg¡q WdS)Nédrérér)ÚrangerZdigitize)rrÚirrrÚtest_mem_digitizesz TestRegression.test_mem_digitizecCstgt t g¡¡ƒdS)N)r rÚuniqueÚarray)rrrrrÚtest_unique_zero_sized%sz%TestRegression.test_unique_zero_sizedcCsVt dd„¡}|t d¡t d¡t d¡ƒ|t d¡t d¡t d¡t d¡ƒdS)NcWs|S)Nr)ÚargsrrrÚ+óz3TestRegression.test_mem_vectorise..)rrr)rrr)rrr)rr)rÚ vectorizeÚzeros)rrZvtrrrÚtest_mem_vectorise)sz!TestRegression.test_mem_vectorisecCs0ttjddd…dgƒttjdd…gƒdS)Nryð?)r rZmgrid)rrrrrÚtest_mgrid_single_element0sz(TestRegression.test_mgrid_single_elementcCsdd„}t |¡}t|ƒdS)NcSsdS)Né{r)rrrrrÚp7sz1TestRegression.test_refcount_vectorize..p)rr*r)rrr/ÚvrrrÚtest_refcount_vectorize5s z&TestRegression.test_refcount_vectorizecCs0tjtjtjdgdd}| tjjt|d¡dS)Nrr)Úrr2)rrÚnanÚassertRaisesZlinalgZLinAlgErrorÚgetattr)rrr/rrrÚtest_poly1d_nan_roots<sz$TestRegression.test_poly1d_nan_rootscCst gdg¡dS)Ngð?)rZpolymul)rrrrrÚtest_mem_polymulAszTestRegression.test_mem_polymulcCst g¡}t |d¡dS)Nz asdasd asdasd)rr%Úappend)rrrrrrÚtest_mem_string_concatEs z%TestRegression.test_mem_string_concatcCsJt dddg¡}t dddddg¡}t ||¡\}}t||||ƒdS)Nrrrr é)rrÚpolydivr)rrÚur0Úqr2rrrÚ test_poly_divJszTestRegression.test_poly_divcCs:t dddg¡}t ddg¡}t||kƒt||kƒdS)Nrrrr )rrr)rrrrrrrÚtest_poly_eqQszTestRegression.test_poly_eqcQCsvdddddg}dddd d ddd ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVgQ}dWdXdYdZdZd[d\d]dWd[d]d^d\d^dWd[d\d_d\d_dYd\d\d\d`dZdWd]dZd^d[d^dYdYd\dadYdYdWdZdad\d\dZdXdWd[d[d\dYdWd_d\d[dZdYdYd_d[d[d\dXd[dZd^dYd[d\d\dZdadWd[d[d[d\dYdZdZd^dYgQ}t ||db¡}t||ƒdS)cNg1$¡½ûÍ±¾g½HPò´B?gÜ{p#½¿gáp“Iè#@g K< l¨sÀéZé[é\é]é^é_é`éaébécréeéfégéhéiéjékéléménéoépéqérésétéuévéwéxéyézr.é|é}é~éé‚éƒé„é…é†é‡éˆé‰éŠé‹éŒéé’é“é”é•é–é—é˜é™éšé›éœééžéŸé é¡é¢é£é¤é¥é¦é§é¨é©éªé«é¬éé®é¯é°g"@g@g@g@g @g@g&@g@g$@g*@g(@r )rZpolyfitr )rÚrefrrZtestedrrrÚtest_polyfit_buildXs$z!TestRegression.test_polyfit_buildcCspd}tjdtjd}t ||¡\}}t|jtjk|ƒd}tjdtjd}t ||¡\}}t|jtjk|ƒdS)NzWrong type, should be complexr)ÚdtypezWrong type, should be float)rÚonesÚcomplexr;rr‘ÚintÚfloat)rÚmsgrr=r2rrrÚtest_polydiv_typemsz TestRegression.test_polydiv_typecCstttjt d¡dddS)N)ré i)Zbins)rÚ ValueErrorrZhistogramddr’)rrrrÚtest_histogramdd_too_many_binsxsz-TestRegression.test_histogramdd_too_many_binscCs\d}tjdtjd}tt |¡jtjk|ƒd}tjdtjd}tt |¡jtjk|ƒdS)NzWrong type, should be complexr)r‘zWrong type, should be float)rr’r“rZpolyintr‘r”r•)rr–rrrrÚtest_polyint_type|sz TestRegression.test_polyint_typecCstt t gg¡¡ƒdS)N)ÚlistrZndenumerater%)rrrrÚtest_ndenumerate_crash…sz%TestRegression.test_ndenumerate_crashcCs tt tjg¡t dg¡ƒdS)N)r rr%r3Zasfarray)rrrrrÚtest_asfarray_noneŠsz!TestRegression.test_asfarray_nonecsZt tj¡jd}td|ddƒ‰‡fdd„}‡fdd„}| t|¡| t|¡dS) Nérgš™™™™™É?rc s@d}t |fd¡}tjjd|ˆd}d|t |||||¡<dS)Nrr:r)Úsize)rr’rÚrandintÚix_)ÚnÚar")ÚthesizerrÚdp“sz4TestRegression.test_large_fancy_indexing..dpcs@d}t |fd¡}tjjd|ˆd}|t |||||¡dS)Nrr:r)r )rr’rr¡r¢)r£r¤r")r¥rrÚdp2™sz5TestRegression.test_large_fancy_indexing..dp2)rr‘ZintpÚitemsizer”r4r™)rrZnbitsr¦r§r)r¥rÚtest_large_fancy_indexingŽsz(TestRegression.test_large_fancy_indexingcCs6t ddg¡}t d|¡}ttj||fj|kƒdS)N)r¤Zf4)ÚbZi4)r)rr‘r+rZr_)rrÚdtrrrrÚtest_void_coercion¢sz!TestRegression.test_void_coercioncCsnddl}ddl}|j}t|jdƒ|_z2yt dt d¡i¡Wntdƒ‚YnXWd|j ¡||_XdS)NrÚwZfoorzticket #1243) ÚosÚsysÚstdoutÚopenÚdevnullrZwhor%ÚAssertionErrorÚclose)rrr®r¯Z oldstdoutrrrÚtest_who_with_0dim_array§s z'TestRegression.test_who_with_0dim_arraycCs:t ¡g}x*|D]"}tt|ttfƒƒt|dkƒqWdS)NÚ)rZget_includerÚ isinstanceÚstrr )rZinclude_dirsÚpathrrrÚtest_include_dirs·s z TestRegression.test_include_dirscCsxttt t dg¡d¡tjƒƒttt dgd¡tjƒƒttt t dg¡d¡tjƒƒttt dgd¡tjƒƒdS)Nrr)rr·rZpolyderrZndarray)rrrrÚtest_polyder_return_typeÀs z'TestRegression.test_polyder_return_typecCsxddlm}tjdddgtjd}ddd g}t d¡ tj¡}tjtjtjg}y|||||ƒWntƒ‚YnXdS) Nr)Ú append_fieldsrrr)r‘r¤rªÚc) Znumpy.lib.recfunctionsr¼rr%Zint32ZeyeZastypeZfloat64r³)rr¼ÚbaseÚnamesÚdataZdlistrrrÚtest_append_fields_dtype_listÇs z,TestRegression.test_append_fields_dtype_listcCsòtjddkrddlm}nddlm}ddg}tj|dƒ|d}t|tjdg|dƒd d dgfg}tj|dƒ|d}t|tjdg|dƒd g}tj|dƒ|d}t|tjdg|dƒdg}tj|dƒ|d}dg}t|tj||dƒdS)Nrr)ÚStringIO)r¤Úu1r)rªrÃrz0 1 2 3)r‘))rr)rrr¤)r¤rÃ)rr)rªrÃ)))rrrr)r¤rÃ)rr)))rr)rr)r¤rÃ)rrrz0 1 2 3 4 5 6 7 8 9 10 11))))rr)rr)r r:))éé)rŸé )r˜é)r¯Úversion_infoÚiorÂrZloadtxtrr%)rrÂr«rrÀrrrÚtest_loadtxt_fields_subarraysÔs z,TestRegression.test_loadtxt_fields_subarrayscCs6tjdtjd}yt |¡Wntƒ‚YnXdS)Nr)r‘)rr+ÚboolZnansumr³)rr¤rrrÚtest_nansum_with_booleanìs z'TestRegression.test_nansum_with_booleancCs^Gdd„dƒ}ttjdƒ}z4ytj|ƒ|dWntk rJtƒ‚YnXWd| ¡XdS)Nc@seZdZdZdS)z)TestRegression.test_py3_compat..Cz3Old-style class in python2, normal class in python3N)Ú__name__Ú __module__Ú__qualname__Ú__doc__rrrrÚC÷srÑr)Úoutput)r±r®r²rÚinfoÚAttributeErrorr³r´)rrÑÚoutrrrÚtest_py3_compatôszTestRegression.test_py3_compatN)rÍrÎrÏÚrlevelrrr#r&r,r-r1r6r7r9r>r?rr—ršr›rržr©r¬rµrºr»rÁrÊrÌrÖrrrrrs6 rÚ__main__)Z __future__rrrr®r¯ZnumpyrZ numpy.testingrrrrr r rZnumpy.testing.utilsrZnumpy.compatr r×rrÍrrrrÚs$t