apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-07-17 16:41:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update pickle_tests.py

Browse source
This commit is contained in:
Max Zwiessele 2015-10-10 23:28:33 +01:00
parent 0a1b3a909f
commit 2dbd799831
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
GPy/testing/pickle_tests.py
View file

@ -60,13 +60,15 @@ class Test(ListDictTestCase):
pio2 = pickle.load(f) pio2 = pickle.load(f)
self.assertListDictEquals(pio._properties, pio2._properties) self.assertListDictEquals(pio._properties, pio2._properties)
with tempfile.TemporaryFile('w+b') as f: f = tempfile.TemporaryFile('w+b')
with f:
pickle.dump(piov, f) pickle.dump(piov, f)
f.seek(0)
pio2 = pickle.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()))
self.assertListDictEquals(dict(piov.items()), dict(pio2.items())) self.assertListDictEquals(dict(piov.items()), dict(pio2.items()))
def test_param(self): def test_param(self):
param = Param('test', np.arange(4*2).reshape(4,2)) param = Param('test', np.arange(4*2).reshape(4,2))