10441 Use self.round_trip_pickle to test in test_period

pandas-dev · scarrucciu · Jun 25, 2015 · Jun 25, 2015 · Jun 25, 2015 · Jun 25, 2015
commit f64f7467cab6464a9ac30581ecb7ef0571602e52
diff --git a/pandas/io/tests/generate_legacy_storage_files.py b/pandas/io/tests/generate_legacy_storage_files.py
@@ -53,11 +53,6 @@ def _create_sp_frame():
     return SparseDataFrame(data, index=dates)
 
 
-def _create_period():
-
-    return Period('2000Q1')
-
-
 def create_data():
     """ create the pickle/msgpack data """
 

diff --git a/pandas/tseries/tests/test_period.py b/pandas/tseries/tests/test_period.py
@@ -2541,13 +2541,11 @@ def test_searchsorted(self):
 
     def test_round_trip(self):
 
+        import pickle
         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))
+        new_p = self.round_trip_pickle(p)
+        self.assertEqual(new_p, p)
 
 def _permute(obj):
     return obj.take(np.random.permutation(len(obj)))