o i @s<ddlZddlmZGdddeZedkredSdS)N)TestCasec@seZdZddZddZdS)DocsJitclassUsageTestcs ddlddlm}m}ddlm}d|fd|ddfg}||Gfdddt}d}||}|t||| |j |j |j j|jd | |j|j |d d j|jd j |d d j|jd | |d d d| |d dd dS)Nr)int32float32jitclassvaluearraycs8eZdZfddZeddZddZeddZd S) z3DocsJitclassUsageTest.test_ex_jitclass..Bagcs||_j|jd|_dS)Ndtype)rzerosrr )selfrnpd/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/numba/tests/doc_examples/test_jitclass.py__init__sz.Bag.__init__cSs|jjSN)r sizer rrrrz8DocsJitclassUsageTest.test_ex_jitclass..Bag.sizecSs(t|jD] }|j||7<q|jSr)rangerr )r valirrr incrementsz=DocsJitclassUsageTest.test_ex_jitclass..Bag.incrementcSs||Srr)xyrrradd$rz7DocsJitclassUsageTest.test_ex_jitclass..Bag.addN) __name__ __module__ __qualname__rpropertyrr staticmethodrrrrrBags  r#r  )numpynumbarrnumba.experimentalrobject assertTrue isinstanceassertPreciseEqualrtestingassert_allcloser r rronesr)r rrrspecr#nmybagrrrtest_ex_jitclass s, z&DocsJitclassUsageTest.test_ex_jitclasscsddlmddlm}ddlm}|Gddd|Gfddd}|gd}||}td D]"}||jj|| | ||t |||jj|d q5dS) Nr)Listrc@s,eZdZUeed<ddZdefddZdS)zBDocsJitclassUsageTest.test_ex_jitclass_type_hints..CounterrcSs d|_dS)NrrrrrrrAs zKDocsJitclassUsageTest.test_ex_jitclass_type_hints..Counter.__init__returncSs|j}|jd7_|S)Nr(r9)r retrrrgetDszFDocsJitclassUsageTest.test_ex_jitclass_type_hints..Counter.getN)rrr int__annotations__rr<rrrrCounter=s r?csFeZdZUed<eed<deffdd ZdefddZdS) zKDocsJitclassUsageTest.test_ex_jitclass_type_hints..ListLoopIteratorcounteritemscs||_|_dSr)rAr@)r rA)r?rrrNs zTDocsJitclassUsageTest.test_ex_jitclass_type_hints..ListLoopIterator.__init__r:cSs|jt|j}|j|Sr)r@r<lenrA)r idxrrrr<Rs zODocsJitclassUsageTest.test_ex_jitclass_type_hints..ListLoopIterator.getN)rrr r>floatrr<rr?r8rrListLoopIteratorIs  rF)gQ @gX9v@grh|?g r() typingr8r,r numba.typedr assertEqualr@rassertAlmostEqualr<rB)r r NumbaListrFrAloop_itrrCrrErtest_ex_jitclass_type_hints7s      z1DocsJitclassUsageTest.test_ex_jitclass_type_hintsN)rrr r7rNrrrrrs -r__main__)unittestnumba.tests.supportrrrmainrrrrs  X