Best way to assert for numpy.array equality?

user_315577 0
Viewed 76893

I want to make some unit-tests for my app, and I need to compare two arrays. Since array.__eq__ returns a new array (so TestCase.assertEqual fails), what is the best way to assert for equality?

Currently I'm using

self.assertTrue((arr1 == arr2).all())

but I don't really like it

numpy unit-testing python
8 Answers
self.assertTrue(np.array_equal(x, y, equal_nan=True))

equal_nan = True if you want to np.nan == np.nan returns True

or you can use numpy.allclose to compare with torelance.

user_30854810

np.linalg.norm(arr1 - arr2) < 1e-6

user_68470860

Use numpy

numpy.array_equal(a, b)
user_10516240
Related