import unittest
from PIL import Image
from paiton_image21.runtime import validate_request

class RequestLimits(unittest.TestCase):
    def test_image_edit_contract(self):
        image=Image.new("RGBA",(1024,1024))
        settings=validate_request("Make the teapot red",1024,1024,mode="edit",image=image)
        self.assertEqual(settings["num_inference_steps"],40)
        self.assertEqual(settings["true_cfg_scale"],1.0)
        self.assertTrue(settings["use_kv_cache"])
        with self.assertRaises(ValueError):
            validate_request("edit",2048,2048,mode="edit",image=image)

    def test_unqualified_requests_fail_before_gpu_work(self):
        invalid=[dict(prompt=""),dict(prompt="x"*513),dict(prompt="x",width=4096,height=4096),
                 dict(prompt="x",width=1024,height=2048),dict(prompt="x",steps=4),
                 dict(prompt="x",guidance=2.0),dict(prompt="x",seed=-1),dict(prompt="x",seed=True),
                 dict(prompt="x",mode="edit"),dict(prompt="x",mode="video"),dict(prompt=None)]
        for kwargs in invalid:
            with self.subTest(kwargs=kwargs),self.assertRaises(ValueError):
                validate_request(**kwargs)

if __name__=="__main__":
    unittest.main()