wehub-resource-sync
36 lines
1.4 KiB
TypeScript
36 lines
1.4 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import { summarizeEncoderCandidates } from "../../../open-sse/services/compression/engines/headroom/encoderComparison.ts";
|
|
|
|
const byteLen = (s: string) => Buffer.byteLength(s, "utf8");
|
|
|
|
test("agrega sizes e elege winner por tokens", () => {
|
|
const messages = [
|
|
{
|
|
role: "user",
|
|
content: JSON.stringify(Array.from({ length: 20 }, (_, i) => ({ id: i, ok: true }))),
|
|
},
|
|
];
|
|
const cmp = summarizeEncoderCandidates(messages, 8, byteLen);
|
|
assert.equal(cmp.arraysCompared, 1);
|
|
assert.ok(cmp.json.bytes > 0 && cmp.gcf.bytes > 0);
|
|
assert.ok(["gcf", "toon", "json"].includes(cmp.winner));
|
|
const sizes: Record<string, number> = { gcf: cmp.gcf.tokens, json: cmp.json.tokens };
|
|
if (cmp.toonAvailable) sizes["toon"] = cmp.toon.tokens;
|
|
const min = Math.min(...Object.values(sizes));
|
|
assert.equal(sizes[cmp.winner], min);
|
|
});
|
|
|
|
test("sem array compactável → zerado, sem winner espúrio", () => {
|
|
const cmp = summarizeEncoderCandidates([{ role: "user", content: "oi" }], 8, byteLen);
|
|
assert.equal(cmp.arraysCompared, 0);
|
|
assert.equal(cmp.json.bytes, 0);
|
|
assert.equal(cmp.gcf.bytes, 0);
|
|
});
|
|
|
|
test("system messages são ignoradas (igual ao smartcrusher)", () => {
|
|
const arr = JSON.stringify(Array.from({ length: 20 }, (_, i) => ({ id: i })));
|
|
const cmp = summarizeEncoderCandidates([{ role: "system", content: arr }], 8, byteLen);
|
|
assert.equal(cmp.arraysCompared, 0);
|
|
});
|