mirror of
https://github.com/JimLiu/baoyu-skills.git
synced 2026-07-12 05:51:44 +08:00
Jim Liu 宝玉
774ad784d8
Move test files from tests/ directory to colocate with source code, convert from .mjs to .ts using tsx runner, add workspaces and npm cache to CI workflow.
102 lines
2.4 KiB
TypeScript
102 lines
2.4 KiB
TypeScript
import assert from "node:assert/strict";
|
|
import test from "node:test";
|
|
|
|
import type { CliArgs } from "../types.ts";
|
|
import {
|
|
buildInput,
|
|
extractOutputUrl,
|
|
parseModelId,
|
|
} from "./replicate.ts";
|
|
|
|
function makeArgs(overrides: Partial<CliArgs> = {}): CliArgs {
|
|
return {
|
|
prompt: null,
|
|
promptFiles: [],
|
|
imagePath: null,
|
|
provider: null,
|
|
model: null,
|
|
aspectRatio: null,
|
|
size: null,
|
|
quality: null,
|
|
imageSize: null,
|
|
referenceImages: [],
|
|
n: 1,
|
|
batchFile: null,
|
|
jobs: null,
|
|
json: false,
|
|
help: false,
|
|
...overrides,
|
|
};
|
|
}
|
|
|
|
test("Replicate model parsing accepts official formats and rejects malformed ones", () => {
|
|
assert.deepEqual(parseModelId("google/nano-banana-pro"), {
|
|
owner: "google",
|
|
name: "nano-banana-pro",
|
|
version: null,
|
|
});
|
|
assert.deepEqual(parseModelId("owner/model:abc123"), {
|
|
owner: "owner",
|
|
name: "model",
|
|
version: "abc123",
|
|
});
|
|
|
|
assert.throws(
|
|
() => parseModelId("just-a-model-name"),
|
|
/Invalid Replicate model format/,
|
|
);
|
|
});
|
|
|
|
test("Replicate input builder maps aspect ratio, image count, quality, and refs", () => {
|
|
assert.deepEqual(
|
|
buildInput(
|
|
"A robot painter",
|
|
makeArgs({
|
|
aspectRatio: "16:9",
|
|
quality: "2k",
|
|
n: 3,
|
|
}),
|
|
["data:image/png;base64,AAAA"],
|
|
),
|
|
{
|
|
prompt: "A robot painter",
|
|
aspect_ratio: "16:9",
|
|
number_of_images: 3,
|
|
resolution: "2K",
|
|
output_format: "png",
|
|
image_input: ["data:image/png;base64,AAAA"],
|
|
},
|
|
);
|
|
|
|
assert.deepEqual(
|
|
buildInput("A robot painter", makeArgs({ quality: "normal" }), ["ref"]),
|
|
{
|
|
prompt: "A robot painter",
|
|
aspect_ratio: "match_input_image",
|
|
resolution: "1K",
|
|
output_format: "png",
|
|
image_input: ["ref"],
|
|
},
|
|
);
|
|
});
|
|
|
|
test("Replicate output extraction supports string, array, and object URLs", () => {
|
|
assert.equal(
|
|
extractOutputUrl({ output: "https://example.com/a.png" } as never),
|
|
"https://example.com/a.png",
|
|
);
|
|
assert.equal(
|
|
extractOutputUrl({ output: ["https://example.com/b.png"] } as never),
|
|
"https://example.com/b.png",
|
|
);
|
|
assert.equal(
|
|
extractOutputUrl({ output: { url: "https://example.com/c.png" } } as never),
|
|
"https://example.com/c.png",
|
|
);
|
|
|
|
assert.throws(
|
|
() => extractOutputUrl({ output: { invalid: true } } as never),
|
|
/Unexpected Replicate output format/,
|
|
);
|
|
});
|