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_draw_circle(self):
""" Test draw circle. """
img = CImg(np.zeros((10, 10)), dtype=cimg.uint8)
img.draw_circle(4, 4, 2, [255])
img.save('circle.png')
arr = np.array([
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 1, 1, 1, 0, 0, 0, 0], [0, 0, 1, 1, 1, 1, 1, 0, 0, 0],
[0, 0, 1, 1, 1, 1, 1, 0, 0, 0], [0, 0, 1, 1, 1, 1, 1, 0, 0, 0],
[0, 0, 0, 1, 1, 1, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
]) * 255
img_expected = CImg(arr, dtype=cimg.uint8)
self.assertEqual(img, img_expected)
with self.assertRaises(RuntimeError):
img.draw_circle(4, 4, 2, [255, 0, 0])
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)
