def test_auto_convert_color_palette_mode_image_pil_image(self): """ Test auto convert palette-mode png image during VLImage loading from PIL image """ with open(PALETTE_MODE, "rb") as f: imageInBytes = f.read() vlImageFromPil = VLImage(PIL.Image.open(io.BytesIO(imageInBytes))) vlImageFromBytes = VLImage(imageInBytes) assert vlImageFromPil.asNPArray().tobytes( ) == vlImageFromBytes.asNPArray().tobytes()
def test_check_ndarray_type(self): """ Test check numpy array conversion """ image = Image.open(ONE_FACE) imageWithOneFace = VLImage(image) assert (np.asarray(image) == imageWithOneFace.asNPArray()).all()