U ,‰d‹ã@s<ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl mZmZm Z ddlmZmZmZddlmZmZmZddlmZddl mZmZddlmZmZmZddlmZm Z m!Z!m"Z"dd l#m$Z$m%Z%dd l&m'Z'ddl(m)Z)ddl*m+Z+dd l&m,Z,e-j.dkZ/ej0 1d¡Z2eƒZ3de3_4de3_5eƒZ6de6_5eƒZ7de7_8de7_5dd„Z9dd„Z:dd„Z;dd„ZGdd„de!ƒZ?Gdd „d e?eƒZ@Gd!d"„d"e?eƒZAGd#d$„d$eƒZBGd%d&„d&eBƒZCGd'd(„d(eƒZDGd)d*„d*eƒZEGd+d,„d,eEƒZFeF G¡Gd-d.„d.eEƒZHeH G¡Gd/d0„d0eEƒZIeI G¡Gd1d2„d2eEƒZJeJ G¡Gd3d4„d4eEƒZKeK G¡Gd5d6„d6eEƒZLeL G¡Gd7d8„d8eEƒZMeM G¡Gd9d:„d:eEƒZNeN G¡Gd;d<„d„d>eEƒZPeP G¡Gd?d@„d@eƒZQGdAdB„dBeƒZRGdCdD„dDeƒZSGdEdF„dFeƒZTeUdGkr8e V¡dS)HéN)ÚtypeofÚnjitÚguvectorize)ÚtypesÚtypingÚutils)Úcompile_isolatedÚFlagsÚ DEFAULT_FLAGS)Ú from_dtype)ÚjitÚ vectorize)Ú LoweringErrorÚTypingErrorÚNumbaTypeError)ÚTestCaseÚCompilationCacheÚMemoryLeakMixinÚtag)Úsupported_ufuncsÚ all_ufuncs)Ú numpy_support)Ú cpu_target)ÚBaseContext)Úufunc_dbéÚwin32Tcst ˆ¡‡fdd„ƒ}dS)zŠAn 'expectedFailure' like decorator that only expects compilation errors caused by unimplemented functions that fail in no-python modecs<zˆ||ŽWn"tk r0t t ¡¡‚YnXtj‚dS©N)rÚunittestZ_ExpectedFailureÚsysÚexc_infoZ_UnexpectedSuccess)ÚargsÚkwargs©Úfunc©ú;/tmp/pip-unpacked-wheel-eu7e0c37/numba/tests/test_ufuncs.pyÚwrapper.s z_unimplemented..wrapperN)Ú functoolsÚwraps)r$r'r%r#r&Ú_unimplemented+sr*cCsTi}d dd„t|jƒDƒ¡}d ||j¡}t|tƒ|ƒ|d}d |j¡|_|S)Nú,cSsg|]}d |¡‘qS)za{0})Úformat©Ú.0Úir%r%r&Ú 8sz'_make_ufunc_usecase..zdef fn({0}): np.{1}({0})Úfnz{0}_usecase)ÚjoinÚrangeÚnargsr,Ú__name__ÚexecÚglobals)ÚufuncÚldictZarg_strZfunc_strr1r%r%r&Ú_make_ufunc_usecase6sr:cCs4i}td |¡tƒ|ƒ|d}d t|ƒ¡|_|S)Nzdef fn(x): return {0}(x)r1úusecase_{0}©r6r,r7Úhashr5©Úufunc_opr9r1r%r%r&Ú_make_unary_ufunc_op_usecase?s r@cCs4i}td |¡tƒ|ƒ|d}d t|ƒ¡|_|S)Nzdef fn(x,y): return x{0}yr1r;r<r>r%r%r&Ú_make_binary_ufunc_op_usecaseFs rAcsPtˆtƒrd?„Z#efd@dA„Z$efdBdC„Z%efdDdE„Z&efdFdG„Z'efdHdI„Z(efdJdK„Z)efdLdM„Z*efdNdO„Z+efdPdQ„Z,efdRdS„Z-efdTdU„Z.efdVdW„Z/efdXdY„Z0efdZd[„Z1efd\d]„Z2efd^d_„Z3efd`da„Z4efdbdc„Z5efddde„Z6efdfdg„Z7efdhdi„Z8efdjdk„Z9efdldm„Z:efdndo„Z;efdpdq„Zefdvdw„Z?efdxdy„Z@efdzd{„ZAefd|d}„ZBefd~d„ZCefd€d„ZDefd‚dƒ„ZEefd„d…„ZFefd†d‡„ZGefdˆd‰„ZHefdŠd‹„ZIefdŒd„ZJefdŽd„ZKefdd‘„ZLefd’d“„ZMefd”d•„ZNefd–d—„ZOd˜d™„ZPefdšd›„ZQefdœd„ZRefdždŸ„ZSefd d¡„ZTefd¢d£„ZUefd¤d¥„ZVeWefd¦d§„ƒZXeWefd¨d©„ƒZYefdªd«„ZZefd¬d„Z[efd®d¯„Z\efd°d±„Z]efd²d³„Z^d´dµ„Z_d¶d·„Z`d¸d¹„Zadºd»„Zbd¼d½„ZcdS)¾Ú TestUFuncsNZifcFc s(| t¡t|ƒ} t|jƒ|} | D]ü}|d}|d} t|tƒ}|rN|‰n|f|j‰| |krdq$|r|t ˆddk¡r|q$ˆdj j|krŽq$| | ||¡}| f|j}|f|j }|jj| |||d}|j}tˆdtjƒr‡fdd„|Dƒ}‡fdd„|Dƒ}ndd„|Dƒ}dd„|Dƒ}d }tjd dP}t d¡| ˆ|žŽd }|D]*}t|jtƒrTt|jƒ |¡rTd }qTW5QRX|ˆ|žŽt||ƒD]€\}}d dddddddg¡ |j| ||||j |¡}ztjj||d|dWn.t k r|rt!d|||ƒn‚YnXq q$dS)NrrS©Úflagscs$g|]}tjˆdj|jjd‘qS©rrY©rJÚzerosÚsizerKÚname©r.Zout_ty©r!r%r&r0Ðsþÿz/TestUFuncs.basic_ufunc_test..cs$g|]}tjˆdj|jjd‘qSr}r~r‚rƒr%r&r0ÕsþÿcSsg|]}tjd|jjd‘qS©rSrY©rJrrKrr‚r%r%r&r0ÛsÿcSsg|]}tjd|jjd‘qSr„r…r‚r%r%r&r0ßsÿFT)ÚrecordÚalwayszinvalid value encounteredÚ zufunc '{0}' failedz inputs ({1}):z{2}zgot({3})z{4}zexpected ({5}):z{6}é)ÚdecimalÚerr_msgz!Output mismatch for invalid input)"Zreset_module_warningsr5r:ÚlistrlrGÚtupleÚninrJÚanyrKÚkindrvÚnoutrmÚcompileÚentry_pointÚndarrayÚwarningsÚcatch_warningsÚsimplefilterÚ issubclassÚcategoryÚRuntimeWarningrHÚmessageÚ startswithrOr2r,ÚtestingÚassert_array_almost_equalÚAssertionErrorÚprint)ror8r|Úskip_inputsÚadditional_inputsrsrtÚkindsÚ positive_onlyÚpyfuncrlÚinput_tupleÚ input_operandrrZis_tupleruZinput_typesÚoutput_typesÚcrÚcfuncÚresultsÚexpectedZinvalid_flagZwarnlistZwarnmsgZthiswarnZ expected_iZresult_iÚmsgr%rƒr&Úbasic_ufunc_test¦s¦ ÿÿ ý ýþþ ÿýùý ÿzTestUFuncs.basic_ufunc_testcCs8|j||tjtjt tjdd¡t tjdd¡gddS)NrSrZ)r|r¡)r®rrfrgri)rorr|r%r%r&Úbasic_int_ufunc_testsþÿzTestUFuncs.basic_int_ufunc_testcCs|jtj|ddS©Nr{)r®rJÚadd©ror|r%r%r&Útest_add_ufuncszTestUFuncs.test_add_ufunccCs|jtj|ddSr°)r®rJÚsubtractr²r%r%r&Útest_subtract_ufuncszTestUFuncs.test_subtract_ufunccCs|jtj|ddSr°)r®rJÚmultiplyr²r%r%r&Útest_multiply_ufuncszTestUFuncs.test_multiply_ufunccCs d}tj}|jtj||ddS©N)r|rs)rrgr®rJÚdivide)ror|Úint_out_typer%r%r&Útest_divide_ufuncszTestUFuncs.test_divide_ufunccCs|jtjdddS©NÚf)r£)r®rJZ logaddexprnr%r%r&Útest_logaddexp_ufunc"szTestUFuncs.test_logaddexp_ufunccCs|jtjdddSr¼)r®rJZ logaddexp2rnr%r%r&Útest_logaddexp2_ufunc%sz TestUFuncs.test_logaddexp2_ufunccCs|jtj|tjddSr¸)r®rJÚtrue_dividerrgr²r%r%r&Útest_true_divide_ufunc(sz!TestUFuncs.test_true_divide_ufunccCs| tj¡dSr)r®rJÚfloor_dividernr%r%r&Útest_floor_divide_ufunc+sz"TestUFuncs.test_floor_divide_ufunccCs,|jtjtjt tjdd¡tjg|ddS)NrSrZ)rsr¡r|)r®rJÚnegativerrerirbr²r%r%r&Útest_negative_ufunc.sþzTestUFuncs.test_negative_ufunccCs|jtj|ddSr°)r®rJZpositiver²r%r%r&Útest_positive_ufunc5szTestUFuncs.test_positive_ufunccCs|jtj|dddS)NT)r|r¤)r®rJÚpowerr²r%r%r&Útest_power_ufunc8s ÿzTestUFuncs.test_power_ufunccCs|jtj|dddS)NZfc©r|r£)r®rJZfloat_powerr²r%r%r&Útest_float_power_ufunc<sz!TestUFuncs.test_float_power_ufunccCs|jtj|dddS©NÚiurÉ)r®rJÚgcdr²r%r%r&Útest_gcd_ufunc?szTestUFuncs.test_gcd_ufunccCs|jtj|dddSrË)r®rJZlcmr²r%r%r&Útest_lcm_ufuncBszTestUFuncs.test_lcm_ufunccCs|jtj|ddSr°)r®rJÚ remainderr²r%r%r&Útest_remainder_ufuncEszTestUFuncs.test_remainder_ufuncc Cs:|jtj|dt t tj¡j¡t d¡ftjfgddS)NZifcué©r|r£r¢)r®rJÚmodrdÚiinfoÚmaxrr²r%r%r&Útest_mod_ufuncHs"ÿÿzTestUFuncs.test_mod_ufunccCs|jtj|ddSr°)r®rJÚfmodr²r%r%r&Útest_fmod_ufuncNszTestUFuncs.test_fmod_ufunccCst|j||t t tj¡j¡tjft t tj¡j¡tjft t tj¡j ¡tjft t tj ¡j ¡tj fgddS)N)r|r¢)r®rJrbrÕrÖrrdrfZfinfoÚminrg)ror|r8r%r%r&Útest_abs_ufuncQsüÿzTestUFuncs.test_abs_ufunccCs|j|tjddS)N)r|r8)rÛrJÚabsoluter²r%r%r&Útest_absolute_ufuncZszTestUFuncs.test_absolute_ufunccCs|jtj|dddS©Nr½rÉ)r®rJÚfabsr²r%r%r&Útest_fabs_ufunc]szTestUFuncs.test_fabs_ufunccCs|jtj|dddS©NÚcfrÉ)r®rJZrintr²r%r%r&Útest_rint_ufunc`szTestUFuncs.test_rint_ufunccCs|jtj|ddSr°)r®rJÚsignr²r%r%r&Útest_sign_ufunccszTestUFuncs.test_sign_ufunccCs|jtj|ddSr°)r®rJZconjr²r%r%r&Útest_conj_ufuncfszTestUFuncs.test_conj_ufunccCs|jtj|dddSrá)r®rJÚexpr²r%r%r&Útest_exp_ufunciszTestUFuncs.test_exp_ufunccCs|jtj|dddSrá)r®rJZexp2r²r%r%r&Útest_exp2_ufunclszTestUFuncs.test_exp2_ufunccCs|jtj|dddSrá)r®rJÚlogr²r%r%r&Útest_log_ufuncoszTestUFuncs.test_log_ufunccCs|jtj|dddSrá)r®rJÚlog2r²r%r%r&Útest_log2_ufuncrszTestUFuncs.test_log2_ufunccCs|jtj|dddSrá)r®rJÚlog10r²r%r%r&Útest_log10_ufuncuszTestUFuncs.test_log10_ufunccCs|jtj|dddSrá)r®rJÚexpm1r²r%r%r&Útest_expm1_ufuncxszTestUFuncs.test_expm1_ufunccCs|jtj|dddSrá)r®rJÚlog1pr²r%r%r&Útest_log1p_ufunc{szTestUFuncs.test_log1p_ufunccCs|jtj|dddSrá)r®rJÚsqrtr²r%r%r&Útest_sqrt_ufunc~szTestUFuncs.test_sqrt_ufunccCs|jtj|ddSr°)r®rJZsquarer²r%r%r&Útest_square_ufuncszTestUFuncs.test_square_ufunccCs|jtj|dddSrÞ)r®rJÚcbrtr²r%r%r&Útest_cbrt_ufunc„szTestUFuncs.test_cbrt_ufunccCsbt tjdd¡tjt tjdd¡tjt tjdd¡tjt tjdd¡tjg}|jtj||ddS)NrSrZ)r¡r|) rrirbrcrdrer®rJÚ reciprocal©ror|Zto_skipr%r%r&Útest_reciprocal_ufunc‡sýz TestUFuncs.test_reciprocal_ufunccCs|jtj|ddSr°)r®rJÚ conjugater²r%r%r&Útest_conjugate_ufunc‘szTestUFuncs.test_conjugate_ufunccCs|jtj|dddSrá)r®rJÚsinr²r%r%r&Útest_sin_ufunc˜szTestUFuncs.test_sin_ufunccCs|jtj|dddSrá)r®rJÚcosr²r%r%r&Útest_cos_ufunc›szTestUFuncs.test_cos_ufunccCs|jtj|dddSrá)r®rJÚtanr²r%r%r&Útest_tan_ufuncžszTestUFuncs.test_tan_ufunccCs|jtj|dddSrá)r®rJÚarcsinr²r%r%r&Útest_arcsin_ufunc¡szTestUFuncs.test_arcsin_ufunccCs|jtj|dddSrá)r®rJÚarccosr²r%r%r&Útest_arccos_ufunc¤szTestUFuncs.test_arccos_ufunccCs|jtj|dddSrá)r®rJZarctanr²r%r%r&Útest_arctan_ufunc§szTestUFuncs.test_arctan_ufunccCs|jtj|dddSrá)r®rJZarctan2r²r%r%r&Útest_arctan2_ufuncªszTestUFuncs.test_arctan2_ufunccCs|jtjdddSr¼)r®rJÚhypotr²r%r%r&Útest_hypot_ufuncszTestUFuncs.test_hypot_ufunccCs|jtj|dddSrá)r®rJÚsinhr²r%r%r&Útest_sinh_ufunc°szTestUFuncs.test_sinh_ufunccCs|jtj|dddSrá)r®rJÚcoshr²r%r%r&Útest_cosh_ufunc³szTestUFuncs.test_cosh_ufunccCs|jtj|dddSrá)r®rJÚtanhr²r%r%r&Útest_tanh_ufunc¶szTestUFuncs.test_tanh_ufunccCs|jtj|dddSrá)r®rJZarcsinhr²r%r%r&Útest_arcsinh_ufunc¹szTestUFuncs.test_arcsinh_ufunccCs|jtj|dddSrá)r®rJZarccoshr²r%r%r&Útest_arccosh_ufunc¼szTestUFuncs.test_arccosh_ufunccCsdt tjdd¡tjt tjdd¡tjt tjdd¡tjt tjdd¡tjg}|jtj||dddS)NrSrZrâ)r¡r|r£) rrirbrcrdrer®rJZarctanhrúr%r%r&Útest_arctanh_ufunc¿s ýÿzTestUFuncs.test_arctanh_ufunccCs|jtj|dddSrÞ)r®rJZdeg2radr²r%r%r&Útest_deg2rad_ufuncÐszTestUFuncs.test_deg2rad_ufunccCs|jtj|dddSrÞ)r®rJZrad2degr²r%r%r&Útest_rad2deg_ufuncÓszTestUFuncs.test_rad2deg_ufunccCs|jtj|dddSrÞ)r®rJÚdegreesr²r%r%r&Útest_degrees_ufuncÖszTestUFuncs.test_degrees_ufunccCs|jtj|dddSrÞ)r®rJÚradiansr²r%r%r&Útest_radians_ufuncÙszTestUFuncs.test_radians_ufunccCs|jtj|ddSr°)r¯rJZbitwise_andr²r%r%r&Útest_bitwise_and_ufuncßsz!TestUFuncs.test_bitwise_and_ufunccCs|jtj|ddSr°)r¯rJZ bitwise_orr²r%r%r&Útest_bitwise_or_ufuncâsz TestUFuncs.test_bitwise_or_ufunccCs|jtj|ddSr°)r¯rJZbitwise_xorr²r%r%r&Útest_bitwise_xor_ufuncåsz!TestUFuncs.test_bitwise_xor_ufunccCs|jtj|ddSr°)r¯rJÚinvertr²r%r%r&Útest_invert_ufuncèszTestUFuncs.test_invert_ufunccCs|jtj|ddSr°)r¯rJZbitwise_notr²r%r%r&Útest_bitwise_not_ufuncësz!TestUFuncs.test_bitwise_not_ufunccCs|jtj|ddSr°)r®rJÚgreaterr²r%r%r&Útest_greater_ufuncúszTestUFuncs.test_greater_ufunccCs|jtj|ddSr°)r®rJÚ greater_equalr²r%r%r&Útest_greater_equal_ufuncýsz#TestUFuncs.test_greater_equal_ufunccCs|jtj|ddSr°)r®rJÚlessr²r%r%r&Útest_less_ufuncszTestUFuncs.test_less_ufunccCs|jtj|ddSr°)r®rJÚ less_equalr²r%r%r&Útest_less_equal_ufuncsz TestUFuncs.test_less_equal_ufunccCs|jtj|ddSr°)r®rJÚ not_equalr²r%r%r&Útest_not_equal_ufuncszTestUFuncs.test_not_equal_ufunccCs|jtj|ddSr°)r®rJÚequalr²r%r%r&Útest_equal_ufunc szTestUFuncs.test_equal_ufunccCs|jtj|ddSr°)r®rJÚlogical_andr²r%r%r&Útest_logical_and_ufuncsz!TestUFuncs.test_logical_and_ufunccCs|jtj|ddSr°)r®rJÚ logical_orr²r%r%r&Útest_logical_or_ufuncsz TestUFuncs.test_logical_or_ufunccCs|jtj|ddSr°)r®rJÚlogical_xorr²r%r%r&Útest_logical_xor_ufuncsz!TestUFuncs.test_logical_xor_ufunccCs|jtj|ddSr°)r®rJZlogical_notr²r%r%r&Útest_logical_not_ufuncsz!TestUFuncs.test_logical_not_ufunccCs|jtj|ddSr°)r®rJÚmaximumr²r%r%r&Útest_maximum_ufuncszTestUFuncs.test_maximum_ufunccCs|jtj|ddSr°)r®rJZminimumr²r%r%r&Útest_minimum_ufuncszTestUFuncs.test_minimum_ufunccCs|jtj|ddSr°)r®rJZfmaxr²r%r%r&Útest_fmax_ufuncszTestUFuncs.test_fmax_ufunccCs|jtj|ddSr°)r®rJZfminr²r%r%r&Útest_fmin_ufunc!szTestUFuncs.test_fmin_ufunccCs&tjddgtjdt tjdd¡fgS)NTFrYrSrZ)rJrhÚbool_rrirnr%r%r&Úbool_additional_inputs(sÿÿz!TestUFuncs.bool_additional_inputscCs|jtj|d| ¡ddS©NZifcbrÓ)r®rJÚisfiniter:r²r%r%r&Útest_isfinite_ufunc.sþzTestUFuncs.test_isfinite_ufunccCs|jtj|d| ¡ddSr;)r®rJÚisinfr:r²r%r%r&Útest_isinf_ufunc4sþzTestUFuncs.test_isinf_ufunccCs|jtj|d| ¡ddSr;)r®rJÚisnanr:r²r%r%r&Útest_isnan_ufunc:sþzTestUFuncs.test_isnan_ufunccCs|jtj|ddSr°)r®rJÚsignbitr²r%r%r&Útest_signbit_ufunc@szTestUFuncs.test_signbit_ufunccCs|jtj|dddSrÞ)r®rJÚcopysignr²r%r%r&Útest_copysign_ufuncCszTestUFuncs.test_copysign_ufunccCs|jtj|dddSrÞ)r®rJZ nextafterr²r%r%r&Útest_nextafter_ufuncFszTestUFuncs.test_nextafter_ufunccCs|jtj|dddSrÞ)r®rJÚmodfr²r%r%r&Útest_modf_ufuncIszTestUFuncs.test_modf_ufunccCs|jtj|dddSrÞ)r®rJÚfrexpr²r%r%r&Útest_frexp_ufuncQszTestUFuncs.test_frexp_ufunccCs|jtj|dddSrÞ)r®rJÚfloorr²r%r%r&Útest_floor_ufuncUszTestUFuncs.test_floor_ufunccCs|jtj|dddSrÞ)r®rJÚceilr²r%r%r&Útest_ceil_ufuncXszTestUFuncs.test_ceil_ufunccCs|jtj|dddSrÞ)r®rJÚtruncr²r%r%r&Útest_trunc_ufunc[szTestUFuncs.test_trunc_ufunccCs|jtj|dddSrÞ)r®rJÚspacingr²r%r%r&Útest_spacing_ufunc^szTestUFuncs.test_spacing_ufunccCsÔ|j}t|ƒ}dtjfdtjfdtjftjddgddt tjdd¡ftjddgddt tjdd¡ftjd dgd dt tjdd¡fg}|}t tjdd¡t tjdd¡g}|}t |||¡D]\}} } |d}|d}| d} | d}|dkr"|t tjdd¡ksÀ|t tjdd¡kr"qÀ|dkrb|t tjdd¡krb|tjkrbt tjdd¡rbqÀ|dkr¢|t tjdd¡kr¢|tjkr¢t tjdd¡r¢qÀt|tjƒs¾t|tjƒrÎt| tjƒsÎqÀ|j j|||| f|d }|j}t|tjƒr$tj|j| jjd}tj|j| jjd}n\t| tjƒr\tj| j| jjd}tj| j| jjd}n$tjd| jjd}tjd| jjd}||| |ƒ||| |ƒt| d| ƒ}|tjtjfkrºdnd}|j|||dqÀdS)NrSrTrWrr[rYrZr]rUr_r¹r´r{rKÚsingleÚdouble)Úprec)r5r:rrdrergrJrhriÚ itertoolsÚproductrbrGrmr’r“r”rr€rKrÚgetattrrfÚ complex64ÚassertPreciseEqual)ror8r|Z ufunc_nameZinputs1Zinputs2r¨r¥Zinput1Zinput2ruZinput1_operandÚinput1_typeZinput2_operandÚinput2_typer©rªÚresultr¬Zscalar_typerUr%r%r&Úbinary_ufunc_mixed_types_testdsŒ ù ÿÿ ÿÿ ÿÿ ÿ þþÿÿÿÿÿþz(TestUFuncs.binary_ufunc_mixed_types_testc Csþttjƒ}tjdddtjddd dd¡tjddd dd¡tjddd dd¡tjddd dd¡tjddd dd¡g}tjddd dd¡tjddd dd¡tjddd dd¡tjddd ddd¡tjddd ddd¡tjddd ddd¡g}t||ƒD]~\}}t tj |j d¡}t tj|j d¡}|jj |||ftd}|j} tj|j|jd} t || ¡| ||ƒ| || ¡q ttjƒ}tjdd dtjdd d dd¡tjdd d ddd¡tjdd d dd¡tjdd d dd¡tjdd d ddd¡tjdd d ddd¡tjdd d ddd¡tjdd d ddd¡g }|}t ||¡D]š\}} t tj |j d¡}t tj | j d¡}t tj t|j | j ƒd¡}|jj ||||ftd}|j} t || ¡} tj| jd d}| || |ƒ| || ¡q^dS) Nér]rYrSé érZr{r[)r:rJrÄÚarangeÚreshaperrOrrirdÚndimrermr’Úno_pyobj_flagsr“ÚshaperKrZr±rVrWrÖ)ror¥Zinput_operandsZoutput_operandsrCr]rrrur©rªr¬Úinput1_operandsÚinput2_operandsrDr[r\r%r%r&Útest_broadcastingsd ú úÿ ÷ÿzTestUFuncs.test_broadcastingc CsD| t¡0dd„}t tjdd¡}t|||ftd}W5QRXdS)NcSst ||¡Sr©rJr±©Úa0Úa1r%r%r&Úmyaddósz2TestUFuncs.test_implicit_output_npm..myaddrSrZr{)ÚassertRaisesrrrirdrre)rornZarr_tyr©r%r%r&Útest_implicit_output_npmñs ÿz#TestUFuncs.test_implicit_output_npmc CsDdd„}tjdddtjddd dd¡tjddd ddd¡tjddd dd¡tjddd dd¡tjddd ddd¡tjddd ddd¡tjddd ddd¡tjddd ddd¡g }|}t ||¡D]l\}}t tj|jd ¡}t tj|jd ¡}|j j |||ftd }|j} t ||¡} | ||ƒ}tj | |¡qÒdS)NcSst ||¡Srrjrkr%r%r&r¥úszATestUFuncs.test_broadcast_implicit_output_npm_nrt..pyfuncr_r[rYr`rarSrZr{)rJrbrcrVrWrrirdrdrmr’Úenable_nrt_flagsr“r±rÚassert_array_equal)ror¥rgrhrCrDr[r\r©rªr¬r]r%r%r&Ú&test_broadcast_implicit_output_npm_nrtùs,÷ÿ z1TestUFuncs.test_broadcast_implicit_output_npm_nrtcCsddd„}t ddd¡ dd¡}tj|dd }| dd¡jd}t|ƒ}|jjrX|jd ks\t ‚t|ƒ}|jj rv|jdkszt ‚t|ƒ}|jdkst ‚|jjrœt ‚|jj r¨t ‚tt |||gd¡ƒ}|||fg7}|||fg7}|||fg7}|D]r\} } |jj|t| ƒt| ƒftd }|| | ƒ}| | | ¡} | |jj| jj¡| |jj | jj ¡tj || ¡qìdS)NcSst ||¡Srrjrkr%r%r&r¥sz=TestUFuncs.test_implicit_output_layout_binary..pyfuncrrSérr‰ÚF©ÚorderrZÚAér{)rJÚlinspacercrhÚTrr|Úc_contiguousÚlayoutrŸÚf_contiguousrŒrVÚpermutationsrmr’rqr“ÚassertEqualrrr)ror¥ÚXÚYÚZÚXtyÚYtyÚZtyZ testcasesÚarg0Zarg1r©r¬r]r%r%r&Ú"test_implicit_output_layout_binarys:ÿ ÿ ÿz-TestUFuncs.test_implicit_output_layout_binarycCsdd„}t ddd¡ dd¡}tj|dd }| dd¡jd}t|ƒ}|jjrX|jd ks\t ‚t|ƒ}|jj rv|jdkszt ‚t|ƒ}|jdkst ‚|jjrœt ‚|jj r¨t ‚|||fD]d}|jj|t|ƒft d} ||ƒ} | |¡}| | jj|jj¡| | jj |jj ¡tj | |¡q²dS) NcSs t |¡Sr©rJrô)rlr%r%r&r¥=sz.pyfuncrrSrtrr‰rurvrZrxr{)rJrzrcrhr{rr|r|r}rŸr~rmr’rqr“r€rrr)ror¥rr‚rƒr„r…r†r‡r©r¬r]r%r%r&Ú!test_implicit_output_layout_unary<s2ÿ ÿ ÿz,TestUFuncs.test_implicit_output_layout_unary)dr5rwrxrer®r¯r³rµr·r»r¾r¿rÁrÃrÅrÆrÈrÊrÎrÏrÑr×rÙrJÚabsrÛrÝràrãrårærèrérërírïrñrórõrörørûrýrÿrrrrrr rr rrrrrrrrrrrrrr r"r$r&r(r*r,r.r0r2r3r5r6r7r8r:r=r?rArCrErFr*rHrJrLrNrPrRr^rirprsrˆrŠr%r%r%r&rz¤sÌý b ID%rzc@s†eZdZdd„Zeggddfdd„Zeggdddfdd„Zd d „Zdd„Zd d„Z dd„Z dd„Zdd„Zdd„Z dd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„Zd)d*„Zd+d,„Zd-d.„Zd/d0„Zd1d2„Zd3d4„Zd5d„Zd6d7„Zd8d9„Zd:d;„Zdd?„Z!d@dA„Z"dBdC„Z#dDdE„Z$dFdG„Z%dHdI„Z&dJdK„Z'dLdM„Z(dNdO„Z)dPdQ„Z*dRdS„Z+dTdU„Z,dVdW„Z-dXdY„Z.dS)ZÚTestArrayOperatorscCs&| |jj|jj¡tj ||¡dSr)r€rKrrJrrž)ror¬Úgotr%r%r&Ú_check_results`sz!TestArrayOperators._check_resultsNcCs‚t|ƒ}t|jƒ}| |¡|} |D]X} | \}}||ks$t|tjƒsFq$|jj| |f|d} | j }| |ƒ}||ƒ}| ||¡q$dSr°)r@rŒrlÚextendrGrrirmr’r“rŽ)roÚoperatorr|r¡r¢rsrtÚ operator_funcrlr¥r¦r§rrr©rªr¬rr%r%r&Ú unary_op_testds" ÿÿz TestArrayOperators.unary_op_testFcCst|ƒ}t|jƒ} | |¡|} tj d¡}| D]Ò}|\} }t t |d|ƒ¡}|}||kr\q0|rrtj d|dd}t|tj ƒrœ| }|}|rÆt | |k¡rÆq0n*| ddd¡ |¡}t|ƒ}|rÆ| |krÆq0|jj| ||f|d}|j}| || ƒ}||| ƒ}| ||¡q0dS)NrSrKrYrédé r{)rArŒrlrrJÚrandomZRandomStaterZas_dtyperXrrGrrirÚuniformZastyperrmr’r“rŽ)rorr|r¡r¢rsrtÚpositive_rhsr‘rlr¥Zrandom_stater¦Zinput_operand1rrZinput_dtypeZinput_type1ZzeroZinput_operand0Zinput_type0r©rªr¬rr%r%r&Úbinary_op_testysB ÿÿÿ z!TestArrayOperators.binary_op_testcCs0dtjfdtjft ddg¡t tjdd¡fgS)NTFrSrZ)rÚbooleanrJrhrirnr%r%r&Úbitwise_additional_inputs¤sýz,TestArrayOperators.bitwise_additional_inputsc OsD| dg¡}|tjtjt tjdd¡t tjdd¡g7}|j||ŽS)Nr¡rSrZ)Ú setdefaultrrfrgrir˜)ror!Úkwsr¡r%r%r&Úbinary_int_op_testsýz%TestArrayOperators.binary_int_op_testcOs$| dg¡}|| ¡7}|j||ŽS)Nr¢)r›ršr)ror!rœr¢r%r%r&Úbinary_bitwise_op_test¶sz)TestArrayOperators.binary_bitwise_op_testcsÀt|ƒ}|}‡fdd„|Dƒ}‡fdd„ˆDƒ} ‡fdd„ˆDƒ} | | }t ||¡D]d\}} t|ƒ}t| ƒ}|jj|||ftd}|j}| ¡}||| ƒ| ¡}||| ƒ| ||¡qVdS)Ncsg|]}tjˆ|d‘qS©rY©rJrh©r.rK)Ú lhs_valuesr%r&r0Âsÿz6TestArrayOperators.inplace_op_test..csg|]}tjˆ|d‘qSrŸr r¡)Ú rhs_valuesr%r&r0Åsÿcsg|]}ˆD]}||ƒ‘qqSr%r%)r.ÚvrK)Ú rhs_dtypesr%r&r0Çsr{) rIrVrWrrmr’rer“ÚcopyrZ)rorr¢r£Z lhs_dtypesr¥r‘r¥Z lhs_inputsZ rhs_arraysZrhs_scalarsZ rhs_inputsÚlhsÚrhsZlhs_typeZrhs_typer©rªr¬rr%)r¢r¥r£r&Úinplace_op_test»s, ÿ ÿÿ z"TestArrayOperators.inplace_op_testc Cs&| |||tjtjftjtjtjf¡Sr)r©rJrfrgre©rorr¢r£r%r%r&Úinplace_float_op_testÖs þz(TestArrayOperators.inplace_float_op_testcCs*| |||tjtjtjftjtjf¡dSr)r©rJrkrcrerbrªr%r%r&Úinplace_int_op_testÝs þz&TestArrayOperators.inplace_int_op_testcCs0| |||¡| |||tjftjtjf¡dSr)r¬r©rJr9rªr%r%r&Úinplace_bitwise_op_testâs ÿz*TestArrayOperators.inplace_bitwise_op_testcCs| d¡dS©Nú+©r’rnr%r%r&Útest_unary_positive_array_opêsz/TestArrayOperators.test_unary_positive_array_opcCs| d¡dS©Nú-r°rnr%r%r&Útest_unary_negative_array_opísz/TestArrayOperators.test_unary_negative_array_opc Cs<|jdtjtjt tjdd¡t tjdd¡g| ¡ddS)Nú~rSrZ)r¡r¢)r’rrfrgriršrnr%r%r&Útest_unary_invert_array_opðsþüz-TestArrayOperators.test_unary_invert_array_opcCs:| ddddgdddg¡| tjdddgdddg¡dS)Nz+=rTçø?r_éûÿÿÿrç@)r«rÚiaddrnr%r%r&Útest_inplace_addúsz#TestArrayOperators.test_inplace_addcCs:| ddddgdddg¡| tjdddgdddg¡dS)Nz-=rTr·r_r¸rr¹)r«rÚisubrnr%r%r&Útest_inplace_subþsz#TestArrayOperators.test_inplace_subcCs:| ddddgdddg¡| tjdddgdddg¡dS)Nz*=rTr·r_r¸rr¹)r«rÚimulrnr%r%r&Útest_inplace_mulsz#TestArrayOperators.test_inplace_mulcCs:| ddddgdddg¡| tjdddgdddg¡dS)Nz//=rTr·r_r¸çô?r¹)r«rÚ ifloordivrnr%r%r&Útest_inplace_floordivsz(TestArrayOperators.test_inplace_floordivcCs:| ddddgdddg¡| tjdddgdddg¡dS) Nz/=rTr·r_r¸rr¹rÀ)r«rÚitruedivrnr%r%r&Útest_inplace_div sz#TestArrayOperators.test_inplace_divcCs:| ddddgdddg¡| tjdddgdddg¡dS)Nz%=rTr·r_r¸ryr¹)r«rÚimodrnr%r%r&Útest_inplace_remaindersz)TestArrayOperators.test_inplace_remaindercCs:| ddddgdddg¡| tjdddgdddg¡dS)Nz**=rTr·r_r¸ryr¹)r«rÚipowrnr%r%r&Útest_inplace_powsz#TestArrayOperators.test_inplace_powc CsJ| ddddddgdddd d g¡| tjdddddgdddd d g¡dS)Nz&=rrSryr_é3é rÒé*éÿ)rrÚiandrnr%r%r&Útest_inplace_ands"z#TestArrayOperators.test_inplace_andc CsJ| ddddddgdddd d g¡| tjdddddgdddd d g¡dS)Nz|=rrSryr_rÉrÊrÒrËrÌ)rrÚiorrnr%r%r&Útest_inplace_ors"z"TestArrayOperators.test_inplace_orc CsJ| ddddddgdddd d g¡| tjdddddgdddd d g¡dS)Nz^=rrSryr_rÉrÊrÒrËrÌ)rrÚixorrnr%r%r&Útest_inplace_xors"z#TestArrayOperators.test_inplace_xorcCsB| dddddgddddg¡| tjddddgddddg¡dS) Nz<<=rr‰éöÿÿÿéÍÿÿÿrSré)r¬rÚilshiftrnr%r%r&Útest_inplace_lshift"sz&TestArrayOperators.test_inplace_lshiftcCsB| dddddgddddg¡| tjddddgddddg¡dS) Nz>>=rr‰rÓrÔrSrrÕ)r¬rÚirshiftrnr%r%r&Útest_inplace_rshift&sz&TestArrayOperators.test_inplace_rshiftcCsœdd„}t d¡}||ƒ}| |d|dko<|d|dk¡t d¡}t|ƒ|ƒ}| |d|dkox|d|dk¡tj ||¡tj ||¡dS)z¶ Verify that the unary positive operator copies values, and doesn't just alias to the input array (mirrors normal Numpy/Python interaction behavior). cSs| }d|d<d|d<|S)Nr_rrrSr%)rmÚa2r%r%r&r½1sz:TestArrayOperators.test_unary_positive_array_op..fr”rrSN)rJrÚ assertTruerrrr)ror½rmrÚÚa3Za4r%r%r&r±*s & &cCs| d¡dSr®©r˜rnr%r%r&Útest_add_array_opCsz$TestArrayOperators.test_add_array_opcCs| d¡dSr²rÝrnr%r%r&Útest_subtract_array_opFsz)TestArrayOperators.test_subtract_array_opcCs| d¡dS)NÚ*rÝrnr%r%r&Útest_multiply_array_opIsz)TestArrayOperators.test_multiply_array_opcCsd}tj}|jd|ddS)Nú/)rs)rrgr˜)rorºr%r%r&Útest_divide_array_opLsz'TestArrayOperators.test_divide_array_opcCsdt d¡tjft d¡tjft d¡tjft d¡tjft d¡tjft d¡tjft d¡tjft d¡tjft d¡tjft d ¡tjftjdd gddt tjdd ¡ftjddgddt tjdd ¡ftjdddgddt tjdd ¡ftjdddgddt tjdd ¡ftjddgddt tjdd ¡ftjdd gddt tjdd ¡fg|_ | d¡dS)NrSéþÿÿÿrrr¸rUr·gÀg@ryrXrYrZr_r[rTr‰r\ér]r^r_z//)rJrbrrcrdrerfrgrhrirlr˜rnr%r%r&Útest_floor_divide_array_opQs$ "" íz-TestArrayOperators.test_floor_divide_array_opcCs| d¡dS)Nú%rÝrnr%r%r&Útest_remainder_array_opksz*TestArrayOperators.test_remainder_array_opcCs|jddddS)Nz**T©r—rÝrnr%r%r&Útest_power_array_opnsz&TestArrayOperators.test_power_array_opcCs|jddddS)Nz<>Trérërnr%r%r&Útest_right_shift_array_optsz,TestArrayOperators.test_right_shift_array_opcCs| d¡dS)Nú&©ržrnr%r%r&Útest_bitwise_and_array_opwsz,TestArrayOperators.test_bitwise_and_array_opcCs| d¡dS)Nú|rïrnr%r%r&Útest_bitwise_or_array_opzsz+TestArrayOperators.test_bitwise_or_array_opcCs| d¡dS)Nú^rïrnr%r%r&Útest_bitwise_xor_array_op}sz,TestArrayOperators.test_bitwise_xor_array_opcCs| d¡dS)Nz==rÝrnr%r%r&Útest_equal_array_op€sz&TestArrayOperators.test_equal_array_opcCs| d¡dS)Nú>rÝrnr%r%r&Útest_greater_array_opƒsz(TestArrayOperators.test_greater_array_opcCs| d¡dS)Nz>=rÝrnr%r%r&Útest_greater_equal_array_op†sz.TestArrayOperators.test_greater_equal_array_opcCs| d¡dS)Nú.rScSsg|]}|tjk‘qSr%)rrgrþr%r%r&r0ÆsrTÚexact)rrU)rOrÚ_compile_flagsr“rQr,ÚsetrrcrdreÚfloatrJZ issubdtyperKZinexactÚintegerÚintr9ÚboolÚ signaturer!Úreturn_typerrZ)ror¥Z arg_typesZ arg_valuesrPr!r©rªrr¬rÚspecialZalltypesrUr%r%r&Ú run_ufuncŸs2 ÿ zTestScalarUFuncs.run_ufunccCsRdd„}ddddddg}tjftjftjftjftjftjfg}| |||¡dS)NcSs t |¡Srr‰©rCr%r%r&Ú_funcÏsz7TestScalarUFuncs.test_scalar_unary_ufunc.._func)ry)rS)çš™™™™™¹?)çš™™™™™É?)rrcrbrerdrfrgr ©rorÚvalsÚtysr%r%r&Útest_scalar_unary_ufuncÎsþz(TestScalarUFuncs.test_scalar_unary_ufunccCsRdd„}ddddddg}tjtjtjtjtjtjg}| |t||ƒt||ƒ¡dS)NcSst ||¡SrrjrBr%r%r&rÚsz@TestScalarUFuncs.test_scalar_binary_uniform_ufunc.._funcryrSr r) rrcrbrerdrfrgr rOrr%r%r&Ú test_scalar_binary_uniform_ufuncÙsÿz1TestScalarUFuncs.test_scalar_binary_uniform_ufunccCsVdd„}ddddddg}tjtjtjtjtjtjg}| |t ||¡t ||¡¡dS)NcSst ||¡SrrjrBr%r%r&räsz>TestScalarUFuncs.test_scalar_binary_mixed_ufunc.._funcryrSr r) rrcrbrerdrfrgr rVrW)ror|rrrr%r%r&Útest_scalar_binary_mixed_ufuncãsþ ÿz/TestScalarUFuncs.test_scalar_binary_mixed_ufuncN) r5rwrxÚ__doc__Úenable_pyobj_flagsrr rrrr%r%r%r&rý“s / rýc@seZdZdZeZdS)ÚTestScalarUFuncsNoPythonz:Same tests as TestScalarUFuncs, but forcing no python modeN)r5rwrxrrerr%r%r%r&rïsrc@s6eZdZdd„Ze ejdkd¡dd„ƒZdd„Z d S) ÚTestUfuncIssuescCsPtdgƒdd„ƒ}tjddd}tjddd}| |||ƒ||||¡dS)Nz(float64,float64)cSst ||¡t ||¡Srrj)Úx1Zx2r%r%r&Úfooøsz+TestUfuncIssues.test_issue_651..foor”r_rY)r rJrbrZ)rorÚaÚbr%r%r&Útest_issue_651ös zTestUfuncIssues.test_issue_651)rSéz4Complex floor division support removed in NumPy 1.22cCs8dd„}t|tjtjgƒ}| |ddƒ| dd¡¡dS)NcSst ||¡Sr)rJrÂrBr%r%r&rsz+TestUfuncIssues.test_issue_713..fooùð?)rrZ complex128r€r“)rorr©r%r%r&Útest_issue_713szTestUfuncIssues.test_issue_713csˆdd„}|‰tddˆƒ‰‡‡‡fdd„}t ddd g¡t d ddg¡g}|D]4}||d ƒ||t d ¡ƒ||t ddd g¡ƒqNdS) zF should return float32, not float64. cSst ||¡Sr)rJrÇrBr%r%r&r sz,TestUfuncIssues.test_issue_2006..fooT©Znopythoncs2ˆ||ƒ}tj |ˆ||ƒ¡ˆ |j|j¡dSr)rJrržr€rK)rCrDr©rªr¥ror%r&Úchecks z.TestUfuncIssues.test_issue_2006..checkrSryr_ry@ÀN)rrJrfrYrdre)rorr#ÚxsrCr%r"r&Útest_issue_2006 s zTestUfuncIssues.test_issue_2006N) r5rwrxrrZskipIfrZ numpy_versionr r%r%r%r%r&rôs ÿ rc@sleZdZdZdZdddddddœZddd „Zd d„Zdd „Zdd„Z e dd„ƒZe dd„ƒZe dd„ƒZ dS)Ú_LoopTypesTesteraìTest code generation for the different loop types defined by ufunc. This test relies on class variables to configure the test. Subclasses of this class can just override some of these variables to check other ufuncs in a different compilation context. The variables supported are: _funcs: the ufuncs to test _compile_flags: compilation flags to use (to force nopython mode) _skip_types: letter types that force skipping the loop when testing if present in the NumPy ufunc signature. _supported_types: only test loops where all the types in the loop signature are in this collection. If unset, all. Note that both, _skip_types and _supported_types must be met for a loop to be tested. The NumPy ufunc signature has a form like 'ff->f' (for a binary ufunc loop taking 2 floats and resulting in a float). In a NumPy ufunc object you can get a list of supported signatures by accessing the attribute 'types'. ZOegGryrr‰))rru)rÚD)rru)rîr')rru)r÷ÚdrcCs0|dkrtjddddg|dS|dkrssz/_LoopTypesTester._check_loop..Ú_skip_typesc3s|]}|ˆkVqdSrr%r7)Ú skip_typesr%r&r:vsÚ_required_typesc3s|]}|ˆkVqdSrr%r7)Úletter_typesr%r&r:{sÿ)rŽr‘rXrÚ_check_ufunc_with_dtypes)ror1r8ÚloopZrequired_typesr%)r>r<r9r&Ú_check_loopjsÿÿz_LoopTypesTester._check_loopcsdd„|Dƒ}dd„|Dƒ}t||ˆjd}‡fdd„t|ƒDƒ}|D]}ˆj |¡qFdd„|Dƒ} |j|Ž|| Žt|| |ƒD]’\} }}ˆ | ||¡\}}|jj } ˆj |j| fd¡}| dkrÈd nd }| dkrØdn|}d ddddg¡}| |j||||¡}ˆj|||||dq„dS)NcSsg|]}t |¡‘qSr%)rJrKrþr%r%r&r0‚sz=_LoopTypesTester._check_ufunc_with_dtypes..cSsg|]}t t|ƒdd¡‘qS)rSrZ)rrirrþr%r%r&r0ƒsr{cs$g|]\}}ˆj||d d¡‘qS)©r5ry)r6Úrepeat)r.r5rÿrnr%r&r0‡sÿcSsg|]}| ¡‘qSr%)r¦)r.rr%r%r&r0‹srSZfFrSrZdDrTrˆz ufunc '{0}' arrays differ ({1}):z args: {2}zexpected {3}zgot {4})rUrÚulps)rrÚ enumerater•Úshuffler“rOÚ_fixup_resultsrKÚcharÚ_ulpsÚgetr5r2r,rZ)ror1r8ZdtypesZarg_dtyZarg_nbtyr©Zc_argsZarrZpy_argsrKÚpy_argÚc_argZtypecharrDrUrr%rnr&r?s4 ÿ ÿÿz)_LoopTypesTester._check_ufunc_with_dtypescCs||fSrr%)rorKrKrLr%r%r&rGžsz_LoopTypesTester._fixup_resultscCs|jD]}| ||¡qdSr)rÚ_inject_test)Úclsr8r@r%r%r&Ú_check_ufunc_loops¡s z#_LoopTypesTester._check_ufunc_loopsc s0‡‡fdd„}t|d ˆjˆ dd¡¡|ƒdS)Ncstˆƒ}| |ˆˆ¡dSr)r:rA)ror1©r@r8r%r&Ú test_template¨sz4_LoopTypesTester._inject_test..test_templateztest_{0}_{1}z->Ú_)Úsetattrr,r5Úreplace)rNr8r@rQr%rPr&rM¦s ÿþz_LoopTypesTester._inject_testcCs|jD]}| |¡qdSr)Ú_ufuncsrO)rNr8r%r%r&Úautogenerate¯s z_LoopTypesTester.autogenerateN)r)r5rwrxrr;rIr6rAr?rGÚclassmethodrOrMrVr%r%r%r&r&!s&û ( r&c@sreZdZeZedd…Ze ej ¡e ej ¡e ej¡e ej¡e ej ¡e ej¡dZdejZdS)ÚTestLoopTypesIntNoPythonNz?bBhHiIlLqQÚfdFDmMO)r5rwrxrerrrUÚremoverJrÇrùÚ left_shiftÚright_shiftr´rÄr=r&r;r%r%r%r&rXµsrXc@s.eZdZeZejejgZdZ de jdZdS)Ú(TestLoopTypesSubtractAndNegativeNoPythonz?bBhHiIlLqQfdFDÚmMOr)N)r5rwrxrerrJr´rÄrUr=r&r;r%r%r%r&r]Æsr]cs8eZdZeZejgZdZde j Z d‡fdd„ Z‡ZS)ÚTestLoopTypesReciprocalNoPythonÚbBhHiIlLqQfdFDr^rcs.t|j|ƒj||d}|dkr*d||dk<|S)NrBÚ bBhHiIlLqQrËr©r`rqr6©ror4r5Úresrpr%r&r6Õsÿz-TestLoopTypesReciprocalNoPython._arg_for_type)r) r5rwrxrerrJrùrUr=r&r;r6ryr%r%rpr&r_Ïs r_cs8eZdZeZejgZdZde j Z d‡fdd„ Z‡ZS)ÚTestLoopTypesPowerNoPythonr`r^rcs6t|j|ƒj||d}|dkr2|dkr2d||dk<|S)NrBrarSr_rrbrcrpr%r&r6èsÿz(TestLoopTypesPowerNoPython._arg_for_type)r) r5rwrxrerrJrÇrUr=r&r;r6ryr%r%rpr&reâs recs8eZdZeZejgZdZde j Z d‡fdd„ Z‡ZS)Ú!TestLoopTypesIntLeftShiftNoPythonrarYrcs@t|j|ƒj||d}|dkr<|jjd}t |d|d¡}|S©NrBrSér)r`rqr6rKÚitemsizerJÚclip©ror4r5rdZ bit_countrpr%r&r6ûsÿz/TestLoopTypesIntLeftShiftNoPython._arg_for_type)r) r5rwrxrerrJr[rUr=r&r;r6ryr%r%rpr&rfõs rfcs8eZdZeZejgZdZde j Z d‡fdd„ Z‡ZS)Ú"TestLoopTypesIntRightShiftNoPythonrarYrcsRt|j|ƒj||d}|dkr<|jjd}t |d|d¡}|dkrNt |¡}|Srg)r`rqr6rKrirJrjr‹rkrpr%r&r6sÿ z0TestLoopTypesIntRightShiftNoPython._arg_for_type)r) r5rwrxrerrJr\rUr=r&r;r6ryr%r%rpr&rls rlc@s6eZdZeZejejejgZ dZ dejZdd„Z dS)Ú TestLoopTypesFloorDivideNoPythonr`r^cCsF|jdkr>t |¡t |¡@}||dk|dk@O}||||<||fS)Nr½rV)rrJr>r@)rorKrKrLÚpredr%r%r&rG4s z/TestLoopTypesFloorDivideNoPython._fixup_resultsN)r5rwrxrerrJrÂrÐÚdivmodrUr=r&r;rGr%r%r%r&rm.s rmc@sjeZdZeZedd…Zer(e e j ¡e e j¡e e j¡e e j ¡e e j¡dZdejZdS)ÚTestLoopTypesFloatNoPythonNr.ZFDmMO)r5rwrxrerrrUÚ iswindowsrZrJrBrÂrÐrorÔr=r&r;r%r%r%r&rpAsrpc@s*eZdZeZedd…ZdZdej Z dS)ÚTestLoopTypesComplexNoPythonNr0r^) r5rwrxrerrrUr=r&r;r%r%r%r&rrPsrrc@sdeZdZeZedd…Ze ej ¡dZ dd„Zdd„Zdd„Z d d „Zdd„Zd d„Zdd„ZdS)ÚTestLoopTypesDatetimeNoPythonNZmMc CsŒtj}t|ƒ}| ||dddg¡| ||dddg¡| ||dddg¡| ||dddg¡| t¡| ||dddg¡W5QRXdS)Núm8[s]úm8[m]úm8[ms])rJr±r:r?ror©ror8r1r%r%r&Útest_addosz&TestLoopTypesDatetimeNoPython.test_addc CsŒtj}t|ƒ}| ||dddg¡| ||dddg¡| ||dddg¡| ||dddg¡| t¡| ||dddg¡W5QRXdS)NúM8[s]úM8[m]rtrvru)rJr´r:r?rorrwr%r%r&Ú test_subtract|sz+TestLoopTypesDatetimeNoPython.test_subtractc Csdtj}t|ƒ}| ||dddg¡| ||dddg¡| t¡| ||dddg¡W5QRXdS)NrtÚqzm8[us]ru)rJr¶r:r?rorrwr%r%r&Ú test_multiply‰sz+TestLoopTypesDatetimeNoPython.test_multiplyc CsŒtj}t|ƒ}| ||dddg¡| ||dddg¡| ||dddg¡| ||dddg¡| t¡| ||dddg¡W5QRXdS)Nrurtr(r|)rJrÀr:r?rorrwr%r%r&Útest_true_divide“sz.TestLoopTypesDatetimeNoPython.test_true_dividec Csdtj}t|ƒ}| ||dddg¡| ||dddg¡| t¡| ||dddg¡W5QRXdS)Nrur|rtr()rJrÂr:r?rorrwr%r%r&Útest_floor_divide sz/TestLoopTypesDatetimeNoPython.test_floor_dividecCs\t|ƒ}| ||dddg¡| ||dddg¡| ||dddg¡| ||dddg¡dS)Nrurtr)rzry)r:r?rwr%r%r&Ú_check_comparisonªs z/TestLoopTypesDatetimeNoPython._check_comparisoncCs0tjtjtjtjtjtjfD]}| |¡qdSr)rJr+r)r%r'r!r#r€)ror8r%r%r&Útest_comparisons³s ÿz.TestLoopTypesDatetimeNoPython.test_comparisons)r5rwrxrerrrUrZrJror=rxr{r}r~rr€rr%r%r%r&rs\s rsc@s(eZdZeZdd„Zdd„Zdd„ZdS)ÚTestUFuncBadArgsNoPythoncCs*dd„}|jtt|tjgtj|jddS)NcSst |¡}|S)zerror: np.add requires two argsrj)rCr]r%r%r&r$¿s z8TestUFuncBadArgsNoPython.test_missing_args..func©rr|©rorrrrgr©ror$r%r%r&Útest_missing_args¾s ÿz*TestUFuncBadArgsNoPython.test_missing_argscCs:dd„}t tjdd¡}|jtt||gd||jddS)NcSst ||||¡}|S)zerror: too many argsrj)rCÚoutZout2r]r%r%r&r$Ész9TestUFuncBadArgsNoPython.test_too_many_args..funcrSrZr_rƒ)rrirgrorrr)ror$Z array_typer%r%r&Útest_too_many_argsÈsÿz+TestUFuncBadArgsNoPython.test_too_many_argscCs*dd„}|jtt|tjgtj|jddS)NcSsd}t |||¡dS)z0error: scalar as a return value is not supportedrNrjrBr%r%r&r$ÓszITestUFuncBadArgsNoPython.test_no_scalar_result_by_reference..funcrƒr„r…r%r%r&Ú"test_no_scalar_result_by_referenceÒs ÿz;TestUFuncBadArgsNoPython.test_no_scalar_result_by_referenceN)r5rwrxrerr†rˆr‰r%r%r%r&r‚»s r‚c@seZdZdd„ZdS)Ú TestUFuncCompilationThreadSafetycsfg‰tdd„ƒ‰‡‡fdd„‰‡fdd„tdƒDƒ}|D]}| ¡q8|D]}| ¡qJ| ˆ¡dS) z{ Test that (lazy) compiling from several threads at once doesn't produce errors (see issue #2403). cSs|dS)NrSr%rr%r%r&rãsz7TestUFuncCompilationThreadSafety.test_lock..fooc slz:tjdtjd}tjdtjdd}tj ˆ|ƒ|¡Wn,tk rf}zˆ |¡W5d}~XYnXdS)N)r”rYgð?)rJZonesrgrrrÚ ExceptionÚappend)rr¬Úe)Úerrorsrr%r&r'çsz;TestUFuncCompilationThreadSafety.test_lock..wrappercsg|]}tjˆd‘qS))Útarget)Ú threadingÚThreadr-)r'r%r&r0ïsz>TestUFuncCompilationThreadSafety.test_lock..rÒN)r r3Ústartr2ZassertFalse)roÚthreadsrÿr%)rŽrr'r&Ú test_lockÜs z*TestUFuncCompilationThreadSafety.test_lockN)r5rwrxr”r%r%r%r&rŠÚsrŠc@seZdZdd„Zdd„ZdS)ÚTestUfuncOnContextc Csrtj}| tj¡}| |t¡t tj¡}| ||¡t ƒ}| t¡}t |¡W5QRX| |jj |f¡dSr)rZtarget_contextÚget_ufunc_inforJr±ZassertIsInstanceÚdictrr€ÚobjectroÚKeyErrorÚ exceptionr!)roÚ targetctxZadd_infor¬ZbadkeyÚraisesr%r%r&Útest_cpu_get_ufunc_infoøsz*TestUfuncOnContext.test_cpu_get_ufunc_infoc CsDttjdƒ}| t¡}| tj¡W5QRX| t |j ƒd¡dS)NÚcpuz< does not support ufunc)rrZtyping_contextroÚNotImplementedErrorr–rJr±ÚassertRegexrHrš)ror›rœr%r%r&Útest_base_get_ufunc_infosþz+TestUfuncOnContext.test_base_get_ufunc_infoN)r5rwrxrr¡r%r%r%r&r•÷sr•c@seZdZdd„ZdS)ÚTestUfuncWriteInputcCsptdgddddd„ƒ}tj dd¡}tj|tjd }|||ƒtj tjdd d d dgtj d |j dd¡dS) Nzvoid(float64[:], uint8[:])z(n)->(n)Tr!cSs(t|jƒD]}|ddkr d||<q dS)NrrrS)r3r€)rCr‡r/r%r%r&r$sz6TestUfuncWriteInput.test_write_input_arg..funcr”r‰rYFr)Zaxis)rrJr•ZrandZ zeros_likerjrrrrhr9r)ror$rCr‡r%r%r&Útest_write_input_args þz(TestUfuncWriteInput.test_write_input_argN)r5rwrxr£r%r%r%r&r¢sr¢Ú__main__)Wr(rVÚrerr•rrZnumpyrJrZnumbarrrZ numba.corerrrZnumba.core.compilerrr r Znumba.np.numpy_supportrrr Znumba.core.errorsrrrZnumba.tests.supportrrrrZnumba.core.typing.npydeclrrZnumba.nprZnumba.core.registryrZnumba.core.baserrrÚ__itemsize__Zis32bitsr2rœrqrZenable_pyobjectZno_rewritesrerqZnrtr*r:r@rArIrQrRrzrŒrýrrr&rXrVr]r_rerfrlrmrprrrsr‚rŠr•r¢r5Úmainr%r%r%r&Úsš >?7\- \