o i@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*rrZ assertRaisesrr- get_nowaitr)rrr0rrr test_waitgs   zWorkerQueueTypeTest.test_waitN)__name__ __module__ __qualname__rr!r1r6r8rrrrr 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_workersr@Z surf_dataZbest_num_workersrrrtest_benchmark_workersxs  z(ThreadsModuleTest.test_benchmark_workerscCs$td|tjttdS)z&Ensure init() sets up the worker queueN)rinitrF_wqrquitr rrr test_inits  zThreadsModuleTest.test_initcCs"tdt|tjdS)z(Ensure quit() cleans up the worker queuerIN)rrJrL assertIsNonerKr 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|]}|duVqdSNrrBrrr 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)rYrZr[r\B)r rrrX)r rYrrZrr[rXr\)r_r`rarb)Nr])r_r`rarb)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|SrQrrrrrr$r"z3ThreadsModuleTest.test_tmap__wait..r^F)Z num_workersrcss|]}|jVqdSrQ)rrBrrrrRsz4ThreadsModuleTest.test_tmap__wait..)r*rrrrS)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 raisedZrastcsrQr)Zsdf exceptionrrrsz,ThreadsModuleTest.test_FuncResult..xN)rrrrNrj ValueErrorr,)rrrrrirtest_FuncResults  z!ThreadsModuleTest.test_FuncResultN) r9r:r;rHrMrOrWrdrhrlrrrrr<ws  r<__main__)ZunittestZpygame.threadsrrrrrZpygamerrr r3ZTestCaser r<r9mainrrrrsnv