o gi#@s|ddlZddlZddlZddlmZmZddlmZm Z ddl Z Gddde j Z Gddde j Z edkr.fnvgh㈵>iN)rtimesleepnparangetesting assert_equal)selfr sleep_timer r r r r test_thread_responses   z'TestParUfuncIssues.test_thread_responsec stdtj}dfdd}||tdgdddfd d }d D]a}tj|tjd }|}t }||}Wdn1sCwY| } | t t d d| } ddt|D} | fddt|D7} t | } || | tj|d|q#dS)z< Testing issue #1998 due to GIL reacquiring N abcdefghijct|dSr printx charactersr r bar,z;TestParUfuncIssues.test_gil_reacquire_deadlock..barz int32(int32)rTrnopythoncs t|d|d|dS)N rrrcbarr r foo2s  z;TestParUfuncIssues.test_gil_reacquire_deadlock..foorr&didtypecS|Sr striprr r r Ez@TestParUfuncIssues.test_gil_reacquire_deadlock..cSg|]}t|dqSr&str.0rr r r GzBTestParUfuncIssues.test_gil_reacquire_deadlock..cg|]}|dqSr4r r7r r r r9Hr:r)ctypes CFUNCTYPEc_int32rrrint32copyrgetvalueclosesortedmap splitlinesrange assertEqualrr rprotor"r)nelemr acopybufgotstdout got_outputexpected_outputr r(r!r test_gil_reacquire_deadlock#s*   z.TestParUfuncIssues.test_gil_reacquire_deadlockN)__name__ __module__ __qualname___numba_parallel_test_rrrRr r r r r s rc@seZdZdZeddZdS)TestParGUfuncIssuesFc stdtj}dfdd}||tdgdddd fd d }d D]a}tj|tjd }|}t }||}Wdn1sDwY| } | t t dd| } ddt|D} | fddt|D7} t | } || | tj|d|q$dS)zR Testing similar issue to #1998 due to GIL reacquiring for Gufunc Nrcrr rrr r r r"]r#z.barz(int32, int32[:])z()->()rTr$cs(t|d|d|d|d<dS)Nr&rrr)routr'r r r)cs  z.foor*r,cSr.r r/rr r r r1wr2zATestParGUfuncIssues.test_gil_reacquire_deadlock..cSr3r4r5r7r r r r9yr:zCTestParGUfuncIssues.test_gil_reacquire_deadlock..cr;r4r r7r r r r9zr:r)r<r=r>rrrr?r@rrArBrCrDrErFrGrrrHr rQr rRTs.   z/TestParGUfuncIssues.test_gil_reacquire_deadlockN)rSrTrUrVrrRr r r r rWPsrW__main__)rr<numpyrnumba.tests.supportrrnumbarrunittestTestCaserrWrSmainr r r r sE1