wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-08-01 12:50:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix python3 test and real values image.

Browse Source
This commit is contained in:
François Orieux
2013-12-10 22:45:16 +01:00
parent bda9b4fb3c
commit 15a96d5300
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/restoration/tests/test_restoration.py
+8 -4
View File
@@ -26,7 +26,9 @@ def test_wiener():
deconvolved = restoration.wiener(data, otf, 0.05,
reg=laplacian,
is_real=False)
np.testing.assert_allclose(deconvolved, np.load(path))
np.testing.assert_allclose(np.real(deconvolved),
np.load(path),
rtol=1e-3)
def test_unsupervised_wiener():
@@ -42,11 +44,13 @@ def test_unsupervised_wiener():
_, laplacian = uft.laplacian(2, data.shape)
otf = uft.ir2tf(psf, data.shape, is_real=False)
np.random.seed(0)
deconvolved = np.real(restoration.unsupervised_wiener(
deconvolved = restoration.unsupervised_wiener(
data, otf, reg=laplacian, is_real=False,
user_params={"callback": lambda x: None})[0])
user_params={"callback": lambda x: None})[0]
path = pjoin(dirname(abspath(__file__)), 'camera_unsup2.npy')
np.testing.assert_allclose(deconvolved, np.load(path))
np.testing.assert_allclose(np.real(deconvolved),
np.load(path),
rtol=1e-3)
def test_richardson_lucy():