def test_if_returns_correct_values(self, template, image, result): transformation = trfs.HistogramMatchingTransformation(template) assert np.all(transformation(image) == result)
def test_if_template_and_input_have_to_have_same_no_of_channels(self): image = np.random.rand(*INPUT_VOLUME_SHAPE_4D) template = np.random.rand(*TEMPLATE_VOLUME_SHAPE_4D) transformation = trfs.HistogramMatchingTransformation(template) with pytest.raises(IndexError): transformation(image)
def test_if_template_has_to_have_same_ndim_as_input_image(self): image = np.random.rand(*INPUT_VOLUME_SHAPE_4D) template = np.random.rand(*INPUT_VOLUME_SHAPE_3D) transformation = trfs.HistogramMatchingTransformation(template) with pytest.raises(AssertionError): transformation(image)
def test_if_4_dimensional_array_should_be_passed(self): image = np.random.rand(*INPUT_VOLUME_SHAPE_3D) transformation = trfs.HistogramMatchingTransformation(image) with pytest.raises(IndexError): transformation(image)
def test_if_returns_same_for_identical_images(self): image = np.random.rand(*INPUT_VOLUME_SHAPE_4D) transformation = trfs.HistogramMatchingTransformation(image) assert np.all(image == transformation(image))
def test_if_returns_with_same_shape(self): image = np.random.rand(*INPUT_VOLUME_SHAPE_4D) transformation = trfs.HistogramMatchingTransformation(image) assert image.shape == transformation(image).shape