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([]);
  });
});