fixed multislice unit test where we need extra parameters (probe FOV,…

… multislice prop distance, wavelength)
AdvancedPhotonSource · Jul 31, 2024 · 5c403b3 · 5c403b3
1 parent a736262
commit 5c403b3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/operators/test_multislice.py b/tests/operators/test_multislice.py
@@ -39,9 +39,12 @@ def setUp(self, depth=7, pw=15, nscan=27):
         self.operator = Multislice(
             nscan=self.scan_shape[-2],
             probe_shape=self.probe_shape[-1],
+            probe_wavelength = 1e-10,
+            probe_FOV_lengths = (1e-5, 1e-5),
             detector_shape=self.detector_shape[-1],
             nz=self.original_shape[-2],
             n=self.original_shape[-1],
+            multislice_propagation_distance = 1e-8,
         )
         self.operator.__enter__()
         self.xp = self.operator.xp