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)