B   ›\k<ã@sÔdZddlmZmZmZddlZddlZddlZddlZddl m Z ddl Z ddl mZddl mZmZmZmZmZmZmZddl mZmZdd lmZmZmZdd lmZe ej d ¡Z!d d „Z"dd„Z#dd„Z$dZ%dZ&dZ'dd„Z(dZ)dd„Z*e*eƒZ+e*eƒZ,dd„Z-e-eƒZ.e-eƒZ/e 0e 1ej2¡e 1ej2¡ej3ej3¡Z4e 0e 1ej5¡e 1ej5¡ej3ej3¡Z6e 0ej7ej7ej3ej3¡Z8Gdd„deƒZ9Gdd„deƒZ:Gd d!„d!eƒZ;e!Gd"d#„d#eƒƒZ¡dS)%z Tests for @cfunc and friends. é)ÚdivisionÚprint_functionÚabsolute_importN)Ú namedtuple)Úunittest_support)ÚcfuncÚcarrayÚfarrayÚtypesÚtypingÚutilsÚnjit)Ú cffi_supportÚ numpy_supporté)ÚTestCaseÚtagÚcaptured_stderr)Ú BaseCacheTestz4CFFI not supported -- please install the cffi modulecCs||S)N©)ÚaÚbrrú5lib/python3.7/site-packages/numba/tests/test_cfunc.pyÚ add_usecasesrcCs ||}|S)Nr)rrÚcrrrÚ div_usecasesrcCs|dS)Nér)rrrrÚsquare_usecase"srzfloat64(float64, float64)zfloat64(int64, int64)zfloat64(float64)cCstƒ||S)N)Úobject)rrrrrÚobjmode_usecase+srécs‡fdd„}|S)Nc s˜ˆ|||fƒ}ˆ|tƒ}|j|d<|j|dd…<|j|dd…<|jj|d<|jj|d<d}x0t ||¡D] \}}||||f||7}qhW||d<dS)Nrréééé) ÚCARRAY_USECASE_OUT_LENÚndimÚshapeÚstridesÚflagsÚ c_contiguousÚ f_contiguousÚnpÚndindex) Úin_ptrÚout_ptrÚmÚnÚin_ÚoutÚsÚiÚj)ÚfuncrrÚcfarray_usecase5s    z-make_cfarray_usecase..cfarray_usecaser)r7r8r)r7rÚmake_cfarray_usecase3s r9cs‡fdd„}|S)Nc s¢ˆ|||ftjd}ˆ|ttjƒ}|j|d<|j|dd…<|j|dd…<|jj|d<|jj|d<d}x0t  ||¡D] \}}||||f||7}qrW||d<dS)N)Údtyperrr!r"r#r$) r,Úfloat32r%r&r'r(r)r*r+r-) r.r/r0r1r2r3r4r5r6)r7rrr8Ms   z3make_cfarray_dtype_usecase..cfarray_usecaser)r7r8r)r7rÚmake_cfarray_dtype_usecaseJs r<c@sXeZdZedƒdd„ƒZedƒedd„ƒƒZdd„Zedƒdd „ƒZd d „Z d d „Z dS)Ú TestCFuncÚ importantcCsœttƒtƒ}| |jd¡| |jd¡| |jt¡|j}|  |t ¡|  d|¡|j }|  |t j¡|j}| t |tj¡j|¡| |ddƒd¡dS)z8 Basic usage and properties of a cfunc. rg@g @g@N)rÚadd_sigrÚ assertEqualÚ__name__Ú __qualname__ZassertIsÚ __wrapped__Ú native_nameÚassertIsInstanceÚstrÚassertInZaddressr Z INT_TYPESÚctypesÚcastÚc_void_pÚvalueÚassertPreciseEqual)ÚselfÚfZsymbolÚaddrZctrrrÚ test_basicos   zTestCFunc.test_basiccCs@ddlm}| ¡\}}ttƒtƒ}| |j¡}| |d¡dS)Nr)Ú cffi_usecasesg@) ÚrQZload_inline_modulerÚ square_sigrZ_numba_test_funcptrÚcffirL)rMrQÚffiÚlibrNÚresrrrÚ test_cffi†s     zTestCFunc.test_cfficCs.ttdtjidtƒ}| | dd¡d¡dS)Nr)Úlocalsr r!g@)rÚdiv_sigr Zint64rrLrH)rMrNrrrÚ test_locals‘szTestCFunc.test_localsc Cs¨ttƒtƒ}tƒ}| | dd¡d¡WdQRX| | ¡d¡tƒ}| dd¡}| |d¡WdQRX| ¡}| d|¡t j dkr˜| d |¡n | d |¡dS) Nr"rg@rRrgz(ZeroDivisionError: ('division by zero',))r!zException ignoredz ignored) rrZrrrLrHr@ÚgetvaluerGÚsysÚ version_info)rMrNÚerrrWrrrÚ test_errors—s    zTestCFunc.test_errorscCs2ttƒtƒ}| ¡}| |j|¡| d|¡dS)Nz fadd double)rr?rZ inspect_llvmrGrD)rMrNZirrrrÚ test_llvm_ir«s zTestCFunc.test_llvm_irc Cs\| t¡ttddtƒWdQRX| ¡}ttƒtƒWdQRX| dt|j ƒ¡dS)z7 Object mode is currently unsupported. T)ZforceobjNzUntyped global name 'object') Ú assertRaisesÚNotImplementedErrorrr?rÚassertTypingErrorrrGrFÚ exception)rMÚraisesrrrÚtest_object_mode±s   zTestCFunc.test_object_modeN) rAÚ __module__rBrrPÚskip_cffi_unsupportedrXr[r`rargrrrrr=ms r=c@sJeZdZej e¡Zej ed¡Z dZ dd„Z dd„Z e dƒdd „ƒZd S) ÚTestCFuncCachezcfunc_cache_usecases.pyZcfunc_caching_test_foddercCs`dt|j|jd}tjtjd|gtjtjd}| ¡\}}|j dkr\t d|j |  ¡fƒ‚dS)Nafif 1: import sys sys.path.insert(0, %(tempdir)r) mod = __import__(%(modname)r) mod.self_test() f = mod.add_usecase assert f.cache_hits == 1 f = mod.outer assert f.cache_hits == 1 f = mod.div_usecase assert f.cache_hits == 1 )ÚtempdirÚmodnamez-c)ÚstdoutÚstderrrz/process failed with code %s: stderr follows %s ) ÚdictrkrlÚ subprocessÚPopenr]Ú executableÚPIPEZ communicateÚ returncodeÚAssertionErrorÚdecode)rMÚcodeÚpopenr3r_rrrÚrun_in_separate_processÂs  z&TestCFuncCache.run_in_separate_processcCs | ¡dS)N)Z self_test)rMÚmodrrrÚ check_moduleÚszTestCFuncCache.check_moduler>cCsÎ| d¡| ¡}| d¡| |jjd¡| |jjd¡| |jjd¡| |jjd¡| |¡| ¡}| d¡| |jjd¡| |jjd¡| |jjd¡| |jjd¡| |¡|  ¡dS)Nrr#r) Z check_pycacheÚ import_moduler@rZ cache_hitsZouterZadd_nocache_usecaserr{ry)rMrzrrrÚ test_cachingÝs      zTestCFuncCache.test_cachingN)rArhrBÚosÚpathÚdirnameÚ__file__ÚhereÚjoinZ usecases_filerlryr{rr}rrrrrj¼s  rjc@s€eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z e dƒdd„ƒZ dd„Z dd„Z dd„Ze dƒdd„ƒZdd„ZdS)Ú TestCArrayz* Tests for carray() and farray(). cCsHt dd¡ d¡ tj¡}tjttjd}|||ƒ||ƒf|jžŽ|S)Né é)rr!)r:)r,ÚarangeÚreshapeÚastyper;Úemptyr%r')rMÚpointer_factoryr7rr3rrrÚrun_carray_usecaseûszTestCArray.run_carray_usecasecCs(| ||¡}| ||¡}| ||¡dS)N)rŒrL)rMr‹ÚpyfuncrÚexpectedÚgotrrrÚcheck_carray_usecases  zTestCArray.check_carray_usecasecCs|j tj¡S)N)rHÚdata_asrJ)rMÚarrrrrÚ make_voidptrszTestCArray.make_voidptrcCs|j t tj¡¡S)N)rHr‘ÚPOINTERZc_float)rMr’rrrÚmake_float32_pointer szTestCArray.make_float32_pointercCs|j t tj¡¡S)N)rHr‘r”Zc_double)rMr’rrrÚmake_float64_pointer szTestCArray.make_float64_pointerc s‚‡fdd„}t d¡ d¡ tj¡j|d}|ˆ |¡|jƒ}|||ƒ|ˆ |¡|jƒ}|||  d¡ƒ|ˆ |¡|j|j ƒ}|||ƒ|ˆ |¡|jtjƒ}|||ƒ|ˆ  |¡|j|j ƒ}|||ƒ|ˆ  |¡|jtj ƒ}|||  tj ¡ƒˆ t¡|ˆ  |¡|jƒWdQRXˆ t¡||jj|jƒWdQRXˆ t¡}|ˆ |¡|jtj ƒWdQRXˆ dt|jƒ¡dS)Ncs$ˆ ||¡ˆ |jj|jj¡dS)N)rLr@rHÚdata)rrŽ)rMrrÚeqs z*TestCArray.check_carray_farray..eqr#)rr!)ÚorderÚKz%mismatching dtype 'int32' for pointer)r,r‡rˆr‰r;Úcopyr•r'ÚsizeZravelr:r“Úint32ZviewrbÚ TypeErrorrHr—rGrFre)rMr7r™r˜Úbaserrfr)rMrÚcheck_carray_farrays,          zTestCArray.check_carray_farrayr>cCs| td¡dS)z, Test pure Python carray(). ÚCN)r r)rMrrrÚ test_carray7szTestCArray.test_carraycCs| td¡dS)z, Test pure Python farray(). ÚFN)r r )rMrrrÚ test_farray>szTestCArray.test_farrayc#sNxHtjtjtjtjtjtjfD](‰t‡fdd„|jDƒƒ}|j |ŽVqWdS)zŠ Generate a bunch of concrete signatures by varying the width and signedness of size arguments (see issue #1923). c3s |]}|tjkrˆn|VqdS)N)r Úintp)Ú.0r)Ú actual_sizerrú Ksz.TestCArray.make_carray_sigs..N) r r¥rZintcÚuintpZuint32ZuintcÚtupleÚargsÚ return_type)rMZ formal_sigr«r)r§rÚmake_carray_sigsDs   zTestCArray.make_carray_sigsc CsØ|}x0| t¡D]"}t|ƒ|ƒ}| |j||j¡qW|}x0| t¡D]"}t|ƒ|ƒ}| |j||j¡qFW| ¡}ttƒ|ƒ}WdQRX| dt |j ƒ¡|}x0| t ¡D]"}t|ƒ|ƒ}| |j||j¡q®WdS)Nz7mismatching dtype 'float32' for pointer type 'float64*') r­Úcarray_float32_usecase_sigrrr•rHrdÚcarray_float64_usecase_sigrGrFreÚcarray_voidptr_usecase_sig)rMZusecaseZ dtype_usecaserÚsigrNrfrrrÚcheck_numba_carray_farrayOs      z$TestCArray.check_numba_carray_farraycCs| tt¡dS)zK Test Numba-compiled carray() against pure Python carray() N)r²Úcarray_usecaseÚcarray_dtype_usecase)rMrrrÚtest_numba_carraygszTestCArray.test_numba_carraycCs| tt¡dS)zK Test Numba-compiled farray() against pure Python farray() N)r²Úfarray_usecaseÚfarray_dtype_usecase)rMrrrÚtest_numba_farraynszTestCArray.test_numba_farrayN)rArhrBÚ__doc__rŒrr“r•r–r rr¢r¤r­r²rµr¸rrrrr„ös( r„c@s4eZdZdZefdd„Zdd„Zdd„Zdd „Zd S) ÚTestCffiStructzÎ typedef struct _big_struct { int i1; float f2; double d3; float af4[9]; } big_struct; typedef struct _error { int bits:4; } error; typedef double (*myfunc)(big_struct*, size_t); cCs ddlm}|ƒ}| |¡|S)Nr)ÚFFI)rTr»Zcdef)rMÚsrcr»rUrrrÚget_ffi†s  zTestCffiStruct.get_fficCsþ| ¡}| d¡}tj|dd}| |tj¡| t|ƒd¡| | d¡tj ¡| | d¡tj ¡| | d¡tj ¡| | d¡tj tj d d ¡| d ¡}tj|dd}| |t j¡| |jd t |¡¡| |jd tj¡| |jtj ¡dS)NÚ big_structT)Úuse_record_dtypeéÚi1Úf2Úd3Úaf4)é )r:r'Úmyfuncrr)r½ÚtypeofrÚmap_typerEr ZRecordr@Úlenrr;Úfloat64Z NestedArrayr Z Signaturer«ÚCPointerr©r¬)rMrUr¾ZnbtyperÆr±rrrÚtest_type_parsings"  z TestCffiStruct.test_type_parsingcs"| ¡}| d¡}tj|dd}tj| d¡dd}tdd„ƒ‰t|ƒ‡fdd„ƒ}| d ¡}| d |¡}xftd ƒD]Z}|d ||_ |d ||_ d|d ||_ x&tdƒD]} |d| ||j | <q°WqtWt | d|¡ƒ} | | d ¡} tj| |¡t |¡d d} ˆ| ƒ} | | | ¡dS)Nr¾T)r¿rÆcSsLd}xBt|jƒD]4}||}||j|j|j7}|||j ¡7}qW|S)Nr)ÚrangerœrÁrÂrÃrÄÚsum)rŸZtmpr5ÚelemrrrÚcalcªs z0TestCffiStruct.test_cfunc_callback..calccst||ƒ}ˆ|ƒS)N)r)Úptrr1rŸ)rÐrrÚfoo³s z/TestCffiStruct.test_cfunc_callback..fooz big_struct[3]z big_struct*r!é{éÕrrÅr…Zsize_t)Úbufferr:r')r½rÇrrÈr rÚnewrIrÍrÁrÂrÃrÄÚintrHr,ZndarrayrÕrZas_dtyper@)rMrUr¾Z nb_big_structr±rÒZmydatarÑr5r6rOrZarrayZexpectr)rÐrÚtest_cfunc_callback¤s,     z"TestCffiStruct.test_cfunc_callbackc CsH| ¡}| t¡}tj| d¡ddWdQRX| dt|jƒ¡dS)NÚerrorT)r¿z0field 'bits' has bitshift, this is not supported) r½rbÚ ValueErrorrrÈrÇr@rFre)rMrUrfrrrÚtest_unsupport_bitsizeÏs z%TestCffiStruct.test_unsupport_bitsizeN)rArhrBZc_sourcer½rÌrØrÛrrrrrºus  +rºÚ__main__)?r¹Z __future__rrrrHr~rpr]Ú collectionsrZnumpyr,ZnumbarZunittestrrr r r r r rrZsupportrrrZtest_dispatcherrZ skipUnlessZ SUPPORTEDrirrrr?rZrSrr%r9r³r¶r<r´r·ZvoidrËr;r¥r®rÊr¯Zvoidptrr°r=rjr„rºrAÚmainrrrrÚsX  $       O: g