def testgetMetadataFromImage(self):
"""Test if metadata is extracted from image correctly"""
png_file = open("data/test.png").read()
handler = Handler(self.tmp_url, png_file, "png", **self.kw)
metadata = handler.getMetadata()
self.assertEquals(metadata.get("Compression"), "Zip")
self.assertEquals(metadata.get("Colorspace"), "sRGB")
self.assertEquals(metadata.get("Alpha color"), "grey74")
def testConvertPNGtoJPG(self):
"""Test conversion of png to jpg"""
png_file = open("data/test.png").read()
handler = Handler(self.tmp_url, png_file, "png", **self.kw)
jpg_file = handler.convert("jpg")
mime = magic.Magic(mime=True)
jpg_mimetype = mime.from_buffer(jpg_file)
self.assertEquals("image/jpeg", jpg_mimetype)
def testgetMetadataFromImage(self):
"""Test if metadata is extracted from image correctly"""
png_file = open("data/test.png").read()
handler = Handler(self.tmp_url, png_file, "png", **self.kw)
metadata = handler.getMetadata()
self.assertEquals(metadata.get("Compression"), "Zip")
self.assertEquals(metadata.get("Colorspace"), "sRGB")
self.assertEquals(metadata.get("Alpha color"), "grey74")
def testConvertPNGtoJPG(self):
"""Test conversion of png to jpg"""
png_file = open("data/test.png").read()
handler = Handler(self.tmp_url, png_file, "png", **self.kw)
jpg_file = handler.convert("jpg")
mime = magic.Magic(mime=True)
jpg_mimetype = mime.from_buffer(jpg_file)
self.assertEquals("image/jpeg", jpg_mimetype)
def getImages(images):
"""This function verify if there is ppm and pbm and converts it to png
before return the list of images in document"""
imagesList = []
images.sort()
id = 0
for image in images:
id += 1
page = int(image.split("/-")[-1].split("_")[0])
extension = image.split(".")[-1]
title = image.split("-")[0] + "%.3d-pag%.3d." % (id, page) + extension
if extension in ("ppm", "pbm"):
img = Handler("/".join(image.split("/")[:-1]), open(image).read(), image.split(".")[-1])
new_image = image.split(".")[0] + ".png"
content = img.convert("png")
open(new_image, "w").write(content)
img = open(new_image).read()
imagesList.append([title.split("/")[-1], img])
else:
imagesList.append([title.split("/")[-1], open(image).read()])
return imagesList
def testsetMetadata(self):
""" Test if metadata are inserted correclty """
handler = Handler(self.tmp_url, "", "png", **self.kw)
self.assertRaises(NotImplementedError, handler.setMetadata)
