reduce iterations

rs-station · Oct 28, 2024 · 4603466 · 4603466
1 parent 317eeb6
commit 4603466
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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])
-Original file line number
+Diff line change
@@ Expand Up / @@ -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)
@@ Expand Down @@