def test_adding_some_values(self): columns = 2 rows = 3 matrix = Matrix(rows, columns) matrix.setValue(2, 1, 2) matrix.setValue(3, 2, 6) self.assertEqual(2, matrix.getValue(2, 1)) self.assertEqual(6, matrix.getValue(3, 2))
def test_representation(self): columns = 2 rows = 3 matrix = Matrix(rows, columns) self.assertEqual(matrix.columnCount, 2) self.assertEqual(matrix.rowCount, 3) for column in range(columns): for row in range(rows): self.assertEqual(matrix.getValue(column, row), 0)