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]])