Skip to main content

Browse the archive

Revision 5b50dfa3e7e114b8ebd87cea0f9eb2170c52e201 authored by Mark van der Wilk on 15 June 2016, 15:28:31 UTC, committed by Mark van der Wilk on 15 June 2016, 15:28:31 UTC 
Separated dict test, added prediction test after unpickling to test AutoFlow.
1 parent 82411ab
Raw File
test_todict.py 
import unittest
import GPflow
import numpy as np
import pickle


class TestPickleAndDict(unittest.TestCase):
    def setUp(self):
        rng = np.random.RandomState(0)
        X = rng.randn(10, 1)
        Y = rng.randn(10, 1)
        self.m = GPflow.gpr.GPR(X, Y, kern=GPflow.kernels.RBF(1))

    def test(self):
        # pickle and reload the model
        s1 = pickle.dumps(self.m)
        m1 = pickle.loads(s1)

        d1 = self.m.to_dict()
        d2 = m1.to_dict()
        for key, val in d1.items():
            assert np.all(val == d2[key])


if __name__ == "__main__":
    unittest.main()
back to top