Please use `assert_almost_equal` instead of `np.array_equal` throughout.