It might make more sense to include `np.all` here, rather than place the burden on the callers of this function.

Also, maybe add a docstring.  (I know it's private, but it might not be clear what its purpose is a few months from now).