o i@sfddlZddlZddlmZmZddlmZmZm Z ddl Z Gddde j Z e dkr1e dSdS)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) NcS|t|SNmathsinxrh/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/numba/tests/test_fastmath.pyfoo z"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 |||Sr r)ryrrrrs z3TestFastMath.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 |vpYd |v||d |vped |v|dS)NcSrr r r rrrr!rz.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_codecCs|t}tdhddddWdn1swY|dt|j|t}tddiddddWdn1sGwY|dt|j|t}td dd ddWdn1spwY|d t|jdS) NZspqrrcS|dSNrr rrr<z5TestFastMath.test_jit_subset_errors..r.zUnrecognized fastmath flags:FcSr,r-rr rrrr/Cr0i9cSr,r-rr rrrr/Jr0z!Expected fastmath option(s) to be) assertRaises ValueErrorrrstr exception)rZraisesrrrtest_jit_subset_errors:s*   z#TestFastMath.test_jit_subset_errorsc Cs dd}tdd|}t|}tjdtj}tdd>t }||}Wdn1s1wY|}t }||} Wdn1sLwY|} Wdn1s_wYtj || | d| | d| | d|| d|dS) NcSrr r r rrrrQrz(TestFastMath.test_vectorize..fooTrDUMP_OPTIMIZEDrr) rr&randomastypefloat32rrgetvaluetestingassert_almost_equalrr) rrrrrslow_capexpectr!fast_capgotr rrrtest_vectorizePs&       zTestFastMath.test_vectorizec Csdd}tjdtj}tddKdg}d}t}t||dd|}Wdn1s0wY|}t}t|||} Wdn1sNwY|} Wdn1sawY| |} ||} tj | | | d || d || d | | d | dS) NcSs|t||d<dS)Nrr )routrrrresz*TestFastMath.test_guvectorize..foor6r7Tz(float32, float32[:])z()->()rrr) r&r8r9r:rrrr;r<r=rr) rrrtypessigr@rr r>rr!r?rArrrtest_guvectorizeds*      zTestFastMath.test_guvectorizeN) __name__ __module__ __qualname__r"r*r+r5rBrFrrrrr s r__main__)r numpyr&Znumba.tests.supportrrZnumbarrrZunittestZTestCaserrGmainrrrrsq