class MatrixTest(unittest.TestCase): # Tests def setUp(self): self.matrix = Matrix("9 8 7\n5 3 2\n6 6 7") # def test_extract_row_from_one_number_matrix(self): # matrix = Matrix("1") # self.assertEqual(matrix.row(1), [1]) def test_rows(self): self.assertEqual([[9, 8, 7], [5, 3, 2], [6, 6, 7]], self.matrix.row()) def test_columns__1(self): self.assertEqual([[9, 5, 6], [8, 3, 6], [7, 2, 7]], self.matrix.column()) def test_matrix(self): new_matrix = Matrix("9 8 7 3\n5 3 2 3\n6 6 7 3") self.assertEqual([[9, 8, 7, 3], [5, 3, 2, 3], [6, 6, 7, 3]], new_matrix.row()) self.assertEqual([[9, 5, 6], [8, 3, 6], [7, 2, 7], [3, 3, 3]], new_matrix.column())
def test_extract_row_from_one_number_matrix(self): matrix = Matrix("1") self.assertEqual(matrix.row(1), [1])
def test_can_extract_row_from_non_square_matrix_with_no_corresponding_column( self): matrix = Matrix("1 2 3\n4 5 6\n7 8 9\n8 7 6") self.assertEqual(matrix.row(4), [8, 7, 6])
def test_extract_row_where_numbers_have_different_widths(self): matrix = Matrix("1 2\n10 20") self.assertEqual(matrix.row(2), [10, 20])
def test_can_extract_row(self): matrix = Matrix("1 2\n3 4") self.assertEqual(matrix.row(2), [3, 4])
def test_matrix(self): new_matrix = Matrix("9 8 7 3\n5 3 2 3\n6 6 7 3") self.assertEqual([[9, 8, 7, 3], [5, 3, 2, 3], [6, 6, 7, 3]], new_matrix.row()) self.assertEqual([[9, 5, 6], [8, 3, 6], [7, 2, 7], [3, 3, 3]], new_matrix.column())