U id@sbddlZddlZddlmZmZddlmZmZm Z ddl Z Gddde j Z e dkr^e dS)N)captured_stdoutoverride_config)njit vectorize guvectorizec@s<eZdZddZddZddZddZd d Zd d Zd S) TestFastMathcCsdd}tdd|}t|}||d|d||jd}||jd}|d||d||d||d|dS) NcSs|t|SNmathsinxrR/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/tests/test_fastmath.pyfoo sz"TestFastMath.test_jit..fooTZfastmath?r fadd fast call fast)r assertEqual inspect_llvm signaturesassertIn assertNotInselfrfastfooslowfoofastllvmslowllvmrrrtest_jit s   zTestFastMath.test_jitcCsZdd}tddhd|}tdhd|}||dtjd|t|dtjdS)NcSs |||Srr)r yrrrrsz3TestFastMath.test_jit_subset_behaviour..fooreassocnszrr)rrnpinf assertTrueisnan)rrrrrrrtest_jit_subset_behaviours z&TestFastMath.test_jit_subset_behaviourcCsdd}tddhd|}t|}||d|d||jd}||jd}|d||d ||d ||d ||d ||d ||d |kpd |k||d |kpd |k|dS)NcSs|t|Srr r rrrr!sz.TestFastMath.test_jit_subset_code..foor"r#rrrrrzfadd reassoc nszzcall reassoc nszzfadd nsz reassoczcall nsz reassoc)rrrrrr&rrrrtest_jit_subset_code s,       z!TestFastMath.test_jit_subset_codec Cs|t}tdhddddW5QRX|dt|j|t }tddiddddW5QRX|dt|j|t}td dd ddW5QRX|d t|jdS) NZspqrrcSs|dSNrr rrr<z5TestFastMath.test_jit_subset_errors..r+zUnrecognized fastmath flags:FcSs|dSr*rr rrrr,Cr-i9cSs|dSr*rr rrrr,Jr-z!Expected fastmath option(s) to be) assertRaises ValueErrorrrstr exception)rZraisesrrrtest_jit_subset_errors:s$ " $  z#TestFastMath.test_jit_subset_errorsc Csdd}tdd|}t|}tjdtj}tddJt}||}W5QRX|}t}||} W5QRX|} W5QRXtj || | d| | d| | d|| d|dS) NcSs|t|Srr r rrrrQsz(TestFastMath.test_vectorize..fooTrDUMP_OPTIMIZEDrr) rr$randomastypefloat32rrgetvaluetestingassert_almost_equalrr) rrrrr slow_capexpectrfast_capgotrrrrtest_vectorizePs     zTestFastMath.test_vectorizec Csdd}tjdtj}tddddg}d}t}t||dd|}W5QRX|}t}t|||} W5QRX|} W5QRX| |} ||} tj | | | d || d || d | | d | dS) NcSs|t||d<dS)Nrr )r outrrrresz*TestFastMath.test_guvectorize..foor3r4Tz(float32, float32[:])z()->()rrr) r$r5r6r7rrrr8r9r:rr) rrr typessigr=rrr;rrr<r>rrrtest_guvectorizeds$    zTestFastMath.test_guvectorizeN) __name__ __module__ __qualname__r r(r)r2r?rCrrrrr s r__main__)r numpyr$Znumba.tests.supportrrZnumbarrrZunittestZTestCaserrDmainrrrrsq