Spaces:
Runtime error
Runtime error
File size: 1,595 Bytes
5f491f6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | import os
import pytest
from app import process_inputs, random_url, open_image_url
def test_api_key_handling():
openai_api_key = os.getenv("OPENAI_API_KEY")
huggingface_api_key = os.getenv("HUGGINGFACE_API_KEY")
assert openai_api_key is not None, "OPENAI_API_KEY is not set"
assert huggingface_api_key is not None, "HUGGINGFACE_API_KEY is not set"
@pytest.mark.asyncio
async def test_input_validation():
valid_image_url = "https://example.com/valid_image.jpg"
invalid_image_url = "invalid_url"
class_names = "cat, dog"
# Test with valid inputs
async for result in process_inputs(class_names, valid_image_url):
assert "results" in result or "Fetching image and running model" in result
# Test with invalid image URL
async for result in process_inputs(class_names, invalid_image_url):
assert "Invalid URL provided" in result
# Test with empty class names
async for result in process_inputs("", valid_image_url):
assert "Provide class names" in result
@pytest.mark.asyncio
async def test_error_handling():
invalid_image_url = "https://example.com/invalid_image.jpg"
class_names = "cat, dog"
# Test error handling in random_url
url = await random_url(None)
assert url is not None
# Test error handling in open_image_url
image = await open_image_url(invalid_image_url)
assert image is None
# Test error handling in process_inputs
async for result in process_inputs(class_names, invalid_image_url):
assert "Something went wrong" in result
|