ó \ K]c@@sVddlmZmZddlZddlZddlZddlZddlZddlZ ddl m Z m Z m Z mZmZddlmZddlmZmZmZddlmZmZddlmZdd lmZdd lmZmZdd lm Z dd l!m"Z"d „Z#d„Z$de%fd„ƒYZ&dej'fgZ(d„Z)d„Z*d„Z+eƒZ,e,j-dee.ƒƒe,j-dƒde fd„ƒYZ/de/fd„ƒYZ0de0fd„ƒYZ1de/fd„ƒYZ2de/fd „ƒYZ3d!e0fd"„ƒYZ4d#e fd$„ƒYZ5e6d%krRej7ƒndS(&i(tprint_functiontabsolute_importNi(tTestCasetoverride_configtoverride_env_configtcaptured_stdouttforbid_codegen(tunittest_support(tjittjitclassttypes(tcompile_isolatedtFlags(tParallelOptions(tNumbaPerformanceWarning(tcompilertprange(tskip_unsupported(t needs_blascC@s|d}|S(Ni((tsomeargtretval((s5lib/python2.7/site-packages/numba/tests/test_debug.pytsimple_nopythons cc@s|V|VdS(N((txty((s5lib/python2.7/site-packages/numba/tests/test_debug.pyt simple_genst SimpleClasscB@seZd„ZRS(cC@s d|_dS(Ni(th(tself((s5lib/python2.7/site-packages/numba/tests/test_debug.pyt__init__ s(t__name__t __module__R(((s5lib/python2.7/site-packages/numba/tests/test_debug.pyRsRcC@s|jS(N(R(tobj((s5lib/python2.7/site-packages/numba/tests/test_debug.pytsimple_class_user%scC@stj||ƒS(N(tnptdot(tatb((s5lib/python2.7/site-packages/numba/tests/test_debug.pytunsupported_parfor(scC@sEtj|ƒ}x/t|ƒD]!}||tj|ƒ||(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyt_check_dump_llvmUscC@s+|jd|j|ƒ|jd|ƒdS(Ns--FUNCTION OPTIMIZED DUMP %ssadd nsw i64 %arg.somearg, 1(RKRN(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyt_check_dump_func_opt_llvmZscC@s+|jd|j|ƒ|jd|ƒdS(Ns--OPTIMIZED DUMP %ssadd nsw i64 %arg.somearg, 1(RKRN(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyt_check_dump_optimized_llvm_scC@s@|jd|j|ƒtjƒdkr<|jd|ƒndS(Ns --ASSEMBLY %stx86_64tAMD64ti386ti686txorl(RURVRWRX(RKRNtplatformtmachine(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyt_check_dump_assemblycs(RRtsetR=R:RIRLRMRORPRRRSRTR\(((s5lib/python2.7/site-packages/numba/tests/test_debug.pyR-5s        tFunctionDebugTestBasecB@seZdZd„ZRS(RcC@sJtƒ5}tttjfƒ}|j|jdƒdƒWdQX|jƒS(Nii(RR RR tint64tassertPreciseEqualt entry_pointtgetvalue(RRCtcres((s5lib/python2.7/site-packages/numba/tests/test_debug.pytcompile_simple_nopythonms (RRRNRd(((s5lib/python2.7/site-packages/numba/tests/test_debug.pyR^istTestFunctionDebugOutputcB@sGeZd„Zd„Zd„Zd„Zd„Zd„Zd„ZRS(cC@s9tdtƒ|jƒ}WdQX|j|dgƒdS(Nt DUMP_BYTECODER.(RR?RdRI(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyttest_dump_bytecodewscC@s9tdtƒ|jƒ}WdQX|j|dgƒdS(NtDUMP_IRR0(RR?RdRI(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyt test_dump_ir|scC@s9tdtƒ|jƒ}WdQX|j|dgƒdS(NtDUMP_CFGR/(RR?RdRI(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyt test_dump_cfgscC@s9tdtƒ|jƒ}WdQX|j|dgƒdS(Nt DUMP_LLVMR2(RR?RdRI(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyttest_dump_llvm†scC@s9tdtƒ|jƒ}WdQX|j|dgƒdS(Nt DUMP_FUNC_OPTR3(RR?RdRI(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyttest_dump_func_opt_llvm‹scC@s9tdtƒ|jƒ}WdQX|j|dgƒdS(NtDUMP_OPTIMIZEDR4(RR?RdRI(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyttest_dump_optimized_llvmscC@s9tdtƒ|jƒ}WdQX|j|dgƒdS(Nt DUMP_ASSEMBLYR5(RR?RdRI(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyttest_dump_assembly•s( RRRgRiRkRmRoRqRs(((s5lib/python2.7/site-packages/numba/tests/test_debug.pyReus      tTestGeneratorDebugOutputcB@s eZdZd„Zd„ZRS(Rc C@s_tƒJ}tttjtjfƒ}|jt|jddƒƒddgƒWdQX|jƒS(Nii( RR RR R_R`tlistRaRb(RRCRc((s5lib/python2.7/site-packages/numba/tests/test_debug.pytcompile_simple_genŸs .cC@sotdtƒ|jƒ}WdQX|j|dgƒ|jd|j|ƒtjdƒ}|j||ƒdS(NRhR0s--GENERATOR INFO: %ssÜ generator state variables: ['x', 'y'] yield point #1: live variables = ['y'], weak live variables = ['x'] yield point #2: live variables = [], weak live variables = ['y'] (RR?RvRIRKRNttextwraptdedent(RRCtexpected_gen_info((s5lib/python2.7/site-packages/numba/tests/test_debug.pyttest_dump_ir_generator¦s (RRRNRvRz(((s5lib/python2.7/site-packages/numba/tests/test_debug.pyRt›s tTestDisableJITcB@s eZdZd„Zd„ZRS(s: Test the NUMBA_DISABLE_JIT environment variable. c C@sUtdtƒ@tƒ0tdtƒtƒ}|j|dƒdƒWdQXWdQXdS(Nt DISABLE_JITtnopythonii(RR?RRRR`(Rtcfunc((s5lib/python2.7/site-packages/numba/tests/test_debug.pyttest_jit¸s c C@sƒtdtƒntƒ^ttƒtƒ}|ƒ}|j|jdƒtdtƒt ƒ}|j||ƒdƒWdQXWdQXdS(NR|iR}( RR?RR tsimple_class_specRR`RRR (RtSimpleJITClassRR~((s5lib/python2.7/site-packages/numba/tests/test_debug.pyt test_jitclass¾s  (RRt__doc__RR‚(((s5lib/python2.7/site-packages/numba/tests/test_debug.pyR{³s tTestEnvironmentOverridecB@seZdZeZd„ZRS(sN Test that environment variables are reloaded by Numba when modified. c C@sz|jƒ}|j|ƒtddƒ3|jƒ}|j|ddddddgƒWdQX|jƒ}|j|ƒdS( Nt NUMBA_DEBUGt1R0R1R2R3R4R5(Rdt assertFalseRRI(RRC((s5lib/python2.7/site-packages/numba/tests/test_debug.pyt test_debugÒs    (RRRƒR>t_numba_parallel_test_Rˆ(((s5lib/python2.7/site-packages/numba/tests/test_debug.pyR„ÊstTestParforsDebugcB@sAeZdZeZd„Zeed„ƒƒZed„ƒZ RS(s5 Tests debug options associated with parfors cC@sPd}t}x-|D]%}|t|jƒkrt}PqqW|j|dƒdS(NsX'parallel=True' was specified but no transformation for parallel execution was possible.s Warning message should be found.(R>tstrtmessageR?t assertTrue(Rt warn_listtmsgt warning_foundtw((s5lib/python2.7/site-packages/numba/tests/test_debug.pytcheck_parfors_warningçs c C@sotjtjddƒ}tjdtƒ2}tjdtƒtt ||fdt ƒ}WdQX|j |ƒdS(sr Test that using parallel=True on a function that does not have parallel semantics warns. itCtrecordtalwaystflagsN( R tArraytfloat64twarningstcatch_warningsR?t simplefilterRR R%tforce_parallel_flagsR’(Rtarr_tyR‘Rc((s5lib/python2.7/site-packages/numba/tests/test_debug.pyt test_warnsñs c C@stttjfdtƒ}tddƒïtƒ"}tttjfdtƒ}WdQX|jƒjdƒ}g|D]}d|krt|^qt}g|D]}d|kr™|^q™}g|D]}d|kr¾|^q¾}t t j d ƒj |d ƒj d ƒƒ}t||t|ƒƒ} d df} d} xEt| || ƒD]1\} } }| j| |ƒ}|j|| ƒqEW|d |d f} d} x-|D]%} | j| Œ}|j|| ƒq›Wtjd t|gƒf} d} x-|D]%} | j| Œ}|j|| ƒqìWWdQXdS(sM Test that NUMBA_DEBUG_ARRAY_OPT_STATS produces valid output R–tNUMBA_DEBUG_ARRAY_OPT_STATSR†Ns sis produced from patterns is fused intosAfter fusion, functions #([0-9]+)iis"('ones function', 'NumPy mapping')Rtuserts6Parallel for-loop #{} is produced from pattern '{}' ats1Parallel for-loop #{} is fused into for-loop #{}.s:After fusion, function {} has {} parallel for-loop(s) #{}.(RR R¡(R R*R R_RœRRRbtsplittinttretcompiletsearchtgrouptrangetlentziptformatRKRR](RRcRCtoutputRtparallel_loop_outputt fuse_outputtafter_fusion_outputt parfor_statetboundstpatterntfmtR)ttrialstlpatterntto_match((s5lib/python2.7/site-packages/numba/tests/test_debug.pyttest_array_debug_opt_statsÿs>   %%%    ( RRRƒR>R‰R’RRRžR·(((s5lib/python2.7/site-packages/numba/tests/test_debug.pyRŠßs   t__main__(8t __future__RRtosRZR¤RwR™tnumpyR!tsupportRRRRRtnumbaRtunittestRR R tnumba.compilerR R tnumba.targets.cpuR t numba.errorsRRRt test_parforsRtmatmul_usecaseRRRtobjectRtint32R€R R%R*RœR]R?R-R^ReRtR{R„RŠRtmain(((s5lib/python2.7/site-packages/numba/tests/test_debug.pytsD      (       4 &Z