klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-05 05:39:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Setup Search with Search_Config to Fix Tests

Browse Source 
- Rename pytest fixture search_config to more appropriate
content_config
- Create search_config pytest fixture
- Use search_config where search being setup, used in tests
This commit is contained in:
Debanjum Singh Solanky
2022-01-14 20:13:14 -05:00
parent c64e0c2965
commit ed144f7984
4 changed files with 80 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/test_image_search.py
View File

@@ -5,14 +5,15 @@ import pytest
from src.main import model
from src.search_type import image_search
from src.utils.helpers import resolve_absolute_path
from src.utils.rawconfig import ContentTypeConfig, SearchTypeConfig
# Test
# ----------------------------------------------------------------------------------------------------
def test_image_search_setup(search_config):
def test_image_search_setup(content_config: ContentTypeConfig, search_config: SearchTypeConfig):
# Act
# Regenerate image search embeddings during image setup
image_search_model = image_search.setup(search_config.image, regenerate=True)
image_search_model = image_search.setup(content_config.image, search_config.image, regenerate=True)
# Assert
assert len(image_search_model.image_names) == 3
@@ -21,9 +22,9 @@ def test_image_search_setup(search_config):
# ----------------------------------------------------------------------------------------------------
@pytest.mark.skip(reason="results inconsistent currently")
def test_image_search(search_config):
def test_image_search(content_config: ContentTypeConfig, search_config: SearchTypeConfig):
# Arrange
model.image_search = image_search.setup(search_config.image, regenerate=False)
model.image_search = image_search.setup(content_config.image, search_config.image, regenerate=False)
query_expected_image_pairs = [("brown kitten next to plant", "kitten_park.jpg"),
("horse and dog in a farm", "horse_dog.jpg"),
("A guinea pig eating grass", "guineapig_grass.jpg")]
@@ -38,11 +39,11 @@ def test_image_search(search_config):
results = image_search.collate_results(
hits,
model.image_search.image_names,
search_config.image.input_directory,
content_config.image.input_directory,
count=1)
actual_image = results[0]["Entry"]
expected_image = resolve_absolute_path(search_config.image.input_directory.joinpath(expected_image_name))
expected_image = resolve_absolute_path(content_config.image.input_directory.joinpath(expected_image_name))
# Assert
assert expected_image == actual_image