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)