ai-chatbot/tests/routes/ordinal-image-references.test.ts

23 lines
877 B
TypeScript

import { expect, test } from "@playwright/test";
import { collectOrdinalImageIndexes } from "@/lib/ai/context-memory/ordinal-image-references";
test.describe("ordinal image references", () => {
test("keeps both sides of a Chinese replacement relationship", () => {
expect(
collectOrdinalImageIndexes("把图二替换到图一的场景中,靠着左边放"),
).toEqual([0, 1]);
expect(collectOrdinalImageIndexes("用图三替换图一的产品")).toEqual([
0, 2,
]);
});
test("still resolves a single compact ordinal", () => {
expect(collectOrdinalImageIndexes("只调整图二")).toEqual([1]);
expect(collectOrdinalImageIndexes("修改第三张图")).toEqual([2]);
});
test("does not treat dimensional wording as an image ordinal", () => {
expect(collectOrdinalImageIndexes("保持图二维布局")).toEqual([]);
});
});