Skip to content
Snippets Groups Projects
Commit 876a9efe authored by Levecque Etienne's avatar Levecque Etienne
Browse files

test: fix issue related to avoid_uniform being removed

parent e6512f85
No related branches found
No related tags found
No related merge requests found
......@@ -55,7 +55,7 @@ class TestFilterJPEGColor(unittest.TestCase):
self.filename = "jpeg_color.jpg"
self.img = Image(self.filename, self.filename)
self.img.set_filter_parameter(True, True)
self.img.set_filter_parameter(True)
self.simple_pipeline = create_pipeline('islow', 60, False, True)
self.double_pipeline = create_pipeline(['islow', 'islow'], [75, 60], False, True)
self.triple_pipeline = create_pipeline(['islow', 'islow', 'islow'], [75, 75, 60], False, True)
......@@ -94,7 +94,7 @@ class TestFilterJPEGGrayscale(unittest.TestCase):
self.filename = "jpeg_grayscale.jpg"
self.img = Image(self.filename, self.filename)
self.img.set_filter_parameter(True, True)
self.img.set_filter_parameter(True)
self.simple_pipeline = create_pipeline('islow', 60, True, True)
self.double_pipeline = create_pipeline(['islow', 'islow'], [75, 60], True, True)
self.triple_pipeline = create_pipeline(['islow', 'islow', 'islow'], [75, 75, 60], True, True)
......@@ -133,7 +133,7 @@ class TestFilterSpatialColor(unittest.TestCase):
self.filename = "spatial_color.tif"
self.img = Image(self.filename, self.filename)
self.img.set_filter_parameter(True, True)
self.img.set_filter_parameter(True)
self.simple_pipeline = create_pipeline('islow', 60, False, False)
self.double_pipeline = create_pipeline(['islow', 'islow'], [75, 60], False, False)
self.triple_pipeline = create_pipeline(['islow', 'islow', 'islow'], [75, 75, 60], False, False)
......@@ -155,8 +155,12 @@ class TestFilterSpatialColor(unittest.TestCase):
true_mask = true_mask | (np.any(backward_blocks <= 0, axis=(1, 2, 3)) |
np.any(backward_blocks >= 255, axis=(1, 2, 3)))
true_mask = true_mask.reshape(self.img.block_view.shape[:2])
print(true_mask.shape)
print(true_mask)
self.img.set_pipeline(self.simple_pipeline)
mask = self.img.filter_blocks(purge=True)
print(mask.shape)
print(mask)
self.assertTrue(np.allclose(mask, true_mask))
def test_triple_pipeline(self):
......@@ -171,7 +175,7 @@ class TestFilterSpatialGrayscale(unittest.TestCase):
self.filename = "spatial_grayscale.tif"
self.img = Image(self.filename, self.filename)
self.img.set_filter_parameter(True, True)
self.img.set_filter_parameter(True)
self.simple_pipeline = create_pipeline('islow', 60, True, False)
self.double_pipeline = create_pipeline(['islow', 'islow'], [75, 60], True, False)
self.triple_pipeline = create_pipeline(['islow', 'islow', 'islow'], [75, 75, 60], True, False)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment