Add explicit check for failing ndarray.dot(TensorVariable)

pymc-devs · Oct 3, 2023 · 4869cf1 · 4869cf1
1 parent f84ae58
commit 4869cf1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/tensor/test_variable.py b/tests/tensor/test_variable.py
@@ -83,10 +83,12 @@ def test_dot_method():
     exp_res = dot(X, y)
     assert equal_computations([res], [exp_res])
 
+    # This doesn't work. Numpy calls TensorVariable.__rmul__ at some point and everything is messed up
     X_val = np.arange(2 * 3).reshape((2, 3))
-    res = as_tensor(X_val).dot(y)
+    res = X_val.dot(y)
     exp_res = dot(X_val, y)
-    assert equal_computations([res], [exp_res])
+    with pytest.raises(AssertionError):
+        assert equal_computations([res], [exp_res])
 
 
 def test_infix_matmul_method():