Adds test just in case

tingiskhan · Nov 2, 2024 · 55c8a8d · 55c8a8d
1 parent cc5f3de
commit 55c8a8d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_union.py b/tests/test_union.py
@@ -15,5 +15,9 @@ def test_union():
     key = jrnd.PRNGKey(123)
     samples = combined.sample(key)
 
+    log_prob = combined.log_prob(samples)
+
     assert samples.shape == combined.event_shape
     assert combined.event_shape[-1] == 7
+
+