|
| 1 | +from datetime import datetime, timedelta |
| 2 | + |
1 | 3 | import numpy as np
|
2 | 4 | import pandas as pd
|
| 5 | +from dateutil import tz |
| 6 | + |
3 | 7 | from pandas import date_range, Index, DataFrame, Series, Timestamp
|
4 | 8 | from pandas.util import testing as tm
|
5 | 9 |
|
@@ -266,3 +270,46 @@ def test_nanosecond_getitem_setitem_with_tz(self):
|
266 | 270 | result.loc[df.index[0], 'a'] = -1
|
267 | 271 | expected = DataFrame(-1, index=index, columns=['a'])
|
268 | 272 | tm.assert_frame_equal(result, expected)
|
| 273 | + |
| 274 | + def test_loc_getitem_across_dst(self): |
| 275 | + # GH 21846 |
| 276 | + idx = pd.date_range('2017-10-29 01:30:00', |
| 277 | + tz='Europe/Berlin', periods=5, freq='30 min') |
| 278 | + series2 = pd.Series([0, 1, 2, 3, 4], |
| 279 | + index=idx) |
| 280 | + |
| 281 | + t_1 = pd.Timestamp('2017-10-29 02:30:00+02:00', tz='Europe/Berlin', |
| 282 | + freq='30min') |
| 283 | + t_2 = pd.Timestamp('2017-10-29 02:00:00+01:00', tz='Europe/Berlin', |
| 284 | + freq='30min') |
| 285 | + result = series2.loc[t_1:t_2] |
| 286 | + expected = pd.Series([2, 3], index=idx[2:4]) |
| 287 | + tm.assert_series_equal(result, expected) |
| 288 | + |
| 289 | + result = series2[t_1] |
| 290 | + expected = 2 |
| 291 | + assert result == expected |
| 292 | + |
| 293 | + def test_loc_incremental_setitem_with_dst(self): |
| 294 | + # GH 20724 |
| 295 | + base = datetime(2015, 11, 1, tzinfo=tz.gettz("US/Pacific")) |
| 296 | + idxs = [base + timedelta(seconds=i * 900) for i in range(16)] |
| 297 | + result = pd.Series([0], index=[idxs[0]]) |
| 298 | + for ts in idxs: |
| 299 | + result.loc[ts] = 1 |
| 300 | + expected = pd.Series(1, index=idxs) |
| 301 | + tm.assert_series_equal(result, expected) |
| 302 | + |
| 303 | + def test_loc_setitem_with_existing_dst(self): |
| 304 | + # GH 18308 |
| 305 | + start = pd.Timestamp('2017-10-29 00:00:00+0200', tz='Europe/Madrid') |
| 306 | + end = pd.Timestamp('2017-10-29 03:00:00+0100', tz='Europe/Madrid') |
| 307 | + ts = pd.Timestamp('2016-10-10 03:00:00', tz='Europe/Madrid') |
| 308 | + idx = pd.date_range(start, end, closed='left', freq="H") |
| 309 | + result = pd.DataFrame(index=idx, columns=['value']) |
| 310 | + result.loc[ts, 'value'] = 12 |
| 311 | + expected = pd.DataFrame([np.nan] * len(idx) + [12], |
| 312 | + index=idx.append(pd.DatetimeIndex([ts])), |
| 313 | + columns=['value'], |
| 314 | + dtype=object) |
| 315 | + tm.assert_frame_equal(result, expected) |
0 commit comments