o g@s<ddlZddlZddlmZddlmZddZddZdS)N)SeriescCstgd}|gd}tgdgdd}t|||gd}tgdgdd}t||dd }tjt|d d |d d gWdn1sSwYtjt|d d |dd gWddS1stwYdS)N))r)rrrindex)rr r)rrr)rr rcSs d|dS)Nzindex z+ is out of bounds for( axis 0 with)? size 5)xr r d/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pandas/tests/series/indexing/test_take.pys ztest_take.. )matchrrr)rtaketmassert_series_equalpytestraises IndexError)seractualexpectedmsgr r r test_takes   "rcCsRttgd}|gd}ttjgdgddgdd}t||dS)N)abc)r r)rrr) categories)rrrr )rpd Categoricalrrr)rresultrr r rtest_take_categoricals r%) rpandasr"rpandas._testing_testingrrr%r r r rs