|
|
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 |
|
|
|
