This commit is contained in:
Evan Lohn
2024-10-06 10:24:29 -04:00
parent bedcc984f5
commit 6e2b29314d

View File

@@ -3,6 +3,7 @@ from typing import cast
from unittest.mock import Mock from unittest.mock import Mock
import pytest import pytest
from pytest_mock import MockerFixture
from danswer.llm.answering.answer import Answer from danswer.llm.answering.answer import Answer
from danswer.one_shot_answer.answer_question import AnswerObjectIterator from danswer.one_shot_answer.answer_question import AnswerObjectIterator
@@ -113,14 +114,14 @@ class FinishedTestException(Exception):
) )
@pytest.mark.skip(reason="not working") @pytest.mark.skip(reason="not working")
def test_run_qa_skip_gen_ai( def test_run_qa_skip_gen_ai(
config: dict[str, Any], questions: list[dict[str, Any]], mocker config: dict[str, Any], questions: list[dict[str, Any]], mocker: MockerFixture
) -> None: ) -> None:
mocker.patch( mocker.patch(
"tests.regression.answer_quality.run_qa._initialize_files", "tests.regression.answer_quality.run_qa._initialize_files",
return_value=("test", questions), return_value=("test", questions),
) )
def arg_checker(question_data: dict, config: dict, question_number: int): def arg_checker(question_data: dict, config: dict, question_number: int) -> None:
assert question_data == questions[0] assert question_data == questions[0]
raise FinishedTestException() raise FinishedTestException()