from mflux import ModelConfig
from tests.image_generation.helpers.image_generation_in_context_test_helper import ImageGeneratorInContextTestHelper


class TestImageGeneratorInContext:
    OUTPUT_IMAGE_FILENAME = "output.png"

    def test_in_context_lora_identity(self):
        ImageGeneratorInContextTestHelper.assert_matches_reference_image(
            reference_image_path="ic_lora_reference_in_context_identity.png",
            output_image_path=TestImageGeneratorInContext.OUTPUT_IMAGE_FILENAME,
            model_config=ModelConfig.dev(),
            steps=25,
            seed=42,
            height=320,
            width=320,
            prompt="In this set of two images, a bold modern typeface with the brand name 'DEMA' is introduced and is shown on a company merchandise product photo; [IMAGE1] a simplistic black logo featuring a modern typeface with the brand name 'DEMA' on a bright light green/yellowish background; [IMAGE2] the design is printed on a green/yellowish hoodie as a company merchandise product photo with a plain white background.",
            image_path="ic_lora_reference_logo.png",
            lora_style="identity",
        )