From db5460591abecf55bd09e2d76721b66efd213557 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Johannes=20Sch=C3=B6nberger?= Date: Tue, 25 Dec 2012 11:33:16 +0100 Subject: [PATCH] Add test case for multi-channel chambolle denoising --- skimage/filter/tests/test_denoise.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/skimage/filter/tests/test_denoise.py b/skimage/filter/tests/test_denoise.py index c8e7bb02..cb3b5da3 100644 --- a/skimage/filter/tests/test_denoise.py +++ b/skimage/filter/tests/test_denoise.py @@ -1,5 +1,5 @@ import numpy as np -from numpy.testing import run_module_suite, assert_raises +from numpy.testing import run_module_suite, assert_raises, assert_equal from skimage import filter, data, color, img_as_float @@ -29,6 +29,12 @@ def test_denoise_tv_chambolle_2d(): < np.sqrt((grad**2).sum()) / 2) +def test_denoise_tv_chambolle_multichannel(): + denoised0 = filter.denoise_tv_chambolle(lena[..., 0], weight=60.0) + denoised = filter.denoise_tv_chambolle(lena, weight=60.0, multichannel=True) + assert_equal(denoised[..., 0], denoised0) + + def test_denoise_tv_chambolle_float_result_range(): # lena image img = lena_gray