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