def test_save(self):
""" Test save. """
img = CImg()
img.load(get_test_image())
for ext in ['.bmp', '.jpg', '.png', '.cimg', '.tiff']:
filename = self._get_testfilename() + ext
img.save(filename)
self.assertTrue(os.path.isfile(filename))
os.remove(filename)
def test_save_load(self):
""" Test save/load half float. """
im = CImg()
arr = np.random.randn(3, 2, 500, 300)
im.fromarray(arr)
self.assertTrue(np.allclose(arr, im.asarray()))
filename = self._get_testfilename() + '.cimg'
im.save(filename)
im2 = CImg()
im2.load(filename)
self.assertTrue(np.allclose(im2.asarray(), im.asarray()))
os.remove(filename)
# save/load half float
filename = self._get_testfilename() + '.cimg'
im.save_cimg_float16(filename)
im3 = CImg()
im3.load_cimg_float16(filename)
self.assertTrue(np.allclose(im2.asarray(), im.asarray()))
os.remove(filename)
def test_load(self):
""" Test load. """
img = CImg()
self.assertRaises(RuntimeError, img.load, 'notexistent.jpg')
img.load(get_test_image())
self._check_image_dimensions(img)
