def testExportGreyscale(self): """Try simple greyscale export""" size = (10, 10) dtype = numpy.uint16 data = model.DataArray(numpy.zeros(size, dtype)) # export png.export(FILENAME, data) # check it's here st = os.stat(FILENAME) # this test also that the file is created self.assertGreater(st.st_size, 0) im = Image.open(FILENAME) self.assertEqual(im.format, "PNG") self.assertEqual(im.size, size)
def testExportRGB(self): """Try simple RGB export""" size = (10, 10, 3) dtype = numpy.uint8 metadata = {model.MD_DIMS: "YXC"} data = model.DataArray(numpy.zeros(size, dtype), metadata) # export png.export(FILENAME, data) # check it's here st = os.stat(FILENAME) # this test also that the file is created self.assertGreater(st.st_size, 0) im = Image.open(FILENAME) self.assertEqual(im.format, "PNG") self.assertEqual(im.size, size[:2])