Spaces:
Running
Running
File size: 1,363 Bytes
89b8989 |
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 pytest
from fastapi.testclient import TestClient
from backend.app.main import app
from unittest.mock import patch, MagicMock
@pytest.fixture
def test_app():
with TestClient(app) as test_client:
yield test_client
@pytest.fixture
def sample_image():
"""Create a sample in-memory image for testing"""
from PIL import Image
from io import BytesIO
# Create a simple red image
img = Image.new('RGB', (100, 100), color='red')
img_byte_arr = BytesIO()
img.save(img_byte_arr, format='JPEG')
img_byte_arr.seek(0)
# Return the image data as bytes
return img_byte_arr.getvalue()
@pytest.fixture
def mock_analyzer():
with patch('transformers.AutoImageProcessor') as mock_processor, \
patch('transformers.AutoModelForImageClassification') as mock_model, \
patch('anthropic.Anthropic') as mock_anthropic:
# Setup mock processor
mock_processor.from_pretrained.return_value = MagicMock()
# Setup mock model
mock_model.from_pretrained.return_value = MagicMock()
# Setup mock anthropic
mock_anthropic.return_value = MagicMock()
yield (
mock_processor.from_pretrained.return_value,
mock_model.from_pretrained.return_value,
mock_anthropic.return_value
)
|