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])
