def test_condense_node_order(self): """condense_node_order condenses nodes in list based on index info """ matrix = self.matrix index = find_smallest_index(matrix) node_order = self.node_order node_order = condense_node_order(matrix, index, node_order) self.assertEqual(node_order[1], None) self.assertEqual(node_order[0].__str__(), '(a:0.5,b:0.5);') self.assertEqual(node_order[2].__str__(), 'c;') self.assertEqual(node_order[3].__str__(), 'd;') self.assertEqual(node_order[4].__str__(), 'e;')
def test_condense_matrix(self): """condense_array joins two rows and columns identified by indices """ matrix = self.matrix index = find_smallest_index(matrix) result = condense_matrix(matrix, index, 9999999999) self.assertFloatEqual(result[0, 0], 5000000.0) self.assertEqual(result[1, 4], 9999999999) self.assertEqual(result[0, 1], 9999999999) self.assertEqual(result[0, 2], 4.5) self.assertEqual(result[2, 0], 4.5) self.assertEqual(result[0, 4], 22.5) self.assertEqual(result[4, 4], 9999999) self.assertEqual(result[4, 0], 22.5)
def test_find_smallest_index(self): """find_smallest_index returns the index of smallest value in array """ matrix = self.matrix index = find_smallest_index(matrix) self.assertEqual(index, (0, 1))
def test_find_smallest_index(self): """find_smallest_index returns the index of smallest value in array """ matrix = self.matrix index = find_smallest_index(matrix) self.assertEqual(index, (0,1))