def test_posterize_checks(): img = np.random.random([256, 256, 3]) with pytest.raises(TypeError) as exc_info: F.posterize(img, 4) assert str(exc_info.value) == "Image must have uint8 channel type" img = np.random.randint(0, 256, [256, 256], dtype=np.uint8) with pytest.raises(TypeError) as exc_info: F.posterize(img, [1, 2, 3]) assert str(exc_info.value) == "If bits is iterable image must be RGB"
def albumentations(self, img): return albumentations.posterize(img, 4)