From da423931b5f10bb11d6662834833cc5a7188607a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jostein=20B=C3=B8=20Fl=C3=B8ystad?= Date: Tue, 18 Jun 2013 22:30:09 +0200 Subject: [PATCH] test_radon_transform: Add helper functions. --- .../transform/tests/test_radon_transform.py | 29 +++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/skimage/transform/tests/test_radon_transform.py b/skimage/transform/tests/test_radon_transform.py index 2f4884ea..7884ef83 100644 --- a/skimage/transform/tests/test_radon_transform.py +++ b/skimage/transform/tests/test_radon_transform.py @@ -6,6 +6,35 @@ from numpy.testing import * from numpy.fft import ifftshift, ifftn import itertools from skimage.transform import * +from skimage.io import imread +from skimage import data_dir + + +__PHANTOM = imread(data_dir + "/phantom.png", as_grey=True)[::2, ::2] + +def _get_phantom(): + return __PHANTOM + + +def _debug_plot(original, result, sinogram=None): + from matplotlib import pyplot as plt + imkwargs = dict(cmap='gray', interpolation='nearest') + if sinogram is None: + plt.figure(figsize=(15, 6)) + sp = 130 + else: + plt.figure(figsize=(11, 11)) + sp = 221 + plt.subplot(sp + 0) + plt.imshow(sinogram, aspect='auto', **imkwargs) + plt.subplot(sp + 1) + plt.imshow(original, **imkwargs) + plt.subplot(sp + 2) + plt.imshow(result, vmin=original.min(), vmax=original.max(), **imkwargs) + plt.subplot(sp + 3) + plt.imshow(result - original, **imkwargs) + plt.colorbar() + plt.show() def rescale(x):