def test_pickle(): import tempfile from numpy.testing import assert_equal tmpdir = tempfile.mkdtemp(prefix='pickle') a = range(10) save_pickle(a, tmpdir + '/res.pkl') b = load_pickle(tmpdir + '/res.pkl') assert_equal(a, b) #cleanup, tested on Windows try: import os os.remove(tmpdir + '/res.pkl') os.rmdir(tmpdir) except (OSError, IOError): pass assert not os.path.exists(tmpdir) #test with file handle from gwstatsmodels.compatnp.py3k import BytesIO fh = BytesIO() save_pickle(a, fh) fh.seek(0, 0) c = load_pickle(fh) fh.close() assert_equal(a, b)
def test_pickle(): import tempfile from numpy.testing import assert_equal tmpdir = tempfile.mkdtemp(prefix='pickle') a = range(10) save_pickle(a, tmpdir+'/res.pkl') b = load_pickle(tmpdir+'/res.pkl') assert_equal(a, b) #cleanup, tested on Windows try: import os os.remove(tmpdir+'/res.pkl') os.rmdir(tmpdir) except (OSError, IOError): pass assert not os.path.exists(tmpdir) #test with file handle from gwstatsmodels.compatnp.py3k import BytesIO fh = BytesIO() save_pickle(a, fh) fh.seek(0,0) c = load_pickle(fh) fh.close() assert_equal(a,b)
def save(self, fname, remove_data=False): '''save a pickle of this instance Parameters ---------- fname : string or filehandle fname can be a string to a file path or filename, or a filehandle. remove_data : bool If False (default), then the instance is pickled without changes. If True, then all arrays with length nobs are set to None before pickling. See the remove_data method. In some cases not all arrays will be set to None. ''' from gwstatsmodels.iolib.smpickle import save_pickle if remove_data: self.remove_data() save_pickle(self, fname)