mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-08 10:39:13 +08:00
Steven Silvester
0e61374a89
Add a helper function to check for low contrast Add a check for low contrast when using imsave Use the low contrast helper in imshow and make sure warnings are always shown Clean up parameter names and add doctests Remove unnecessary warning context Remove unnecessary warning context Add dtype ranges for 64bit types Update tests with new warnings Fix doctest logic Fix doctest logic Add a low contrast test with multiple dtypes Fix check for color images Fix color check again Add support for int32 types Relax assertion for 32bit builds Add a low contrast test with multiple dtypes Add a low contrast test with multiple dtypes Fix check for color images Fix color check again Add support for int32 types
109 lines
2.8 KiB
Python
109 lines
2.8 KiB
Python
from __future__ import division
|
|
|
|
import numpy as np
|
|
from skimage import io
|
|
from skimage._shared._warnings import expected_warnings
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
def setup():
|
|
io.reset_plugins()
|
|
|
|
# test images. Note that they don't have their full range for their dtype,
|
|
# but we still expect the display range to equal the full dtype range.
|
|
im8 = np.array([[0, 64], [128, 240]], np.uint8)
|
|
im16 = im8.astype(np.uint16) * 256
|
|
im64 = im8.astype(np.uint64)
|
|
imf = im8 / 255
|
|
im_lo = imf / 1000
|
|
im_hi = imf + 10
|
|
|
|
|
|
|
|
def n_subplots(ax_im):
|
|
"""Return the number of subplots in the figure containing an ``AxesImage``.
|
|
|
|
Parameters
|
|
----------
|
|
ax_im : matplotlib.pyplot.AxesImage object
|
|
The input ``AxesImage``.
|
|
|
|
Returns
|
|
-------
|
|
n : int
|
|
The number of subplots in the corresponding figure.
|
|
|
|
Notes
|
|
-----
|
|
This function is intended to check whether a colorbar was drawn, in
|
|
which case two subplots are expected. For standard imshows, one
|
|
subplot is expected.
|
|
"""
|
|
return len(ax_im.get_figure().get_axes())
|
|
|
|
|
|
def test_uint8():
|
|
ax_im = io.imshow(im8)
|
|
assert ax_im.cmap.name == 'gray'
|
|
assert ax_im.get_clim() == (0, 255)
|
|
# check that no colorbar was created
|
|
assert n_subplots(ax_im) == 1
|
|
assert ax_im.colorbar is None
|
|
|
|
|
|
def test_uint16():
|
|
ax_im = io.imshow(im16)
|
|
assert ax_im.cmap.name == 'gray'
|
|
assert ax_im.get_clim() == (0, 65535)
|
|
assert n_subplots(ax_im) == 1
|
|
assert ax_im.colorbar is None
|
|
|
|
|
|
def test_float():
|
|
ax_im = io.imshow(imf)
|
|
assert ax_im.cmap.name == 'gray'
|
|
assert ax_im.get_clim() == (0, 1)
|
|
assert n_subplots(ax_im) == 1
|
|
assert ax_im.colorbar is None
|
|
|
|
|
|
def test_low_dynamic_range():
|
|
with expected_warnings(["Low image dynamic range"]):
|
|
ax_im = io.imshow(im_lo)
|
|
assert ax_im.get_clim() == (im_lo.min(), im_lo.max())
|
|
# check that a colorbar was created
|
|
assert n_subplots(ax_im) == 2
|
|
assert ax_im.colorbar is not None
|
|
|
|
|
|
def test_outside_standard_range():
|
|
plt.figure()
|
|
with expected_warnings(["out of standard range"]):
|
|
ax_im = io.imshow(im_hi)
|
|
assert ax_im.get_clim() == (im_hi.min(), im_hi.max())
|
|
assert n_subplots(ax_im) == 2
|
|
assert ax_im.colorbar is not None
|
|
|
|
|
|
def test_nonstandard_type():
|
|
plt.figure()
|
|
with expected_warnings(["Non-standard image type",
|
|
"Low image dynamic range"]):
|
|
ax_im = io.imshow(im64)
|
|
assert ax_im.get_clim() == (im64.min(), im64.max())
|
|
assert n_subplots(ax_im) == 2
|
|
assert ax_im.colorbar is not None
|
|
|
|
|
|
def test_signed_image():
|
|
plt.figure()
|
|
im_signed = np.array([[-0.5, -0.2], [0.1, 0.4]])
|
|
ax_im = io.imshow(im_signed)
|
|
assert ax_im.get_clim() == (-0.5, 0.5)
|
|
assert n_subplots(ax_im) == 2
|
|
assert ax_im.colorbar is not None
|
|
|
|
|
|
if __name__ == '__main__':
|
|
np.testing.run_module_suite()
|