Check that also works on numpy arrays