TST: Enable probe support in tests

AdvancedPhotonSource · Jun 17, 2024 · 6be8995 · 6be8995
1 parent 9b8e20c
commit 6be8995
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/tests/ptycho/test_ptycho.py b/tests/ptycho/test_ptycho.py
@@ -374,8 +374,11 @@ def test_consistent_lstsq_grad(self):
             probe_options=ProbeOptions(
                 force_orthogonality=True,
                 use_adaptive_moment=True,
+                probe_support=0.1,
+            ),
+            object_options=ObjectOptions(
+                use_adaptive_moment=True,
             ),
-            object_options=ObjectOptions(use_adaptive_moment=True,),
         )
 
         _save_ptycho_result(
@@ -583,8 +586,13 @@ def test_consistent_rpie(self):
                 num_batch=5,
                 num_iter=16,
             ),
-            probe_options=ProbeOptions(force_orthogonality=True,),
-            object_options=ObjectOptions(smoothness_constraint=0.01,),
+            probe_options=ProbeOptions(
+                force_orthogonality=True,
+                probe_support=0.1,
+            ),
+            object_options=ObjectOptions(
+                smoothness_constraint=0.01,
+            ),
         )
 
         _save_ptycho_result(