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)