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