Fix final failing test on Py2.6 build

This commit is contained in:
Steven Silvester
2014-12-24 08:49:51 -06:00
parent 2e0deb9595
commit 23ae571689
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -186,7 +186,7 @@ def setup_test():
Set a random seed
"""
warnings.simplefilter('default')
from scipy import signal, ndimage, special, optimize, linalg, sparse
from scipy import signal, ndimage, special, optimize, linalg
from scipy.io import loadmat
from skimage import viewer, filter
np.random.seed(0)
@@ -99,7 +99,7 @@ def test_2d_cg_mg():
lx = 70
ly = 100
data, labels = make_2d_syntheticdata(lx, ly)
with expected_warnings(PYAMG_EXPECTED_WARNING):
with expected_warnings('%s|\A\Z' % PYAMG_EXPECTED_WARNING):
labels_cg_mg = random_walker(data, labels, beta=90, mode='cg_mg')
assert (labels_cg_mg[25:45, 40:60] == 2).all()
assert data.shape == labels.shape