Add prune tokens tests