Функция Python `assert` ValueError: Значение истинности массива с более чем одним элементом неоднозначно.Используйте a.any () или a.all ()

Question

Я пытаюсь написать функцию assert в Python, чтобы проверить, является ли вывод моей функции определенным массивом значений.

Используя assert simulate(15,0,3) == np.array([15.,15.,15.,15.]), я получаю следующую ошибку:

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

Как проверить, что выход функции представляет собой массив, которыйимеет более одного значения?

Answer 1

функция assert не работает хорошо при работе с numpy.Вместо этого используйте allclose().

это должно работать

print(np.allclose(simulate(15,0,3), np.array([15,15,15,15])))