It's better to use `assert_array_almost_equal` instead of `assert_true(np.allclose())` to get meaningful failure messages.