import pytest

from omniff.models.image_edit import ImageEditModel


def test_image_edit_interface():
    model = ImageEditModel(model_id="stabilityai/sdxl-turbo", device="cpu")
    assert not model.is_loaded


def test_image_edit_infer_not_loaded():
    model = ImageEditModel(model_id="stabilityai/sdxl-turbo", device="cpu")
    with pytest.raises(RuntimeError, match="not loaded"):
        model.infer({"image_path": "test.jpg", "prompt": "make it blue"})