Unittest – как проверить, равны ли два возможных значения NaN ⇐ Python

[Anonymous]

В моем тестовом примере я предполагаю, что если два значения равны NaN, то они равны. Как это выразить с помощью утверждений unittest? Две общие функции, представленные ниже, не обрабатывают этот случай.

v1 = np.nan
v2 = np.nan
self.assertEquals(v1, v2)
self.assertTrue(v1 == v2)


Решение, которое сейчас работает для меня, — это использование логического выражения внутри AssertTrue:

self.assertTrue(v1 == v2 or (np.isnan(v1) and np.isnan(v2))


Подробнее здесь: https://stackoverflow.com/questions/517 ... -are-equal