From 248113ec6004ea91ae5fcb1d868f63425c706518 Mon Sep 17 00:00:00 2001 From: mj-will Date: Mon, 11 Dec 2023 14:44:51 +0000 Subject: [PATCH] test: add integration test that includes plotting --- tests/test_sampling/test_standard_sampling.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/test_sampling/test_standard_sampling.py b/tests/test_sampling/test_standard_sampling.py index 1002659c..83474f17 100644 --- 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"""