move def test_round_trip_pickle to tseries.test.test_period.py

pandas-dev · scarrucciu · Jun 25, 2015 · Jun 25, 2015 · Jun 25, 2015 · Jun 25, 2015
commit 22d9ad5b141fcc3663050a05d54634a0a092272b
diff --git a/pandas/tseries/tests/test_period.py b/pandas/tseries/tests/test_period.py
@@ -6,6 +6,9 @@
 
 """
 
+import pickle
+import os
+
 from datetime import datetime, date, timedelta
 
 from numpy.ma.testutils import assert_equal
@@ -2537,6 +2540,16 @@ def test_searchsorted(self):
             ValueError, 'Different period frequency: H',
             lambda: pidx.searchsorted(pd.Period('2014-01-01', freq='H')))
 
+    def test_round_trip_pickle(self):
+
+        p = Period('2000Q1')
+        pickle_path = os.path.join(tm.get_data_path(),
+                                   'period.pickle')
+
+        with open(pickle_path, 'wb') as f: pickle.dump(p, f)
+
+        self.assertEqual(p, pd.read_pickle(pickle_path))
+
 def _permute(obj):
     return obj.take(np.random.permutation(len(obj)))