def test_get_ltm_cells(self): "get_ltm_cells converts indices to be below the diagonal" cells = [(0,0),(0,1),(0,2),(1,0),(1,1),(1,2),(2,0),(2,1),(2,2)] result = get_ltm_cells(cells) self.assertEqual(result, [(2, 0), (1, 0), (2, 1)]) cells = [(0,1),(0,2)] result = get_ltm_cells(cells) self.assertEqual(result, [(2, 0), (1, 0)])
def test_get_ltm_cells(self): "get_ltm_cells converts indices to be below the diagonal" cells = [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)] result = get_ltm_cells(cells) self.assertEqual(result, [(2, 0), (1, 0), (2, 1)]) cells = [(0, 1), (0, 2)] result = get_ltm_cells(cells) self.assertEqual(result, [(2, 0), (1, 0)])
def test_get_values_from_matrix(self): """get_values_from_matrix returns the special and other values from matrix""" matrix = self.matrix cells = [(1, 0), (0, 1), (2, 0), (2, 1)] #test that works for a symmetric matrix cells_sym = get_ltm_cells(cells) special_vals, other_vals = get_values_from_matrix(matrix, cells_sym,\ cells2=None, is_symmetric=True) special_vals.sort() other_vals.sort() self.assertEqual(special_vals, [5, 9, 10]) self.assertEqual(other_vals, [13, 14, 15]) #test that work for a non symmetric matrix special_vals, other_vals = get_values_from_matrix(matrix, cells,\ cells2=None, is_symmetric=False) special_vals.sort() other_vals.sort() self.assertEqual(special_vals, [2, 5, 9, 10]) self.assertEqual(other_vals, [1, 3, 4, 6, 7, 8, 11, 12, 13, 14, 15, 16]) #test that works on a symmetric matrix when cells2 is defined cells2 = [(3, 0), (3, 2), (0, 3)] cells2_sym = get_ltm_cells(cells2) special_vals, other_vals = get_values_from_matrix(matrix, cells_sym,\ cells2=cells2_sym, is_symmetric=True) special_vals.sort() other_vals.sort() self.assertEqual(special_vals, [5, 9, 10]) self.assertEqual(other_vals, [13, 15]) #test that works when cells2 is defined and not symmetric special_vals, other_vals = get_values_from_matrix(matrix, cells, cells2=cells2,\ is_symmetric=False) special_vals.sort() other_vals.sort() self.assertEqual(special_vals, [2, 5, 9, 10]) self.assertEqual(other_vals, [4, 13, 15])
def test_get_values_from_matrix(self): """get_values_from_matrix returns the special and other values from matrix""" matrix = self.matrix cells = [(1,0),(0,1),(2,0),(2,1)] #test that works for a symmetric matrix cells_sym = get_ltm_cells(cells) special_vals, other_vals = get_values_from_matrix(matrix, cells_sym,\ cells2=None, is_symmetric=True) special_vals.sort() other_vals.sort() self.assertEqual(special_vals, [5,9,10]) self.assertEqual(other_vals, [13,14,15]) #test that work for a non symmetric matrix special_vals, other_vals = get_values_from_matrix(matrix, cells,\ cells2=None, is_symmetric=False) special_vals.sort() other_vals.sort() self.assertEqual(special_vals, [2,5,9,10]) self.assertEqual(other_vals, [1,3,4,6,7,8,11,12,13,14,15,16]) #test that works on a symmetric matrix when cells2 is defined cells2 = [(3,0),(3,2),(0,3)] cells2_sym = get_ltm_cells(cells2) special_vals, other_vals = get_values_from_matrix(matrix, cells_sym,\ cells2=cells2_sym, is_symmetric=True) special_vals.sort() other_vals.sort() self.assertEqual(special_vals, [5,9,10]) self.assertEqual(other_vals, [13,15]) #test that works when cells2 is defined and not symmetric special_vals, other_vals = get_values_from_matrix(matrix, cells, cells2=cells2,\ is_symmetric=False) special_vals.sort() other_vals.sort() self.assertEqual(special_vals, [2,5,9,10]) self.assertEqual(other_vals, [4,13,15])