improving test coverage

Signed-off-by: Nathaniel <[email protected]>

Author: NathanielF

causalpy/tests/test_integration_pymc_examples.py

@@ -710,6 +710,10 @@ def test_iv_reg_vs_prior(mock_pymc_sample):
         result.idata, "beta_z"
     )
     assert isinstance(summary, pd.DataFrame)
+    with pytest.raises(ValueError):
+        summary = result.model.vs_prior_outcome.get_shrinkage_factors(
+            result.idata, "beta_z"
+        )
 
 
 @pytest.mark.integration
@@ -745,6 +749,10 @@ def test_iv_reg_vs_prior_hs(mock_pymc_sample):
         result.idata, "beta_z"
     )
     assert isinstance(summary, pd.DataFrame)
+    with pytest.raises(ValueError):
+        summary = result.model.vs_prior_outcome.get_inclusion_probabilities(
+            result.idata, "beta_z"
+        )
 
 
 @pytest.mark.integration

causalpy/tests/test_variable_selection_priors.py

@@ -78,7 +78,7 @@ def test_create_variable_in_model_context_horseshoe(coords):
 
 def test_create_prior_spike_and_slab(coords):
     """Test create_prior for spike-and-slab."""
-    vs_prior = VariableSelectionPrior("spike_and_slab")
+    vs_prior = VariableSelectionPrior("spike_and_slab", hyperparams={"pi_alpha": 5})
 
     with pm.Model(coords=coords) as model:
         beta = vs_prior.create_prior(name="beta", n_params=5, dims="features")