def test_dump_load_mrg(self): rng = MRG_RandomStream() with open("test", "wb") as f: dump(rng, f) with open("test", "rb") as f: rng = load(f) assert type(rng) == MRG_RandomStream
def test_dump_zip_names(self): foo_1 = aesara.shared(0, name="foo") foo_2 = aesara.shared(1, name="foo") foo_3 = aesara.shared(2, name="foo") with open("model.zip", "wb") as f: dump((foo_1, foo_2, foo_3, np.array(3)), f) keys = list(np.load("model.zip").keys()) assert keys == ["foo", "foo_2", "foo_3", "array_0", "pkl"] foo_3 = np.load("model.zip")["foo_3"] assert foo_3 == np.array(2) with open("model.zip", "rb") as f: foo_1, foo_2, foo_3, array = load(f) assert array == np.array(3)
def test_dump_load(): x = GpuArraySharedVariable( "x", GpuArrayType("float32", (1, 1), name="x", context_name=test_ctx_name), [[1]], False, ) with open("test", "wb") as f: dump(x, f) with open("test", "rb") as f: x = load(f) assert x.name == "x" np.testing.assert_allclose(x.get_value(), [[1]])