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 pandas as pd import pandas._testing as tm def test_getitem_callable(): # GH 12533 s = pd.Series(4, index=list("ABCD")) result = s[lambda x: "A"] assert result == s.loc["A"] result = s[lambda x: ["A", "B"]] tm.assert_series_equal(result, s.loc[["A", "B"]]) result = s[lambda x: [True, False, True, True]] tm.assert_series_equal(result, s.iloc[[0, 2, 3]]) def test_setitem_callable(): # GH 12533 s = pd.Series([1, 2, 3, 4], index=list("ABCD")) s[lambda x: "A"] = -1 tm.assert_series_equal(s, pd.Series([-1, 2, 3, 4], index=list("ABCD"))) def test_setitem_other_callable(): # GH 13299 inc = lambda x: x + 1 s = pd.Series([1, 2, -1, 4]) s[s < 0] = inc expected = pd.Series([1, 2, inc, 4]) tm.assert_series_equal(s, expected)