def test_error_handling(self):
with np.testing.assert_raises(TypeError):
sp.seed_competition(self.seeds, image=0)
with np.testing.assert_raises(TypeError):
sp.seed_competition(self.seeds.flatten(), graph=0)
with np.testing.assert_raises(TypeError):
sp.dynamic_arc_weight(self.seeds, image=0)
with np.testing.assert_raises(ValueError):
sp.seed_competition(self.seeds, np.ones(self.seeds.size))
with np.testing.assert_raises(ValueError):
sp.dynamic_arc_weight(self.seeds, np.ones(self.seeds.size))
with np.testing.assert_raises(ValueError):
sp.seed_competition(self.seeds)
with np.testing.assert_raises(ValueError):
sp.seed_competition(self.seeds, image=self.image,
graph=csgraph.csgraph_from_dense(self.image))
with np.testing.assert_raises(ValueError):
sp.dynamic_arc_weight(self.seeds, self.image, alpha=-1.0)
with np.testing.assert_raises(ValueError):
sp.dynamic_arc_weight(self.seeds, self.image, mode='fake')
def test_dynamic_arc_weight_grid_exp_decay(self):
costs, roots, preds, labels, avgs = sp.dynamic_arc_weight(self.seeds, self.image, mode='exp')
sqrt2 = np.sqrt(2.0)
np.testing.assert_equal(costs, np.array([[0, sqrt2, 1.5],
[sqrt2, sqrt2, 1.5],
[1, 0, 1]]))
np.testing.assert_equal(roots, np.array([[0, 0, 7],
[0, 7, 7],
[7, 7, 7]]))
np.testing.assert_equal(preds, np.array([[-1, 0, 5],
[0, 7, 8],
[7, -1, 7]]))
np.testing.assert_equal(labels, np.array([[1, 1, 2],
[1, 2, 2],
[2, 2, 2]]))
np.testing.assert_equal(avgs, np.array([[[1, 5], [1.5, 5.5], [3.125, 8.5]],
[[1.5, 5.5], [2.5, 8.5], [3.25, 9]],
[[2, 8.5], [2, 9], [2.5, 9]]]))
def test_dynamic_arc_weight_grid_label(self):
costs, roots, preds, labels, trees = sp.dynamic_arc_weight(
self.seeds, self.image, mode='label')
sqrt2 = np.sqrt(2.0)
np.testing.assert_equal(
costs, np.array([[0, sqrt2, 1.5], [sqrt2, sqrt2, 1.5], [1, 0, 1]]))
np.testing.assert_equal(roots,
np.array([[0, 0, 7], [0, 7, 7], [7, 7, 7]]))
np.testing.assert_equal(preds,
np.array([[-1, 0, 5], [0, 7, 8], [7, -1, 7]]))
np.testing.assert_equal(labels,
np.array([[1, 1, 2], [1, 2, 2], [2, 2, 2]]))
expected_trees = {
1: (3, np.array([5 / 3, 17 / 3])),
2: (6, np.array([17 / 6, 8.5]))
}
np.testing.assert_(expected_trees.keys() == trees.keys())
for e_tree, tree in zip(expected_trees.values(), trees.values()):
np.testing.assert_(e_tree[0] == tree[0])
np.testing.assert_allclose(e_tree[1], tree[1])
