I would import assert_array_almost_equal from np.testing for consitency