from seedream_mcp.tools.core.schemas import (
ImageToImageInput,
MultiImageFusionInput,
SequentialGenerationInput,
TextToImageInput,
)
def test_text_to_image_parameter_order() -> None:
assert list(TextToImageInput.model_fields.keys()) == [
"prompt",
"optimize_prompt_options",
"size",
"watermark",
"response_format",
"stream",
"request_count",
"parallelism",
"auto_save",
"save_path",
"custom_name",
]
def test_image_to_image_parameter_order() -> None:
assert list(ImageToImageInput.model_fields.keys()) == [
"prompt",
"optimize_prompt_options",
"image",
"size",
"watermark",
"response_format",
"stream",
"request_count",
"parallelism",
"auto_save",
"save_path",
"custom_name",
]
def test_multi_image_fusion_parameter_order() -> None:
assert list(MultiImageFusionInput.model_fields.keys()) == [
"prompt",
"optimize_prompt_options",
"image",
"size",
"watermark",
"response_format",
"stream",
"request_count",
"parallelism",
"auto_save",
"save_path",
"custom_name",
]
def test_sequential_generation_parameter_order() -> None:
assert list(SequentialGenerationInput.model_fields.keys()) == [
"prompt",
"optimize_prompt_options",
"image",
"size",
"watermark",
"max_images",
"response_format",
"stream",
"request_count",
"parallelism",
"auto_save",
"save_path",
"custom_name",
]