comparando matrices numpy que contienen NaN

Para mi prueba de unidad, quiero comprobar si dos matrices son idénticas. Ejemplo reducido:

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

Esto no funciona porque nan! = Nan. ¿Cuál es la mejor manera de proceder?

Gracias por adelantado.