Update pickle_tests.py

This commit is contained in:
Max Zwiessele 2015-10-10 23:21:41 +01:00
parent 55fb393751
commit 2a01e9008a

View file

@ -22,6 +22,7 @@ from GPy.models.gp_regression import GPRegression
from functools import reduce from functools import reduce
from GPy.util.caching import Cacher from GPy.util.caching import Cacher
from pickle import PicklingError from pickle import PicklingError
import GPy
def toy_model(): def toy_model():
X = np.linspace(0,1,50)[:, None] X = np.linspace(0,1,50)[:, None]
@ -64,8 +65,7 @@ class Test(ListDictTestCase):
with f: with f:
pickle.dump(piov, f) pickle.dump(piov, f)
f.seek(0)
pio2 = GPy.load(f) pio2 = GPy.load(f)
#py3 fix #py3 fix
#self.assertListDictEquals(dict(piov.items()), dict(pio2.iteritems())) #self.assertListDictEquals(dict(piov.items()), dict(pio2.iteritems()))