def test_rand_equation(optimize, n, reg, n_out, global_dim):
eq, _, size_dict = helpers.rand_equation(n, reg, n_out, d_min=2, d_max=5, seed=42, return_size_dict=True)
views = helpers.build_views(eq, size_dict)
expected = contract(eq, *views, optimize=False)
actual = contract(eq, *views, optimize=optimize)
assert np.allclose(expected, actual)
def test_rand_equation(optimize, n, reg, n_out, global_dim):
eq, _, size_dict = helpers.rand_equation(n, reg, n_out, d_min=2, d_max=5, seed=42, return_size_dict=True)
views = helpers.build_views(eq, size_dict)
expected = contract(eq, *views, optimize=False)
actual = contract(eq, *views, optimize=optimize)
assert np.allclose(expected, actual)
def test_rand_equation(optimize, n, reg, n_out):
eq, shapes = helpers.rand_equation(n,
reg,
n_out,
d_min=2,
d_max=5,
seed=42)
views = [np.random.rand(*s) for s in shapes]
expected = contract(eq, *views, optimize=False)
actual = contract(eq, *views, optimize=optimize)
assert np.allclose(expected, actual)