Files
scikit-image/skimage/io/_plugins/test_freeimage.py
T
2012-02-02 16:56:54 -08:00

28 lines
846 B
Python

import os
import skimage as si
import skimage.io as sio
import skimage.io._plugins.freeimage_plugin as fi
from numpy.testing import *
def test_read():
sio.use_plugin('freeimage', 'imread')
img = sio.imread(os.path.join(si.data_dir, 'color.png'))
assert img.shape == (370, 371, 3)
assert all(img[274,135] == [0, 130, 253])
def test_metadata():
meta = fi.read_metadata(os.path.join(si.data_dir, 'multipage.tif'))
assert meta[('EXIF_MAIN', 'BitsPerSample')] == 8
assert meta[('EXIF_MAIN', 'Software')].startswith('ImageMagick')
meta = fi.read_multipage_metadata(os.path.join(si.data_dir, 'multipage.tif'))
assert len(meta) == 2
assert meta[0][('EXIF_MAIN', 'BitsPerSample')] == 8
assert meta[1][('EXIF_MAIN', 'Software')].startswith('ImageMagick')
if __name__ == "__main__":
run_module_suite()