Merge pull request #43 from yoshikisd/bugfix/pytest_reconstruct

Set fixed RNG seed for reconstruction pytests

tests/models/test_fancy_ptycho.py

@@ -5,6 +5,8 @@ import torch as t
 import cdtools
 from matplotlib import pyplot as plt
 
+# Force all reconstructions to use the same RNG seed
+t.manual_seed(0)
 
 def test_center_probe(lab_ptycho_cxi):
     dataset = cdtools.datasets.Ptycho2DDataset.from_cxi(lab_ptycho_cxi)

tests/models/test_simple_ptycho.py

@@ -1,6 +1,10 @@
 import pytest
 import cdtools
 from matplotlib import pyplot as plt
+import torch as t
+
+# Force all reconstructions to use the same RNG seed
+t.manual_seed(0)
 
 @pytest.mark.slow
 def test_simple_ptycho(lab_ptycho_cxi, reconstruction_device, show_plot):