File size: 599 Bytes
f78888d b634622 f78888d b634622 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import pytest
import os
from PIL import Image
from local_model import query_local
from remote_model import query_remote, client
def test_local_model():
img = Image.new('RGB', (100, 100), color='red')
result = query_local(img, "test!")
assert isinstance(result, str)
assert len(result) > 0
@pytest.mark.skipif(os.getenv("HF_TOKEN") is None, reason="skipping remote model test no HF_TOKEN")
def test_remote_model():
img = Image.new('RGB', (100, 100), color='blue')
result = query_remote(img, "test?", client)
assert isinstance(result, str)
assert len(result) > 0
|