test: add integration test that includes plotting

mj-will · Dec 13, 2023 · 248113e · 248113e
1 parent 3dbb1a2
commit 248113e
Showing 1 changed file with 15 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,21 @@ def test_constant_volume_mode(integration_model, tmpdir):
     fs.run(plot=False)
 
 
+@pytest.mark.slow_integration_test
+def test_sampling_with_plotting(integration_model, tmpdir):
+    """Test sampling with plots enabled"""
+    output = str(tmpdir.mkdir("test"))
+    fs = FlowSampler(
+        integration_model,
+        output=output,
+        nlive=100,
+        plot=True,
+        proposal_plots=True,
+        stopping=1.0,
+    )
+    fs.run(plot=True)
+
+
 @pytest.mark.slow_integration_test
 def test_truncate_log_q(integration_model, tmpdir):
     """Test sampling with truncate_log_q"""