e8e61cc0d25d63e9325c75938e82a8f6a27e9848,test/augmenters/test_contrast.py,TestHistogramEqualization,test_basic_functionality_integrationtest,#TestHistogramEqualization#,1466

Before Change

                            assert len(caught_warnings) == 4
                        else:
                            assert len(caught_warnings) == 1
                        hits = []
                        for w in caught_warnings:
                            hits.append(
                                "Got image with 5 channels in _IntensityChannelBasedApplier (parents: ExampleHistEq)"
                                in str(w.message)
                            )
                        assert np.any(hits)
                        // assert "Got image with 5 channels in _IntensityChannelBasedApplier (parents: ExampleHistEq)"\
                        //        in str(caught_warnings[-1].message)

After Change

                        img_aug = aug.augment_image(img)
                        // Verify
                        assert len(caught_warnings) == 1
                        assert (
                            "Got image with 5 channels in _IntensityChannelBasedApplier (parents: ExampleHistEq)"\
                            in str(caught_warnings[-1].message)
                        )

                expected = img
                if nb_channels is None or nb_channels == 1:
                    expected = cv2.equalizeHist(expected)

Instances