def test_save_image(datadirs):
_, formatsdir, _, _ = datadirs
formats = ['gif', 'png', 'jpg', 'bmp', 'tif', 'npy']
for format in formats:
inpath = formatsdir + 'nut_color.' + format
outpath = formatsdir + 'temp_nut_color.' + format
image = ni.load_image(inpath)
ni.save_image(outpath, image)
loaded = ni.load_image(outpath)
os.remove(outpath)
# Saved and loaded JPG images can vary greatly (lossy format) when using
# skimage and a direct comparision fails under Windows 7, Anaconda.
# Therefore, only shape and mean value are verified for JPG images.
if format == 'jpg':
assert abs(np.mean(image) - np.mean(loaded)) < 0.1
assert loaded.shape == (213, 320, 3)
else:
nt.assert_allclose(image, loaded)
def assert_equal_image(imagepath, image, rtol=0.01, atol=0.01):
if CREATE_DATA:
ni.save_image(imagepath, image)
expected = ni.load_image(imagepath)
nt.assert_allclose(expected, image, rtol=rtol, atol=atol)