Name	Size	Mode	Actions
__pycache__/	-	0755	rm
test_alter_index.py	9723	0644	editdlrm
test_boolean.py	3556	0644	editdlrm
test_callable.py	810	0644	editdlrm
test_datetime.py	18978	0644	editdlrm
test_delitem.py	1404	0644	editdlrm
test_get.py	4324	0644	editdlrm
test_getitem.py	5647	0644	editdlrm
test_iloc.py	743	0644	editdlrm
test_indexing.py	27503	0644	editdlrm
test_loc.py	4405	0644	editdlrm
test_mask.py	1602	0644	editdlrm
test_multiindex.py	694	0644	editdlrm
test_numeric.py	3367	0644	editdlrm
test_setitem.py	793	0644	editdlrm
test_take.py	963	0644	editdlrm
test_where.py	12318	0644	editdlrm
test_xs.py	454	0644	editdlrm
__init__.py	0	0644	editdlrm

import pytest import pandas as pd from pandas import Series import pandas._testing as tm def test_take(): ser = Series([-1, 5, 6, 2, 4]) actual = ser.take([1, 3, 4]) expected = Series([5, 2, 4], index=[1, 3, 4]) tm.assert_series_equal(actual, expected) actual = ser.take([-1, 3, 4]) expected = Series([4, 2, 4], index=[4, 3, 4]) tm.assert_series_equal(actual, expected) msg = lambda x: f"index {x} is out of bounds for( axis 0 with)? size 5" with pytest.raises(IndexError, match=msg(10)): ser.take([1, 10]) with pytest.raises(IndexError, match=msg(5)): ser.take([2, 5]) def test_take_categorical(): # https://github.com/pandas-dev/pandas/issues/20664 ser = Series(pd.Categorical(["a", "b", "c"])) result = ser.take([-2, -2, 0]) expected = Series( pd.Categorical(["b", "b", "a"], categories=["a", "b", "c"]), index=[1, 1, 0] ) tm.assert_series_equal(result, expected)