B \@sjddlmZddlZddlZddlmZddlm Z ddl m Z m Z Gdddej ZedkrfedS))print_functionN)jit) NumbaWarning deprecatedc@s<eZdZddZddZddZddZd d Zd d Zd S) TestBuiltinsc Csdd}tjdd}tdtt|}|dd|t|d||djt|d t |dj ||djt|d t |dj WdQRXdS) NcSs i}||S)N)xyarr8lib/python3.7/site-packages/numba/tests/test_warnings.pyadd sz1TestBuiltins.test_type_infer_warning..addT)recordalwaysrztype inferencez object mode) warningscatch_warnings simplefilterrr assertEquallencategoryassertInstrmessage)selfr wcfuncrrr test_type_infer_warning s  z$TestBuiltins.test_type_infer_warningc stjdtjdfdd}tjdd}tdttdd |}||t |d ||d j t| d t |d j ||d j t| dt |d j WdQRXdS)N)dtypecsS)Nrr)r rr return_external_array szDTestBuiltins.test_return_type_warning..return_external_arrayT)r rF)Z_nrtrrz return typerz object mode)nponesfloat32rrrrrrrrrrr)rr rrr)r r test_return_type_warnings  z%TestBuiltins.test_return_type_warningc sbtjdtjdfdd}tjdd0}tdtt|}||t |dWd QRXd S) z9 Rerun test_return_type_warning with nrt r)rcsS)Nrr)r rr r 8szMTestBuiltins.test_return_type_warning_with_nrt..return_external_arrayT)r rrN) r!r"r#rrrrrrr)rr rrr)r r !test_return_type_warning_with_nrt2s  z.TestBuiltins.test_return_type_warning_with_nrtc CsVdd}tjdd8}tdtt|dd}|dd|t|d WdQRXdS) NcSs g}||S)Nr)rr r rrr r Esz7TestBuiltins.test_no_warning_with_forceobj..addT)r r)Zforceobjrrr)rrrrrrr)rr rrrrr test_no_warning_with_forceobjDs    z*TestBuiltins.test_no_warning_with_forceobjc Csdd}tjdd}tdttjdtjd}t|}|||t |d||d j t| d t |d j ||d j t| d t |d j ||d j t| d t |d j | dt |d j WdQRXdS)NcSs0i}x&t|jdD]}||d9<qWdS)Nrr)rangeshape)rr irrr do_loopRsz1TestBuiltins.test_loop_lift_warn..do_loopT)r rr)rrztype inferencerrz object modez lifted loops)rrrrr!r"r#rrrrrrr)rr*rrrrrr test_loop_lift_warnQs z TestBuiltins.test_loop_lift_warnc Cstddd}tjddd}td||t|d||djt|d t |dj |dt |dj WdQRXdS) NZfoocSsdS)Nrrrrr barnsz)TestBuiltins.test_deprecated..barT)r rrrr-) rrrrrrrDeprecationWarningrrr)rr-rrrr test_deprecatedms zTestBuiltins.test_deprecatedN) __name__ __module__ __qualname__rr$r%r&r,r/rrrr r s  r__main__)Z __future__rrZnumpyr!Znumba.unittest_supportZunittest_supportZunittestZnumbarZ numba.errorsrrZTestCaserr0mainrrrr s   q