75 lines
4.1 KiB
TypeScript
75 lines
4.1 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
|
|
|
import {
|
|
buildImageResolutionOptions,
|
|
formatImageResolutionSize,
|
|
imageResolutionChoices,
|
|
imageResolutionOption,
|
|
imageSizeForResolution,
|
|
supportsImageResolutionPresets,
|
|
} from "../src/lib/image-resolution-tiers";
|
|
import { defaultImageCapabilityConfig } from "../src/lib/model-capabilities";
|
|
import { resolveImageRequestSize } from "../src/services/api/image-validation";
|
|
|
|
const sizes = [
|
|
"1024x1024", "1360x1024", "1024x1360", "1536x1024", "1024x1536", "1024x1280", "1280x1024", "2048x878", "1824x1024", "1024x1824",
|
|
"2048x2048", "2304x1728", "1728x2304", "2496x1664", "1664x2496", "1792x2240", "2240x1792", "3136x1344", "2752x1536", "1536x2752",
|
|
"2880x2880", "3264x2448", "2448x3264", "3504x2336", "2336x3504", "2560x3200", "3200x2560", "3808x1632", "3840x2160", "2160x3840",
|
|
];
|
|
|
|
describe("image resolution tiers", () => {
|
|
test("精确像素预设保持请求原值，比例协议发送比例", () => {
|
|
const profile = defaultImageCapabilityConfig("openai-image", "test");
|
|
profile.size = { parameter: "size", values: ["1920x1080", "3840x2160", "2160x3840", "1824x1024"], default: "1920x1080", allowCustom: false };
|
|
expect(imageResolutionChoices(profile.size.values)).toEqual(["1k", "2k", "4k"]);
|
|
expect(imageSizeForResolution(buildImageResolutionOptions(profile.size.values), "1k", "16:9")).toBe("1824x1024");
|
|
expect(resolveImageRequestSize(profile, undefined, "1920x1080")).toEqual({ parameter: "size", value: "1920x1080" });
|
|
profile.size = { parameter: "aspect_ratio", values: ["16:9"], default: "16:9", allowCustom: false };
|
|
expect(resolveImageRequestSize(profile, undefined, "16:9")).toEqual({ parameter: "aspect_ratio", value: "16:9" });
|
|
profile.size = { parameter: "size", values: [], default: "auto", allowCustom: true };
|
|
expect(() => resolveImageRequestSize(profile, undefined, "1920x1080")).toThrow("16 的倍数");
|
|
});
|
|
test("将 Xiaobaishu 的精确尺寸整理为 1K、2K、4K 各十种比例", () => {
|
|
const options = buildImageResolutionOptions(sizes);
|
|
|
|
expect(options.filter((item) => item.tier === "1k")).toHaveLength(10);
|
|
expect(options.filter((item) => item.tier === "2k")).toHaveLength(10);
|
|
expect(options.filter((item) => item.tier === "4k")).toHaveLength(10);
|
|
expect(imageResolutionOption(options, "1536x2752")).toMatchObject({ tier: "2k", ratio: "9:16" });
|
|
});
|
|
|
|
test("切换档位时按相同比例还原精确像素并显示比例与档位", () => {
|
|
const options = buildImageResolutionOptions(sizes);
|
|
|
|
expect(imageSizeForResolution(options, "4k", "9:16")).toBe("2160x3840");
|
|
expect(formatImageResolutionSize("1536x2752", options)).toBe("9:16 · 2K");
|
|
});
|
|
|
|
test("保留自动尺寸并在摘要中显示中文标签", () => {
|
|
const values = ["auto", ...sizes];
|
|
|
|
expect(imageResolutionChoices(values)).toEqual(["auto", "1k", "2k", "4k"]);
|
|
expect(buildImageResolutionOptions(values)).toHaveLength(30);
|
|
expect(formatImageResolutionSize("auto", [])).toBe("自动");
|
|
});
|
|
|
|
test("允许自定义尺寸时仍保留自动与分辨率预设", () => {
|
|
expect(supportsImageResolutionPresets({ parameter: "size", values: ["auto", ...sizes], allowCustom: true })).toBe(true);
|
|
});
|
|
|
|
test("默认图片能力为 OpenAI 兼容模型提供完整 1K、2K、4K 档位", () => {
|
|
const profile = defaultImageCapabilityConfig("openai-image", "gpt-image-2-4K");
|
|
|
|
expect(supportsImageResolutionPresets(profile.size)).toBe(true);
|
|
expect(imageResolutionChoices(profile.size.values)).toEqual(["auto", "1k", "2k", "4k"]);
|
|
expect(buildImageResolutionOptions(profile.size.values)).toHaveLength(30);
|
|
});
|
|
|
|
test("识别适合全景图的 2:1 自定义尺寸", () => {
|
|
const options = buildImageResolutionOptions(["1440x720", "2048x1024", "3840x1920"]);
|
|
|
|
expect(options.map((item) => item.tier)).toEqual(["1k", "2k", "4k"]);
|
|
expect(imageResolutionOption(options, "3840x1920")).toMatchObject({ tier: "4k", ratio: "2:1" });
|
|
});
|
|
});
|