|
4 | 4 | import numpy as np
|
5 | 5 | import pytest
|
6 | 6 |
|
| 7 | +import pandas as pd |
7 | 8 | from pandas import DataFrame, Series, Timedelta, Timestamp, date_range
|
8 | 9 | import pandas._testing as tm
|
9 | 10 | from pandas.tests.indexing.common import Base
|
@@ -302,3 +303,29 @@ def test_iat_dont_wrap_object_datetimelike():
|
302 | 303 | assert result is ser2[1]
|
303 | 304 | assert isinstance(result, timedelta)
|
304 | 305 | 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 |
0 commit comments