Add test

pymc-devs · Jan 15, 2025 · b065709 · b065709
1 parent 3faae09
commit b065709
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/model/test_model_api.py b/tests/model/test_model_api.py
@@ -30,7 +30,9 @@ def model_wrapped3(mu):
         pm.Normal("x", mu, 1.0, dims="obs")
 
     mw3 = model_wrapped3(0.0, coords=coords)
+    mw4 = model_wrapped(np.array([np.nan]), coords=coords)
 
     np.testing.assert_equal(model.point_logps(), mw.point_logps())
     np.testing.assert_equal(mw.point_logps(), mw2.point_logps())
     assert mw3["mu"] in mw3.data_vars
+    assert "mu" not in mw4