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_from_file(self):
""" Test construction from file. """
im = CImg(get_test_image())
self.assertEqual(im.shape, (3, 1, 797, 1200))
def test_load_tiff(self):
""" Test loading a TIFF file. """
img = CImg()
img.load_tiff(get_test_image('tiff'))
self._check_image_dimensions(img)
def test_load_png(self):
""" Test loading a PNG file. """
img = CImg()
img.load_png(get_test_image('png'))
self._check_image_dimensions(img)
def test_load_jpeg(self):
""" Test loading a JPEG file. """
img = CImg()
img.load_jpeg(get_test_image('jpg'))
self._check_image_dimensions(img)
def test_load_bmp(self):
""" Test loading a BMP file. """
img = CImg()
img.load_bmp(get_test_image('bmp'))
self._check_image_dimensions(img)
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)