test: add sampling integration test for truncate_log_q

mj-will · Nov 21, 2023 · a1dcbb2 · a1dcbb2
1 parent b4bbeb0
commit a1dcbb2
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/test_sampling/test_standard_sampling.py b/tests/test_sampling/test_standard_sampling.py
@@ -459,6 +459,24 @@ def test_constant_volume_mode(integration_model, tmpdir):
     fs.run(plot=False)
 
 
+@pytest.mark.slow_integration_test
+def test_truncate_log_q(integration_model, tmpdir):
+    """Test sampling with truncate_log_q"""
+    output = str(tmpdir.mkdir("test"))
+    fs = FlowSampler(
+        integration_model,
+        output=output,
+        nlive=500,
+        plot=False,
+        proposal_plots=False,
+        constant_volume_mode=False,
+        latent_prior="flow",
+        truncate_log_q=True,
+    )
+    fs.run(plot=False)
+    assert fs.ns.finalised
+
+
 @pytest.mark.slow_integration_test
 def test_prior_sampling(integration_model, tmpdir):
     """Test prior sampling"""