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)