From 46034667db389ba20eb4f56c187b007356b4e4c6 Mon Sep 17 00:00:00 2001 From: tjlane Date: Mon, 28 Oct 2024 16:58:26 +0100 Subject: [PATCH] reduce iterations --- test/functional/test_compute_iterative_tv_map.py | 2 ++ test/unit/test_tv.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/test/functional/test_compute_iterative_tv_map.py b/test/functional/test_compute_iterative_tv_map.py index 00eef52..65fbdda 100644 --- a/test/functional/test_compute_iterative_tv_map.py +++ b/test/functional/test_compute_iterative_tv_map.py @@ -48,6 +48,8 @@ def test_script_produces_consistent_results( str(kweight_parameter), "-x", "0.01", + "--max-iterations", + "5" ] compute_iterative_tv_map.main(cli_args) diff --git a/test/unit/test_tv.py b/test/unit/test_tv.py index 35cad34..183ee25 100644 --- a/test/unit/test_tv.py +++ b/test/unit/test_tv.py @@ -78,7 +78,7 @@ def test_tv_denoise_zero_weight(random_difference_map: Map) -> None: ) random_difference_map.canonicalize_amplitudes() output.canonicalize_amplitudes() - pd.testing.assert_frame_equal(random_difference_map, output, atol=1e-3, rtol=1e-2) + pd.testing.assert_frame_equal(random_difference_map, output, atol=1e-2, rtol=1e-2) @pytest.mark.parametrize("weights_to_scan", [None, DEFAULT_WEIGHTS_TO_SCAN])