Adding tests around find method when passed a function