o 6a @s:ddlZddlZddlmZmZmZmZGdddZdS)N)assert_ assert_raisesassert_array_equal HAS_REFCOUNTc@s4eZdZddZddZddZddZd d Zd S) TestTakec Csddgddgg}ddgddgg}gd }d dg}tjd tjd tjttjd tjd tjd g}dtddd dd ddd }tttdgf}|D]}tjt |tj rW|n||d } t | j } |D]n} | jd kr|d| j| d _d| j| d_|D]T} |D]O} || | }|tur| jd kr| d | tt| j| | ddq| jd kr| d | | j| | dd}t|| |q| j| | dd}t|jd| jkqq~qeqJdS)N1234)raisewrapcliprdtype)rr)rr )i2r )r)modeaxis)npemptyintptuple IndexErrorintobjectrarray issubdtypenumberlistTcopysizeshapeputrtakerr)selfaa_strmodesindices index_arrays real_indicestypesttatresult index_arrayrindex real_indexresr9F/usr/lib/python3/dist-packages/numpy/core/tests/test_item_selection.py test_simple sL       zTestTake.test_simplecCsddtdD}dD]N}t|}tgd}|j||dd|d~tr3ttdd |Dt|d ddd }|j||dd|d~trYttd d |Dq dS) NcSsg|]}tqSr9)r).0ir9r9r: 2sz-TestTake.test_refcounting.. )rrr)rrr r r@)outrcs|] }t|dkVqdSr Nsys getrefcountr<or9r9r: 9z,TestTake.test_refcounting..rcsrCrDrErHr9r9r:rJ?rK)rangerr r)rrall)r*objectsrr+br9r9r:test_refcounting1s zTestTake.test_refcountingcCs*td}dd}tt|jd|ddS)Nr?säUTF8r@)r)rarangedecoder ValueErrorr))r*dkr9r9r:test_unicode_modeAs  zTestTake.test_unicode_modecCs:tgd}|}|tjgtjdt||dS)Nrrr rAr?r)rr r% partitionint16r)r* a_originalr+r9r9r:test_empty_partitionFszTestTake.test_empty_partitioncCs@tgd}|tjgtjd}tgd}t||dS)NrXr)rrrr r r@)rr argpartitionr[r)r*r+rOr9r9r:test_empty_argpartitionPsz TestTake.test_empty_argpartitionN)__name__ __module__ __qualname__r;rPrWr]r_r9r9r9r:r s ' r) rFnumpyr numpy.testingrrrrrr9r9r9r:s