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