def fn():
rstate = random_initialize(12345678)
r0, v0 = random_uint32(rstate, (2, 2))
r1, v1 = random_uint32(r0, (2, 2))
r2, v2 = random_uint32(r1, (1,))
r3, v3 = random_uint32(r2, ())
return r3, v0, v1, v2, v3
def test_with_random(a, b):
rstate = random_initialize(123)
rstate, v0 = random_uint32(rstate, (2, 3))
_, v1 = random_uint32(rstate, (2, 3))
v0 = array_cast(v0, xtype.f64)
v1 = array_cast(v1, xtype.f64)
return a * np.sum(v0).item() + b * np.sum(v1).item()
def test_random():
rstate = random_initialize(1234)
r0, v0 = random_uint32(rstate, (2, 3))
r1, v1 = random_uint32(rstate, (2, 3))
assert isinstance(rstate, np.random.RandomState)
assert isinstance(r0, np.random.RandomState)
assert isinstance(r1, np.random.RandomState)
assert isinstance(v0, np.ndarray)
assert isinstance(v1, np.ndarray)
assert v0.dtype == "uint32"
assert v1.dtype == "uint32"
assert v0.shape == (2, 3)
assert v1.shape == (2, 3)
assert not np.allclose(v0, v1)
def step_init():
"""Myia function to initialize rstate."""
return random_initialize(1234)
def init():
return random_initialize(12345678)
def f() -> RandomStateWrapper:
rstate: RandomStateWrapper = random_initialize(10)
r0, _ = random_uint32(rstate, ())
return r0