sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-27 09:25:06 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Update file summarization unit tests now that multiple files are allowed

Browse source
This commit is contained in:
sabaimran 2024-11-04 17:45:54 -08:00
parent 7543360210
commit 1f372bf2b1
2 changed files with 3 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_offline_chat_director.py
View file

@ -10,7 +10,7 @@ from khoj.processor.conversation.utils import message_to_log
from khoj.routers.helpers import aget_relevant_information_sources from khoj.routers.helpers import aget_relevant_information_sources
from tests.helpers import ConversationFactory from tests.helpers import ConversationFactory
SKIP_TESTS = True SKIP_TESTS = False
pytestmark = pytest.mark.skipif( pytestmark = pytest.mark.skipif(
SKIP_TESTS, SKIP_TESTS,
reason="Disable in CI to avoid long test runs.", reason="Disable in CI to avoid long test runs.",
@ -337,7 +337,6 @@ def test_summarize_one_file(client_offline_chat, default_user2: KhojUser):
# Assert # Assert
assert response_message != "" assert response_message != ""
assert response_message != "No files selected for summarization. Please add files using the section on the left." assert response_message != "No files selected for summarization. Please add files using the section on the left."
assert response_message != "Only one file can be selected for summarization."
@pytest.mark.django_db(transaction=True) @pytest.mark.django_db(transaction=True)
@ -375,7 +374,6 @@ def test_summarize_extra_text(client_offline_chat, default_user2: KhojUser):
# Assert # Assert
assert response_message != "" assert response_message != ""
assert response_message != "No files selected for summarization. Please add files using the section on the left." assert response_message != "No files selected for summarization. Please add files using the section on the left."
assert response_message != "Only one file can be selected for summarization."
@pytest.mark.django_db(transaction=True) @pytest.mark.django_db(transaction=True)
@ -404,7 +402,7 @@ def test_summarize_multiple_files(client_offline_chat, default_user2: KhojUser):
response_message = response.json()["response"] response_message = response.json()["response"]
# Assert # Assert
assert response_message == "Only one file can be selected for summarization." assert response_message is not None
@pytest.mark.django_db(transaction=True) @pytest.mark.django_db(transaction=True)
@ -460,7 +458,6 @@ def test_summarize_different_conversation(client_offline_chat, default_user2: Kh
# Assert # Assert
assert response_message != "" assert response_message != ""
assert response_message != "No files selected for summarization. Please add files using the section on the left." assert response_message != "No files selected for summarization. Please add files using the section on the left."
assert response_message != "Only one file can be selected for summarization."
@pytest.mark.django_db(transaction=True) @pytest.mark.django_db(transaction=True)

5
tests/test_openai_chat_director.py
View file

@ -312,7 +312,6 @@ def test_summarize_one_file(chat_client, default_user2: KhojUser):
# Assert # Assert
assert response_message != "" assert response_message != ""
assert response_message != "No files selected for summarization. Please add files using the section on the left." assert response_message != "No files selected for summarization. Please add files using the section on the left."
assert response_message != "Only one file can be selected for summarization."
@pytest.mark.django_db(transaction=True) @pytest.mark.django_db(transaction=True)
@ -344,7 +343,6 @@ def test_summarize_extra_text(chat_client, default_user2: KhojUser):
# Assert # Assert
assert response_message != "" assert response_message != ""
assert response_message != "No files selected for summarization. Please add files using the section on the left." assert response_message != "No files selected for summarization. Please add files using the section on the left."
assert response_message != "Only one file can be selected for summarization."
@pytest.mark.django_db(transaction=True) @pytest.mark.django_db(transaction=True)
@ -371,7 +369,7 @@ def test_summarize_multiple_files(chat_client, default_user2: KhojUser):
response_message = response.json()["response"] response_message = response.json()["response"]
# Assert # Assert
assert response_message == "Only one file can be selected for summarization." assert response_message is not None
@pytest.mark.django_db(transaction=True) @pytest.mark.django_db(transaction=True)
@ -435,7 +433,6 @@ def test_summarize_different_conversation(chat_client, default_user2: KhojUser):
assert ( assert (
response_message_conv1 != "No files selected for summarization. Please add files using the section on the left." response_message_conv1 != "No files selected for summarization. Please add files using the section on the left."
) )
assert response_message_conv1 != "Only one file can be selected for summarization."
@pytest.mark.django_db(transaction=True) @pytest.mark.django_db(transaction=True)