Skip to content

Commit 5f5acc3

Browse files
TST: confirming tests for some fixed issues (#14117)
* TST: assert indexing with list of Periods works (GH7710) * TST: assert median for timedelta with NaT works (GH8617) * TST: assert no conversion to float when resampling datetime64 values (13119)
1 parent 70bb179 commit 5f5acc3

File tree

3 files changed

+32
-0
lines changed

3 files changed

+32
-0
lines changed

pandas/tseries/tests/test_period.py

+7
Original file line numberDiff line numberDiff line change
@@ -2180,6 +2180,13 @@ def test_getitem_nat(self):
21802180
pd.Period('2011-01', freq='M'))
21812181
self.assertIs(s[pd.NaT], tslib.NaT)
21822182

2183+
def test_getitem_list_periods(self):
2184+
# GH 7710
2185+
rng = period_range(start='2012-01-01', periods=10, freq='D')
2186+
ts = Series(lrange(len(rng)), index=rng)
2187+
exp = ts.iloc[[1]]
2188+
tm.assert_series_equal(ts[[Period('2012-01-02', freq='D')]], exp)
2189+
21832190
def test_slice_with_negative_step(self):
21842191
ts = Series(np.arange(20),
21852192
period_range('2014-01', periods=20, freq='M'))

pandas/tseries/tests/test_resample.py

+17
Original file line numberDiff line numberDiff line change
@@ -2022,6 +2022,23 @@ def test_resample_timedelta_values(self):
20222022
res = df['time'].resample('2D').first()
20232023
tm.assert_series_equal(res, exp)
20242024

2025+
def test_resample_datetime_values(self):
2026+
# GH 13119
2027+
# check that datetime dtype is preserved when NaT values are
2028+
# introduced by the resampling
2029+
2030+
dates = [datetime(2016, 1, 15), datetime(2016, 1, 19)]
2031+
df = DataFrame({'timestamp': dates}, index=dates)
2032+
2033+
exp = Series([datetime(2016, 1, 15), pd.NaT, datetime(2016, 1, 19)],
2034+
index=date_range('2016-01-15', periods=3, freq='2D'),
2035+
name='timestamp')
2036+
2037+
res = df.resample('2D').first()['timestamp']
2038+
tm.assert_series_equal(res, exp)
2039+
res = df['timestamp'].resample('2D').first()
2040+
tm.assert_series_equal(res, exp)
2041+
20252042

20262043
class TestPeriodIndex(Base, tm.TestCase):
20272044
_multiprocess_can_split_ = True

pandas/tseries/tests/test_timeseries.py

+8
Original file line numberDiff line numberDiff line change
@@ -4400,6 +4400,14 @@ def test_intercept_astype_object(self):
44004400
result = df.values.squeeze()
44014401
self.assertTrue((result[:, 0] == expected.values).all())
44024402

4403+
def test_nat_operations(self):
4404+
# GH 8617
4405+
s = Series([0, pd.NaT], dtype='m8[ns]')
4406+
exp = s[0]
4407+
self.assertEqual(s.median(), exp)
4408+
self.assertEqual(s.min(), exp)
4409+
self.assertEqual(s.max(), exp)
4410+
44034411

44044412
class TestTimestamp(tm.TestCase):
44054413
def test_class_ops_pytz(self):

0 commit comments

Comments
 (0)