def test_fromarray(self): """ Test construction from array. """ arr = np.ones((100, 50)) im = CImg() im.fromarray(arr) self.assertTrue(np.allclose(arr, im.asarray().squeeze())) invalid_arr = np.ones((2, 3, 4, 5, 6)) self.assertRaises(RuntimeError, im.fromarray, invalid_arr)
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)