def shape(M):
    a = 0
    b = len(M)
    for i in M:
        if a<len(i):
            a=len(i)
    return b,a

%run -i -e test.py LinearRegressionTestCase.test_shape

 def test_shape(self):

        for _ in range(10):
            r,c = np.random.randint(low=1,high=25,size=2)
            matrix = np.random.randint(low=-10,high=10,size=(r,c))
            self.assertEqual(shape(matrix.tolist()),(r,c),'Wrong answer')

E

ERROR: test_shape (__main__.LinearRegressionTestCase)

Traceback (most recent call last):
File "C:UsersderekDesktopudacity???3linear_algebra-mastertest.py", line 14, in test_shape

self.assertEqual(shape(matrix.tolist()),(r,c),'Wrong answer')

NameError: global name 'shape' is not defined

Ran 1 test in 0.004s

FAILED (errors=1)