1import { describe, expect, it } from "vite-plus/test";
2
3import { modelShortName } from "./modelShortName.ts";
4
5describe("modelShortName", () => {
6 it.each([
7 ["Claude Opus 5.5", ["Claude "], "Opus 5.5"],
8 ["GPT-6.1-Sol", ["GPT-"], "6.1 Sol"],
9 ["GPT 6.0 Astra", ["GPT-", "GPT "], "6.0 Astra"],
10 ["Grok 4 Pro", [], undefined],
11 ["Claude Opus 5.5", [], "Opus 5.5"],
12 ["NVIDIA: Nemotron 3 Super", [], "Nemotron 3 Super"],
13 ["GPT-5.5", ["GPT-"], undefined],
14 ["Gemini 3.1 Pro", ["Gemini "], "3.1 Pro"],
15 ] as const)("gives %s on families %j the short name %s", (name, families, expected) => {
16 expect(modelShortName(name, families)).toBe(expected);
17 });
18});