Skip to content

Commit b8acbde

Browse files
committed
move tests of df.__getitem__ and s.__getitem__ to tests/indexing/test_scalar.py
1 parent 721684f commit b8acbde

File tree

3 files changed

+13
-21
lines changed

3 files changed

+13
-21
lines changed

pandas/tests/frame/test_indexing.py

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -3682,14 +3682,3 @@ def test_functions_no_warnings(self):
36823682
with tm.assert_produces_warning(False):
36833683
df['group'] = pd.cut(df.value, range(0, 105, 10), right=False,
36843684
labels=labels)
3685-
3686-
def test_getitem_zerodim_np_array(self):
3687-
# GH24924
3688-
df = DataFrame([[1, 2], [3, 4]])
3689-
3690-
# should not raise an error
3691-
result = df[np.array(0)]
3692-
3693-
# expected series
3694-
sr = pd.Series([1, 3], name=0)
3695-
tm.assert_series_equal(result, sr)

pandas/tests/indexing/test_scalar.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -205,3 +205,16 @@ def test_iat_setter_incompatible_assignment(self):
205205
result.iat[0, 0] = None
206206
expected = DataFrame({"a": [None, 1], "b": [4, 5]})
207207
tm.assert_frame_equal(result, expected)
208+
209+
def test_getitem_zerodim_np_array(self):
210+
# GH24924
211+
# dataframe __getitem__
212+
df = DataFrame([[1, 2], [3, 4]])
213+
result = df[np.array(0)]
214+
expected = Series([1, 3], name=0)
215+
tm.assert_series_equal(result, expected)
216+
217+
# series __getitem__
218+
s = Series([1, 2])
219+
result = s[np.array(0)]
220+
assert result == 1

pandas/tests/series/indexing/test_indexing.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -838,13 +838,3 @@ def test_head_tail(test_data):
838838
assert_series_equal(test_data.series.head(0), test_data.series[0:0])
839839
assert_series_equal(test_data.series.tail(), test_data.series[-5:])
840840
assert_series_equal(test_data.series.tail(0), test_data.series[0:0])
841-
842-
843-
def test_getitem_with_zerodim_np_array():
844-
# GH24924
845-
sr = Series([1, 2])
846-
847-
# should not raise an error
848-
result = sr[np.array(0)]
849-
850-
assert result == 1

0 commit comments

Comments
 (0)