o OZh@s|ddlZddlmZmZmZmZmZddlmZm Z m Z ddl Z Gdddej Z Gdddej Zedkr.fcSr )Nrrrrrf2rzCWorkerQueueTypeTest.test_usage_with_different_functions..f2r r)rrdowaitstop assertEqualresult)selfrrwqfrZfr2rrr#test_usage_with_different_functions s  z7WorkerQueueTypeTest.test_usage_with_different_functionscCsdS)zSTests function placement on queue and execution after blocking function completion.Nrrrrrtest_dozWorkerQueueTypeTest.test_docCst}|t|jd|jD] }||qtdD] }|dd|q||jD] }| |q/| |j t dS)z$Ensure stop() stops the worker queuercSr r rrrrrCz/WorkerQueueTypeTest.test_stop..N)rZ assertGreaterlenpool assertTrueis_aliverangerr assertFalseassertIsqueuegetr)rrtirrr test_stop9s   zWorkerQueueTypeTest.test_stopcCs`td}||jg}||jdtd||dd|||j d dS)Nr g?r) rrZ threadloopappendtimesleeprrr+r'r))rrlrrrtest_threadloopLs  z#WorkerQueueTypeTest.test_threadloopcCsHt}tdD] }|dd|q||t|jj|dS)NicSr r rrrrrr$or%z/WorkerQueueTypeTest.test_wait..) rr*rr assertRaisesrr- get_nowaitr)rrr0rrr test_waitgs   zWorkerQueueTypeTest.test_waitN)__name__ __module__ __qualname__rr!r1r6r9rrrrr s  r c@sDeZdZddZddZddZddZd d Zd d Zd dZ dS)ThreadsModuleTestcCsn t}||t|d|kodkndd}ddtdddD}t||}||tdS) zfEnsure benchmark_workers performance measure functions properly with both default and specified inputsr@cSst|ddS)N)r?)r Z smoothscale)datarrrsmooth_scale_benchszDThreadsModuleTest.test_benchmark_workers..smooth_scale_benchcSsg|] }t||fddqS)r )r.0rrrr sz. N)rZbenchmark_workersassertIsInstanceintr(r*)rZoptimal_workersrAZ surf_dataZbest_num_workersrrrtest_benchmark_workersxs  z(ThreadsModuleTest.test_benchmark_workerscCs$td|tjttdS)z&Ensure init() sets up the worker queueN)rinitrG_wqrquitr rrr test_inits  zThreadsModuleTest.test_initcCs"tdt|tjdS)z(Ensure quit() cleans up the worker queuerJN)rrKrM assertIsNonerLr rrr test_quits zThreadsModuleTest.test_quitcCsxddtd}}tt||}tt||}|||td}dd}tt||dd}|tdd|DdS) NcSr r rrrrrr$r%z-ThreadsModuleTest.test_tmap..dcSsddS)Nr rrrrrrr$r%F)Z stop_on_errorcss|]}|duVqdSNrrCrrr sz.ThreadsModuleTest.test_tmap..)r*listrmaprr(all)rfuncr@ZtmappedZmappedZdata2Zalways_exceptsZtmapped2rrr test_tmaps  zThreadsModuleTest.test_tmapcCs|tdgd}tdgdgd}tdgdgd}tdgdgd}|gd||gd||gd||gd|dS) z+Using a None as func and multiple sequencesN)r rr)!,7)rZr[r\r]B)r rrrY)r rZrr[rr\rYr])r`rarbrc)Nr^)r`rarbrc)r_N)Zfailrr)rresZres2Zres3Zres4rrr0todo_test_tmap__None_func_and_multiple_sequencesszBThreadsModuleTest.todo_test_tmap__None_func_and_multiple_sequencescCsNtd}tdd|ddd\}}|dd|D}|t|t|dS) NicSs|SrRrrrrrr$r"z3ThreadsModuleTest.test_tmap__wait..r_F)Z num_workersrcss|]}|jVqdSrR)rrCrrrrSsz4ThreadsModuleTest.test_tmap__wait..)r*rrrrT)rrrresultsr2rrrtest_tmap__waits z!ThreadsModuleTest.test_tmap__waitcsftdd}|d||jd||jdtdfdd}t|}|d ||jd S) z:Ensure FuncResult sets its result and exception attributescSr r rrrrrr$r%z3ThreadsModuleTest.test_FuncResult..rrzno exception should be raisedZrastcsrRr)Zsdf exceptionrrrsz,ThreadsModuleTest.test_FuncResult..xN)rrrrOrk ValueErrorr,)rrrrrjrtest_FuncResults  z!ThreadsModuleTest.test_FuncResultN) r:r;r<rIrNrPrXrerirmrrrrr=ws  r=__main__)ZunittestZpygame.threadsrrrrrZpygamerrr r3ZTestCaser r=r:mainrrrrsnv