Skip to content

Commit 793a5c9

Browse files
author
y-p
committed
TST: PeriodIndex pickle roundtrip does not recreate freq GH2891
1 parent 900a552 commit 793a5c9

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

pandas/tests/test_series.py

+10
Original file line numberDiff line numberDiff line change
@@ -1324,6 +1324,16 @@ def test_timeseries_repr_object_dtype(self):
13241324
ts2 = ts.ix[np.random.randint(0, len(ts) - 1, 400)]
13251325
repr(ts).splitlines()[-1]
13261326

1327+
def test_timeseries_periodindex(self):
1328+
# GH2891
1329+
import pickle
1330+
from pandas import period_range
1331+
prng = period_range('1/1/2011', '1/1/2012', freq='M')
1332+
ts = Series(np.random.randn(len(prng)), prng)
1333+
new_ts = pickle.loads(pickle.dumps(ts))
1334+
self.assertEqual(new_ts.index.freq,'M')
1335+
1336+
13271337
def test_iter(self):
13281338
for i, val in enumerate(self.series):
13291339
self.assertEqual(val, self.series[i])

pandas/tseries/tests/test_period.py

+6
Original file line numberDiff line numberDiff line change
@@ -1946,6 +1946,12 @@ def test_append_concat(self):
19461946
self.assert_(isinstance(result.index, PeriodIndex))
19471947
self.assertEquals(result.index[0], s1.index[0])
19481948

1949+
def test_pickle_freq(self):
1950+
# GH2891
1951+
import pickle
1952+
prng = period_range('1/1/2011', '1/1/2012', freq='M')
1953+
new_prng = pickle.loads(pickle.dumps(prng))
1954+
self.assertEqual(new_prng.freq,'M')
19491955

19501956
def _permute(obj):
19511957
return obj.take(np.random.permutation(len(obj)))

0 commit comments

Comments
 (0)