def test_node_tostore(): gp, args = create_problem_node(nargs=4) Add.nargs = 4 n = Add(list(range(len(args))), ytr=gp._ytr, mask=gp._mask) n.position = 10 assert n.eval(args) n1 = n.tostore() assert n1.nargs == n.nargs assert n1.position == n.position assert np.all(n1.weight == n.weight) assert n1.hy is None
def test_node_pickle(): import pickle import tempfile gp, args = create_problem_node() n = Add(list(range(len(args))), ytr=gp._ytr, mask=gp._mask) n.position = 10 assert n.eval(args) with tempfile.TemporaryFile('w+b') as io: pickle.dump(n, io) io.seek(0) n1 = pickle.load(io) assert n1._mask.SSE(n._mask) == 0