как задать массив Numpy в Cython

Пытаюсь освоить cython и ООП. Написал код:

import numpy as np
N = 10
import cython_001
cython_001.test(1, N)

Для него написал cython:

cpdef int test(int x, int N):
    import numpy as np
    cdef int i
    y = np.arange(N, dtype='float64')
    for i in range(N):
        y[i] = i
    print(y[5])
    return y

Появляется сообщение:

TypeError: only size-1 arrays can be converted to Python scalars
Exception ignored in: 'cython_001.test'
TypeError: only size-1 arrays can be converted to Python scalars

1. Как правильно задать тип массива numpy 'y'?
2. Как в будущем задать двумерный массив?