comparando matrizes numpy contendo NaN

Para o meu teste unitário, quero verificar se duas matrizes são idênticas. Exemplo reduzido:

a = np.array([1, 2, np.NaN])
b = np.array([1, 2, np.NaN])
if np.all(a==b):
    print 'arrays are equal'

Isso não funciona porque nan! = Nan. Qual é a melhor maneira de proceder?

Desde já, obrigado.