File size: 675 Bytes
113dbd0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from main import create_image_from_prompt, inpaint_image_from_prompt
def test_create_image_from_prompt_sync():
imagebytesresult = create_image_from_prompt("a test prompt")
assert imagebytesresult is not None
def test_inpaint_from_prompt_sync():
img_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo.png"
mask_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo_mask.png"
imagebytesresult = inpaint_image_from_prompt("a test prompt", img_url, mask_url, True)
assert imagebytesresult is not None
|