def test_2d_cg():
lx = 70
ly = 100
data, labels = make_2d_syntheticdata(lx, ly)
labels_cg = random_walker(data, labels, beta=90, mode='cg_mg')
assert (labels_cg[25:45, 40:60] == 2).all()
return data, labels_cg
def test_2d():
lx = 70
ly = 100
data, labels = make_2d_syntheticdata(lx, ly)
labels_bf = random_walker(data, labels, beta=90)
assert (labels_bf[25:45, 40:60] == 2).all()
return data, labels_bf
def test_3d():
n=30
lx, ly, lz = n, n, n
data, labels = make_3d_syntheticdata(lx, ly, lz)
labels = random_walker(data, labels)
assert (labels.reshape(data.shape)[13:17,13:17,13:17] == 2).all()
return data, labels
def test_2d_inactive():
lx = 70
ly = 100
data, labels = make_2d_syntheticdata(lx, ly)
labels[10:20, 10:20] = -1
labels[46:50, 33:38] = -2
labels = random_walker(data, labels, beta=90)
assert (labels.reshape((lx, ly))[25:45, 40:60] == 2).all()
return data, labels
def test_3d_inactive():
n=30
lx, ly, lz = n, n, n
data, labels = make_3d_syntheticdata(lx, ly, lz)
old_labels = np.copy(labels)
labels[5:25, 26:29, 26:29] = -1
after_labels = np.copy(labels)
labels = random_walker(data, labels)
assert (labels.reshape(data.shape)[13:17,13:17,13:17] == 2).all()
return data, labels, old_labels, after_labels
