сравнение пустых массивов, содержащих NaN

Для моего модульного теста я хочу проверить, идентичны ли два массива. Сокращенный пример:

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

Это не работает, потому что nan != nan. Как лучше поступить?