'Answer annotation problem.'

def answer_problem(self, correctness):

    if (correctness == 'correct'):
        choice = 0
    elif (correctness == 'partially-correct'):
        choice = 2
    else:
        choice = 1
    answer = 'Student    comment'
    self.problem_page.q(css='div.problem    textarea.comment').fill(answer)
    self.problem_page.q(css='div.problem    span.tag'.format(choice=choice)).nth(choice).click()
