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)
