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)