def test_image_no_parameters(self, factory):
image = factory()
serializer = B64ImageSerializer(image)
image_itk = image.get_sitk_image()
image_pil = B64ImageSerializer.convert_itk_to_pil(image_itk)
image_bytes = B64ImageSerializer.create_thumbnail_as_b64(image_pil)
assert serializer.data["content"] == image_bytes
decoded_image_pil = PILImage.open(
BytesIO(b64decode(serializer.data["content"])))
assert decoded_image_pil.size == image_pil.size
def test_image_resizes(self, factory, max_dimension):
if max_dimension == "random":
max_dimension = random.randint(1, 255)
else:
max_dimension = settings.RETINA_DEFAULT_THUMBNAIL_SIZE
image = factory()
serializer_context = {"width": max_dimension, "height": max_dimension}
serializer = B64ImageSerializer(image, context=serializer_context)
image_itk = image.get_sitk_image()
image_pil = B64ImageSerializer.convert_itk_to_pil(image_itk)
image_pil.thumbnail((max_dimension, max_dimension), PILImage.ANTIALIAS)
decoded_image_pil = PILImage.open(
BytesIO(b64decode(serializer.data["content"])))
assert decoded_image_pil.size == image_pil.size
assert max(decoded_image_pil.size) == max_dimension