Add unit test for warn_missing_rvs=False

aesara-devs · Jun 30, 2022 · b6fed3b · b6fed3b
1 parent ee7ad4d
commit b6fed3b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_joint_logprob.py b/tests/test_joint_logprob.py
@@ -1,3 +1,5 @@
+import warnings
+
 import aesara
 import aesara.tensor as at
 import numpy as np
@@ -261,6 +263,10 @@ def test_warn_random_not_found():
     with pytest.warns(UserWarning):
         factorized_joint_logprob({y_rv: y_vv})
 
+    with warnings.catch_warnings():
+        warnings.simplefilter("error")
+        factorized_joint_logprob({y_rv: y_vv}, warn_missing_rvs=False)
+
 
 def test_multiple_rvs_to_same_value_raises():
     x_rv1 = at.random.normal(name="x1")