33 lines
980 B
Python
33 lines
980 B
Python
import asyncio
|
|
import os
|
|
|
|
from image_processor.icons_finder import get_icon
|
|
from image_processor.icons_vectorstore_utils import get_icons_vectorstore
|
|
from ppt_generator.models.content_type_models import IconQueryCollectionModel
|
|
from ppt_generator.models.query_and_prompt_models import IconQueryCollectionWithData
|
|
|
|
|
|
def test_get_icon():
|
|
vector_store = get_icons_vectorstore()
|
|
|
|
for query in [
|
|
"thermometer high",
|
|
"rainstorm",
|
|
"sea level icon",
|
|
"solar panel icon",
|
|
"efficient light bulb",
|
|
"sustainable leaf",
|
|
]:
|
|
asyncio.run(
|
|
get_icon(
|
|
vector_store,
|
|
IconQueryCollectionWithData(
|
|
icon_query=IconQueryCollectionModel(queries=[query]),
|
|
index=0,
|
|
),
|
|
os.path.join(
|
|
os.getenv("APP_DATA_DIRECTORY"), f"generated_icons/{query}.png"
|
|
),
|
|
)
|
|
)
|
|
break
|