15 lines
571 B
TypeScript
15 lines
571 B
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { inferParamBFromIdOrName } from "./model-param-b.js";
|
|
|
|
describe("shared/model-param-b", () => {
|
|
it("extracts the largest valid b-sized parameter token", () => {
|
|
expect(inferParamBFromIdOrName("llama-8b mixtral-22b")).toBe(22);
|
|
expect(inferParamBFromIdOrName("Qwen 0.5B Instruct")).toBe(0.5);
|
|
});
|
|
|
|
it("ignores malformed, zero, and non-delimited matches", () => {
|
|
expect(inferParamBFromIdOrName("abc70beta 0b x70b2")).toBeNull();
|
|
expect(inferParamBFromIdOrName("model 0b")).toBeNull();
|
|
});
|
|
});
|