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