36 lines
884 B
Python
36 lines
884 B
Python
import numpy as np
|
|
import skimage.io as io
|
|
from skimage._shared import testing
|
|
|
|
|
|
testing.pytest.importorskip('astropy')
|
|
|
|
|
|
@testing.pytest.fixture(autouse=True)
|
|
def _reset_plugin():
|
|
yield
|
|
io.reset_plugins()
|
|
|
|
|
|
def test_imread():
|
|
# Make sure we get an import exception if Astropy isn't there
|
|
# (not sure how useful this is, but it ensures there isn't some other
|
|
# error when trying to load the plugin)
|
|
try:
|
|
io.use_plugin('fits')
|
|
except ImportError:
|
|
raise ()
|
|
|
|
|
|
def _same_ImageCollection(collection1, collection2):
|
|
"""
|
|
Ancillary function to compare two ImageCollection objects, checking that
|
|
their constituent arrays are equal.
|
|
"""
|
|
if len(collection1) != len(collection2):
|
|
return False
|
|
for ext1, ext2 in zip(collection1, collection2):
|
|
if not np.all(ext1 == ext2):
|
|
return False
|
|
return True
|