add test cases for indirect Raises through compute_hypothesis_test

Author: thegialeo

tests/test_diagnostics/test_diagnostics_metrics.py

@@ -247,3 +247,45 @@ def test_compute_hypothesis_test_different_distributions(summary_network, monkey
     )
 
     assert mmd_observed >= np.quantile(mmd_null, 0.68)
+
+
+@pytest.mark.parametrize("summary_network", [lambda data: np.random.rand(data.shape[0], 5), None])
+def test_compute_hypothesis_test_mismatched_last_dimension(summary_network, monkeypatch):
+    """Test that a ValueError is raised for mismatched last dimensions."""
+    observed_data = np.random.rand(10, 5)
+    reference_data = np.random.rand(20, 4)
+    num_null_samples = 10
+
+    mock_approximator = bf.approximators.ContinuousApproximator(
+        adapter=None,
+        inference_network=None,
+        summary_network=None,
+    )
+
+    monkeypatch.setattr(mock_approximator, "summary_network", summary_network)
+
+    with pytest.raises(ValueError):
+        bf.diagnostics.metrics.compute_mmd_hypothesis_test(
+            observed_data, reference_data, mock_approximator, num_null_samples
+        )
+
+
+@pytest.mark.parametrize("summary_network", [lambda data: np.random.rand(data.shape[0], 5), None])
+def test_compute_hypothesis_test_num_null_samples_exceeds_reference_samples(summary_network, monkeypatch):
+    """Test that a ValueError is raised when num_null_samples exceeds the number of reference samples."""
+    observed_data = np.random.rand(10, 5)
+    reference_data = np.random.rand(5, 5)
+    num_null_samples = 10
+
+    mock_approximator = bf.approximators.ContinuousApproximator(
+        adapter=None,
+        inference_network=None,
+        summary_network=None,
+    )
+
+    monkeypatch.setattr(mock_approximator, "summary_network", summary_network)
+
+    with pytest.raises(ValueError):
+        bf.diagnostics.metrics.compute_mmd_hypothesis_test(
+            observed_data, reference_data, mock_approximator, num_null_samples
+        )