ff1bfcb0499605f2cedf9b2aa53dafff2cd4a8c0,test/augmenters/test_contrast.py,TestCLAHE,_test_single_image_3d_rgb_to_x,#Any#Any#Any#,1098

Before Change

        mock_all_channel_clahe._augment_images.side_effect = \
            side_effect_all_channel_clahe

        mock_change_colorspace_inv = mock.Mock()
        mock_change_colorspace_inv._augment_images.side_effect = \
            side_effect_change_colorspace_inv

        clahe = iaa.CLAHE(
            clip_limit=1,
            tile_grid_size_px=3,
            tile_grid_size_px_min=2,
            from_colorspace=iaa.CLAHE.RGB,
            to_colorspace=to_colorspace)
        clahe.all_channel_clahe = mock_all_channel_clahe
        clahe.intensity_channel_based_applier.change_colorspace = \
            mock_change_cs
        clahe.intensity_channel_based_applier.change_colorspace_inv = \
            mock_change_colorspace_inv

        img3d_aug = clahe.augment_image(np.copy(img3d))
        expected1 = img3d + 1
        expected2 = np.copy(expected1)
        expected2[..., channel_idx] += 2
        expected3 = np.copy(expected2) + 3
        assert np.array_equal(img3d_aug, expected3)

        mock_change_cs = mock_change_cs._augment_images
        mock_all_channel_clahe = mock_all_channel_clahe._augment_images
        mock_change_colorspace_inv = mock_change_colorspace_inv._augment_images

        assert mock_change_cs.call_count == 1
        assert mock_all_channel_clahe.call_count == 1
        assert mock_change_colorspace_inv.call_count == 1

After Change

    @classmethod
    def _test_single_image_3d_rgb_to_x(cls, to_colorspace, channel_idx):
        fname_cs = "imgaug.augmenters.color.change_colorspace_"
        with mock.patch(fname_cs) as mock_cs:
            img = [
                [0, 1, 2, 3, 4],
                [5, 6, 7, 8, 9],

Instances