i?J3!!*c2 ;rctjdvr|jd|ytjdvr|jd|yttj)N)) )r] )r]r<)r] BINARY_OP))r] BINARY_ADD)r PYVERSIONassertInNotImplementedErrorr%rUs r_check_dump_bytecodez"DebugTestBase._check_dump_bytecodeKsF ??B B MM+s + __ * MM, ,%eoo6 6rc(|jd|y)NzCFG dominatorsrergs r_check_dump_cfgzDebugTestBase._check_dump_cfgSs &,rcB|jd|jz|y)Nz--IR DUMP: %s--re func_namergs r_check_dump_irzDebugTestBase._check_dump_irVs '$..8#>rc(|jd|y)Nz --propagate--rjrgs r_check_dump_typeinferz#DebugTestBase._check_dump_typeinferYs os+rc|jd|tjjdjj dk(r|j |d|yy)Nz --LLVM DUMP auto_parallelFz#store i64 %\"\.\d", i64\* %"retptr")rer FlagsoptionsdefaultrY assertRegexrgs r_check_dump_llvmzDebugTestBase._check_dump_llvm\sL mS) >> ! !/ 2 : : B Be K   S"H# N Lrcf|jd|jz||jd|y)Nz--FUNCTION OPTIMIZED DUMP %sadd nsw i64 %arg.somearg, 1rmrgs r_check_dump_func_opt_llvmz'DebugTestBase._check_dump_func_opt_llvmas) 4t~~EsK 3S9rcf|jd|jz||jd|y)Nz--OPTIMIZED DUMP %srzrmrgs r_check_dump_optimized_llvmz(DebugTestBase._check_dump_optimized_llvmfs) +dnn||~ ||~s AA;;Bctdd5|j}ddd|jdg|jd|jz|t j d}|j||y#1swYcxYw)NrTrBz--GENERATOR INFO: %sz generator state variables: ['x', 'y'] yield point #1: live variables = ['y'], weak live variables = ['x'] yield point #2: live variables = [], weak live variables = ['y'] )rrr[rerntextwrapdedent)r%rUexpected_gen_infos rtest_dump_ir_generatorz/TestGeneratorDebugOutput.test_dump_ir_generatorsw Y -))+C. dV, ,t~~=sC$OO- '-. -s BB N)r'r(r)rnrrrrrrrsI .rrceZdZdZdZdZy)TestDisableJITz: Test the NUMBA_DISABLE_JIT environment variable. ctdd5t5tdt}|j |ddddddddy#1swYxYw#1swYyxYw)N DISABLE_JITTnopythonrr])rrr rr)r%rs rtest_jitzTestDisableJIT.test_jitsT ]D 1!*T*?;''a!4"2 1!!2 1s" A$/AA$A! A$$A-cXtdd5t5ttt}|}|j |j dtdt}|j ||dddddddy#1swYxYw#1swYyxYw)NrTr"r) rrrsimple_class_specr rr$r r,)r%SimpleJITClassr+rs r test_jitclasszTestDisableJIT.test_jitclasss ]D 1!!<*;!<[!I$&''q1*T*+<=''c A6"2 1!!2 1s# B A+BB B B  B)N)r'r(r)__doc__rrrrrrrs5 7rrceZdZdZdZdZy)TestEnvironmentOverridezN Test that environment variables are reloaded by Numba when modified. Fc|j}|j|tdd5|j}|j|gdddd|j}|j|y#1swY+xYw)N NUMBA_DEBUG1)rBrCrDrErFrG)r assertFalserr[rgs r test_debugz"TestEnvironmentOverride.test_debugs{**,   4..0C  # #C*H I 5**, 5 4s %A==BN)r'r(r)r_numba_parallel_test_rrrrrrs " rrcdeZdZdZdZdZdZeedZ eedZ edZ y) TestParforsDebugz5 Tests debug options associated with parfors Fctd}d}|D]}|t|jvsd}n|j|dy)NzX'parallel=True' was specified but no transformation for parallel execution was possible.FT Warning message should be found.strmessage assertTruer% warn_listmsg warning_foundws rcheck_parfors_warningz&TestParforsDebug.check_parfors_warningsA2 Ac!))n$ $   'IJrctd}d}|D]}|t|jvsd}n|j|dy)Nzprange or pndindex loop will not be executed in parallel due to there being more than one entry to or exit from the loop (e.g., an assertion).FTrrrs r(check_parfors_unsupported_prange_warningz9TestParforsDebug.check_parfors_unsupported_prange_warningsC, Ac!))n$ $   'IJrc,tjtjdd}tjd5}tj dt t||fdtddd|jy#1swYxYw)zr Test that using parallel=True on a function that does not have parallel semantics warns. rCTrecordalwaysparallelN) r Arrayfloat64warningscatch_warnings simplefilterrr r2r)r%arr_tyrs r test_warnszTestParforsDebug.test_warnsspU]]As3  $ $D 1Q  ! !(,C D 1D&&!D 12D E2 ""1%2 1s 4B  Bctjd5}tjdtt t j fdtddd|jy#1swYxYw)zG Test that prange with multiple exits issues a warning TrrrN) rrrrr r rr=r)r%rs rtest_unsupported_prange_warnsz.TestParforsDebug.test_unsupported_prange_warnss[  $ $D 1Q  ! !(,C D /D%++$ /0B C2 55a82 1s AA22A;cttjfdtt dd5t 5}ttjfdtdddj jd}|Dcgc] }d|vs| c}}|Dcgc] }d|vs| c}}|Dcgc] }d |vs| c}}ttjd j|d jd }t||t|z}d } d} t||| D]*\} } } | j!| | }|j#|| ,|d z|d zf} d} |D]#} | j | }|j#|| %tj$d t'|gf} d} |D]#} | j | }|j#|| % dddy#1swYxYwcc}wcc}wcc}w#1swYyxYw)zM Test that NUMBA_DEBUG_ARRAY_OPT_STATS produces valid output TrNUMBA_DEBUG_ARRAY_OPT_STATSrN zis produced from patternz is fused intozAfter fusion, functionz #([0-9]+)rr)z"('ones function', 'NumPy mapping'))ruserz6Parallel for-loop #{} is produced from pattern '{}' atz1Parallel for-loop #{} is fused into for-loop #{}.z:After fusion, function {} has {} parallel for-loop(s) #{}.)r r rr:rrrsplitintrecompilesearchgrouprangelenzipformatrer'r)r%rUoutputrparallel_loop_output fuse_outputafter_fusion_output parfor_stateboundspatternfmtr9trialslpatternto_matchs rtest_array_debug_opt_statsz+TestParforsDebug.test_array_debug_opt_statss  ,ekk^d+,<= !> D "c3ekk^d34DE#\\^))$/F"FFq&@A&EFF !#;Fqo&:F; #DFq&>!&CFD rzz,7>>$Q' )).q3L<'#.B*CCEF /GLC'*63G+2(4#68::a2 h/(4 $a')9:GEC%%3::w/ h/& (00!S,5HIGNC-%3::w/ h/.QE D"" G;DE DsY H'G%%+H G2G2H% G7/G73H: G<G<DH%G/ *HH N) r'r(r)rrrrr rrrrrrrrrsg "K K & &994040rr__main__)/osrrrrnumpyr.numba.tests.supportrrrrrrr numbar r numba.corer r rnumba.core.errorsrrnumba.experimentalrunittestrrobjectr int32rr,r2r:r=r?rrrrrrr'mainrrrrs  """--5' &5;;'(  6'H6'r M #33#3L.}.07]7.3*j0xj0Z zHMMOr