From cfd4512a1a357ac92f611ba46c099f86d814f7a0 Mon Sep 17 00:00:00 2001
From: Tomas Stolker <tomas.stolker@phys.ethz.ch>
Date: Mon, 21 Dec 2020 11:58:28 +0100
Subject: [PATCH] Updated unit tests for BadPixelSigmaFilterModule

---
 tests/test_processing/test_badpixel.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/tests/test_processing/test_badpixel.py b/tests/test_processing/test_badpixel.py
index 4ce5aaefe..c381e1e7a 100644
--- a/tests/test_processing/test_badpixel.py
+++ b/tests/test_processing/test_badpixel.py
@@ -51,14 +51,14 @@ def test_bad_pixel_sigma_filter(self) -> None:
                                            image_out_tag='sigma1',
                                            map_out_tag='None',
                                            box=9,
-                                           sigma=5.,
-                                           iterate=1)
+                                           sigma=3.,
+                                           iterate=5)
 
         self.pipeline.add_module(module)
         self.pipeline.run_module('sigma1')
 
         data = self.pipeline.get_data('sigma1')
-        assert np.sum(data) == pytest.approx(0.007314386854009355, rel=self.limit, abs=0.)
+        assert np.sum(data) == pytest.approx(0.006513475520308432, rel=self.limit, abs=0.)
         assert data.shape == (5, 11, 11)
 
     def test_bad_pixel_map_out(self) -> None:
@@ -68,22 +68,22 @@ def test_bad_pixel_map_out(self) -> None:
                                            image_out_tag='sigma2',
                                            map_out_tag='bpmap',
                                            box=9,
-                                           sigma=5.,
-                                           iterate=1)
+                                           sigma=2.,
+                                           iterate=3)
 
         self.pipeline.add_module(module)
         self.pipeline.run_module('sigma2')
 
         data = self.pipeline.get_data('sigma2')
-        assert data[0, 0, 0] == pytest.approx(0.00032486907273264834, rel=self.limit, abs=0.)
+        assert data[0, 0, 0] == pytest.approx(-2.4570591355257687e-05, rel=self.limit, abs=0.)
         assert data[0, 5, 5] == pytest.approx(9.903775276151606e-06, rel=self.limit, abs=0.)
-        assert np.sum(data) == pytest.approx(0.007314386854009355, rel=self.limit, abs=0.)
+        assert np.sum(data) == pytest.approx(0.011777887008566097, rel=self.limit, abs=0.)
         assert data.shape == (5, 11, 11)
 
         data = self.pipeline.get_data('bpmap')
-        assert data[0, 0, 0] == pytest.approx(1., rel=self.limit, abs=0.)
+        assert data[0, 1, 1] == pytest.approx(1., rel=self.limit, abs=0.)
         assert data[0, 5, 5] == pytest.approx(0., rel=self.limit, abs=0.)
-        assert np.sum(data) == pytest.approx(604., rel=self.limit, abs=0.)
+        assert np.sum(data) == pytest.approx(519.0, rel=self.limit, abs=0.)
         assert data.shape == (5, 11, 11)
 
     def test_bad_pixel_map(self) -> None: