From 55fb393751321dd029aaa5c3b52d1911d852f40f Mon Sep 17 00:00:00 2001 From: Max Zwiessele Date: Sat, 10 Oct 2015 23:13:08 +0100 Subject: [PATCH] Update pickle_tests.py --- GPy/testing/pickle_tests.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/GPy/testing/pickle_tests.py b/GPy/testing/pickle_tests.py index 64357b39..d4d96a24 100644 --- a/GPy/testing/pickle_tests.py +++ b/GPy/testing/pickle_tests.py @@ -59,14 +59,17 @@ class Test(ListDictTestCase): f.seek(0) pio2 = pickle.load(f) self.assertListDictEquals(pio._properties, pio2._properties) - - with tempfile.TemporaryFile('w+b') as f: + + f = tempfile.TemporaryFile('w+b') + + with f: pickle.dump(piov, f) f.seek(0) - pio2 = pickle.load(f) - #py3 fix - #self.assertListDictEquals(dict(piov.items()), dict(pio2.iteritems())) - self.assertListDictEquals(dict(piov.items()), dict(pio2.items())) + + pio2 = GPy.load(f) + #py3 fix + #self.assertListDictEquals(dict(piov.items()), dict(pio2.iteritems())) + self.assertListDictEquals(dict(piov.items()), dict(pio2.items())) def test_param(self): param = Param('test', np.arange(4*2).reshape(4,2))