Add tests for the yes_or_no method