write test before the add function is written

testMatrix.py

import unittest
from matrixOperations import add

class TestMatrix(unittest.TestCase):
    def test_add(self):
        A = [[1,1,1],[1,1,1],[1,1,1]]
        B = [[1,1,1],[1,1,1],[1,1,1]]
        sum = [[2,2,2],[2,2,2],[2,2,2]]
        self.assertEqual(add(A,B),sum)