11 lines
602 B
Python
11 lines
602 B
Python
from spriteforge.studio.models import GenerationRecipe, GenerationRequest
|
|
from spriteforge.studio.providers import create_provider, provider_names
|
|
|
|
|
|
def test_diagnostic_provider_exercises_batch_without_gpu():
|
|
request=GenerationRequest(project_id="p",asset_id="agent",shot_id="shot_1",count=3,
|
|
recipe=GenerationRecipe(provider="diagnostic",model="test",seed=9,prompt="x",width=96,height=80))
|
|
output=create_provider("diagnostic").generate(request,lambda _: b"")
|
|
assert len(output)==3 and all(blob.startswith(b"\x89PNG") for blob in output)
|
|
assert "diagnostic" in provider_names()
|