test_n_size_ndarray_creation 문제에 대해 질문드립니다.

n이 주어짐에 따라 행렬을 만드는 문제인데 계속 통과하지 못해서 질문드립니다. returns의 설명이 행렬의 모든 원소를 할당시키는 것인지 0,0과 n-1,n-1만 할당시키는 것인지 잘 모르겠습니다. 다음은 제가 작성한 코드입니다.

def n_size_ndarray_creation(n, dtype=np.int):
    X = np.zeros((n,n),dtype=dtype)
    count=0
    for i in range(n):
        for j in range(n):
            X[i][j]=count
            count+=1
    return X