with pytest.raise() 로 에러 일으키기

minsing-jin·2023년 11월 23일
0

Ragchain에서 나온 코드중

def test_ko_strategy_qa_evaluator(strategy_qa_evaluator):
    with pytest.raises(ValueError):
        strategy_qa_evaluator.evaluate(validate_passages=True)

    result = strategy_qa_evaluator.evaluate(validate_passages=False)
    assert len(result.each_results) == 5
    assert result.each_results.iloc[0][
               'question'] == 'Are more people today related to Genghis Khan than Julius Caesar?'

with의 아래에 예외가 발생하면 pytest.raises()안의 파라미터에서 지정한 에러를 일으킨다.

위의 코드를 예시로 들면 strategy_qa_evaluator.evaluate(validate_passages=True)의 validate_passages가 True가 아닌 False라면 pytest.raises(ValueError)ValueError를 일으킨다.

네이스한 사이트

profile
why not? 정신으로 맨땅에 헤딩하고 있는 코린이

0개의 댓글