Skip to content

Commit 5861b4e

Browse files
committed
Updated tests
- Moved from series/indexing/test_indexing to indexing/test_scalar - Added unit tests for loc get/set - Added GH issue number as comment in tests
1 parent 1321905 commit 5861b4e

File tree

2 files changed

+27
-11
lines changed

2 files changed

+27
-11
lines changed

pandas/tests/indexing/test_scalar.py

+27
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import numpy as np
55
import pytest
66

7+
import pandas as pd
78
from pandas import DataFrame, Series, Timedelta, Timestamp, date_range
89
import pandas._testing as tm
910
from pandas.tests.indexing.common import Base
@@ -302,3 +303,29 @@ def test_iat_dont_wrap_object_datetimelike():
302303
assert result is ser2[1]
303304
assert isinstance(result, timedelta)
304305
assert not isinstance(result, Timedelta)
306+
307+
308+
def test_multiindex_series_loc_get():
309+
series = pd.Series([1, 2], index=[[1, 2], [3, 4]])
310+
assert series.loc[1, 3] == 1
311+
312+
313+
def test_multiindex_series_loc_set():
314+
series = pd.Series([1, 2], index=[[1, 2], [3, 4]])
315+
series.loc[1, 3] = 3
316+
assert series.loc[1, 3] == 3
317+
318+
319+
def test_multiindex_series_at_get():
320+
# GH 26989
321+
# Series with multiindex at getter raises TypeError
322+
series = pd.Series([1, 2], index=[[1, 2], [3, 4]])
323+
assert series.at[1, 3] == 1
324+
325+
326+
def test_multiindex_series_at_set():
327+
# GH 26989
328+
# Series with multiindex at setter raises ValueError
329+
series = pd.Series([1, 2], index=[[1, 2], [3, 4]])
330+
series.at[1, 3] = 3
331+
assert series.at[1, 3] == 3

pandas/tests/series/indexing/test_indexing.py

-11
Original file line numberDiff line numberDiff line change
@@ -923,14 +923,3 @@ def test_getitem_2d_no_warning():
923923
series = pd.Series([1, 2, 3], index=[1, 2, 3])
924924
with tm.assert_produces_warning(None):
925925
series[:, None]
926-
927-
928-
def test_multiindex_at_get():
929-
series = pd.Series([1, 2], index=[[1, 2], [3, 4]])
930-
assert series.at[1, 3] == 1
931-
932-
933-
def test_multiindex_at_set():
934-
series = pd.Series([1, 2], index=[[1, 2], [3, 4]])
935-
series.at[1, 3] = 3
936-
assert series.at[1, 3] == 3

0 commit comments

Comments
 (0)