def testFindOneFragment(self):
a = dok_matrix((2, 2))
frags, branches = find_fragments(a)
good_frags = [{0}, {1}]
self.assertEqual(frags, good_frags)
good_branches = [0, 0]
self.assertEqual(branches, good_branches)
def testFindThreeFragments(self):
# does not increase coverage, but that's okay
a = dok_matrix((5, 5))
a[0, 4] = 1
a[4, 0] = 1
frags, branches = find_fragments(a)
good_frags = [{0, 4}, {1}, {2}, {3}]
self.assertEqual(frags, good_frags)
good_branches = [0, 0, 0, 0]
self.assertEqual(branches, good_branches)
def testFindTwoFragments(self):
a_array = [[0., 1., 1., 0., 0.], [1., 0., 0., 0., 0.],
[1., 0., 0., 0., 0.], [0., 0., 0., 0., 1.],
[0., 0., 0., 1., 0.]]
a = dok_matrix(a_array)
frags, branches = find_fragments(a)
good_frags = [{0, 1, 2}, {3, 4}]
self.assertEqual(frags, good_frags)
good_branches = [0, 0]
self.assertEqual(branches, good_branches)