diff --git a/test/test_symbolic_jit.py b/test/test_symbolic_jit.py index 94608db0..c3b78550 100644 --- a/test/test_symbolic_jit.py +++ b/test/test_symbolic_jit.py @@ -124,7 +124,6 @@ class TestSymbolicJit(unittest.TestCase): np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6) assert_jit_cache_len(jf, 1) - @unittest.skip("two vars not supported") def test_two_vars_plus1_ij(self): def f(a, b): return (a@b+1).realize() jf = TinyJit(f) @@ -139,7 +138,6 @@ class TestSymbolicJit(unittest.TestCase): np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6) assert_jit_cache_len(jf, 1) - @unittest.skip("two vars not supported") def test_two_vars_plus1_ji(self): def f(a, b): return (a@b+1).realize() jf = TinyJit(f) diff --git a/test/test_symbolic_ops.py b/test/test_symbolic_ops.py index 29f914c3..9338ea0d 100644 --- a/test/test_symbolic_ops.py +++ b/test/test_symbolic_ops.py @@ -108,7 +108,6 @@ class TestSymbolicOps(unittest.TestCase): expected = f(a, b).numpy() np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6) - @unittest.skip("two vars not supported") def test_two_vars_plus1_ij(self): def f(a, b): return (a@b+1).realize() for i in range(1, 5): @@ -121,7 +120,6 @@ class TestSymbolicOps(unittest.TestCase): expected = f(a, b).numpy() np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6) - @unittest.skip("two vars not supported") def test_two_vars_plus1_ji(self): # reverse the order of variables def f(a, b): return (a@b+1).realize()