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)