cSAXS/cdtools
0
0
Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-10-16 21:05:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

get tests working

Browse Source
This commit is contained in:
Abe Levitan
2021-06-21 16:55:34 -04:00
parent 8f58242efe
commit 76e6d5efae
18 changed files with 291 additions and 1072 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/tools/test_image_processing.py
+3 -3
View File
@@ -88,8 +88,7 @@ def test_find_subpixel_shift():
shift = t.tensor((0.8,0.75))
im = interactions.ptycho_2D_sinc(test_probe, test_obj, shift)
print(im.shape)
im = interactions.ptycho_2D_sinc(test_probe, test_obj, shift, multiple_modes=False)
retrieved_shift = image_processing.find_subpixel_shift(im, test_probe, search_around=(0,0), resolution=50)
# tolerance of 0.03 on this measurement
@@ -104,7 +103,8 @@ def test_find_shift():
shift = t.tensor((0.8,0.75))
im = interactions.ptycho_2D_sinc(test_probe, test_obj, shift)[:-40,:-6]
im = interactions.ptycho_2D_sinc(test_probe, test_obj, shift,
multiple_modes=False)[:-40,:-6]
retrieved_shift = image_processing.find_shift(im, test_probe[40:,6:], resolution=50)
# tolerance of 0.03 on this measurement