stringadmin
82 lines
2.2 KiB
JavaScript
82 lines
2.2 KiB
JavaScript
const assert = require("node:assert/strict");
|
|
|
|
const {
|
|
buildDashscopeImageSuperResolveBody,
|
|
buildDashscopeVideoStyleTransformBody,
|
|
normalizeImageUpscaleFactor,
|
|
normalizeVideoStyleTransformOptions,
|
|
} = require("../src/aiUpscaleHelpers");
|
|
const { extractVideoUrl } = require("../src/aiTaskWorker");
|
|
|
|
function testImageSuperResolveBody() {
|
|
const body = buildDashscopeImageSuperResolveBody({
|
|
imageUrl: "https://example.com/input.png",
|
|
scale: "4x",
|
|
});
|
|
|
|
assert.equal(body.model, "wanx2.1-imageedit");
|
|
assert.deepEqual(body.input, {
|
|
function: "super_resolution",
|
|
prompt: "图像超分。",
|
|
base_image_url: "https://example.com/input.png",
|
|
});
|
|
assert.equal(body.parameters.upscale_factor, 4);
|
|
assert.equal(body.parameters.n, 1);
|
|
assert.equal(Object.hasOwn(body.parameters, "watermark"), false);
|
|
}
|
|
|
|
function testVideoStyleTransformBody() {
|
|
const body = buildDashscopeVideoStyleTransformBody({
|
|
videoUrl: "https://example.com/input.mp4",
|
|
style: 7,
|
|
videoFps: 99,
|
|
minLen: 540,
|
|
useSR: true,
|
|
animateEmotion: false,
|
|
});
|
|
|
|
assert.equal(body.model, "video-style-transform");
|
|
assert.deepEqual(body.input, {
|
|
video_url: "https://example.com/input.mp4",
|
|
});
|
|
assert.equal(body.parameters.style, 7);
|
|
assert.equal(body.parameters.video_fps, 25);
|
|
assert.equal(body.parameters.min_len, 540);
|
|
assert.equal(body.parameters.use_SR, true);
|
|
assert.equal(body.parameters.animate_emotion, false);
|
|
}
|
|
|
|
function testNormalizers() {
|
|
assert.equal(normalizeImageUpscaleFactor("4x"), 4);
|
|
assert.equal(normalizeImageUpscaleFactor("3"), 2);
|
|
assert.deepEqual(normalizeVideoStyleTransformOptions({ style: "9", videoFps: 8, minLen: 999 }), {
|
|
style: 0,
|
|
videoFps: 15,
|
|
minLen: 720,
|
|
useSR: true,
|
|
animateEmotion: true,
|
|
});
|
|
}
|
|
|
|
function testVideoStyleResultExtraction() {
|
|
assert.equal(
|
|
extractVideoUrl({
|
|
output: {
|
|
task_status: "SUCCEEDED",
|
|
output_video_url: "https://dashscope-result.example.com/result.mp4",
|
|
},
|
|
}),
|
|
"https://dashscope-result.example.com/result.mp4",
|
|
);
|
|
}
|
|
|
|
function main() {
|
|
testImageSuperResolveBody();
|
|
testVideoStyleTransformBody();
|
|
testNormalizers();
|
|
testVideoStyleResultExtraction();
|
|
console.log("ai upscale helper contract tests passed");
|
|
}
|
|
|
|
main();
|