DemonGatanjieuandClaude Opus 5.5
7b09909e5b
tests/ (Python backend tests, .mjs frontend rule tests) is no longer ignored. tools/run_tests.mjs runs the structure check, every tests/*.mjs and the Python tests with $PYTHON, the portable python_embeded or python. docs/guides/testing.md says how to run and write them; README links it. Developer notes no longer point at 18 test files that were removed earlier (AGENTS.md table, backend, frontend and update-guide notes). Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
60 lines
3.5 KiB
JavaScript
60 lines
3.5 KiB
JavaScript
import assert from 'node:assert/strict';
|
|
import { applyParameterChanges, loadParameterSets, parameterChanges } from '../web/modules/node_parameter_sets.js';
|
|
|
|
// Both sources, with the same values saved in a material and in a recipe parameter set.
|
|
const shared = [123, 'fixed', 30, 5.5, 'euler', 'karras', 1];
|
|
globalThis.fetch = async (url) => ({
|
|
ok: true,
|
|
json: async () => (String(url).includes('/materials/')
|
|
? { materials: [
|
|
{ name: 'Slow', timestamp: 10, blocks: [{ title: 'KSampler', widgets_values: shared }] },
|
|
{ name: 'Fast', timestamp: 30, blocks: [{ title: 'KSampler', widgets_values: [1, 'fixed', 8, 2, 'lcm', 'sgm_uniform', 1] }] },
|
|
] }
|
|
: { groups: [
|
|
{ recipe_filename: 'r.json', recipe_name: 'Portrait', notebooks: [{ name: 'Slow set', timestamp: 20, nodes: [{ title: 'KSampler', widgets_values: shared }] }] },
|
|
{ recipe_filename: 'unbound', notebooks: [{ name: 'Loose', timestamp: 99, nodes: [{ widgets_values: [9] }] }] },
|
|
{ recipe_filename: 'gone.json', recipe_name: 'gone.json', notebooks: [{ name: 'Old', timestamp: 98, nodes: [{ widgets_values: [8] }] }] },
|
|
] }),
|
|
});
|
|
|
|
{
|
|
const entries = await loadParameterSets('KSampler');
|
|
assert.deepEqual(entries.map(entry => entry.name), ['Fast', 'Slow']);
|
|
assert.deepEqual(entries[1].sources.map(source => [source.kind, source.label]), [['material', 'Slow'], ['recipe', 'Portrait']]);
|
|
assert.equal(entries[1].timestamp, 20);
|
|
}
|
|
|
|
// What would change: seeds, model files and missing choices stay.
|
|
const widget = (name, value, values) => ({ name, value, ...(values ? { options: { values } } : {}) });
|
|
function sampler() {
|
|
return {
|
|
id: 3, type: 'KSampler',
|
|
widgets: [widget('seed', 5), widget('control_after_generate', 'fixed', ['fixed', 'randomize']), widget('steps', 20),
|
|
widget('cfg', 7), widget('sampler_name', 'euler', ['euler', 'dpmpp_2m']), widget('scheduler', 'normal', ['normal', 'karras']), widget('denoise', 1)],
|
|
};
|
|
}
|
|
{
|
|
const node = sampler();
|
|
const changes = parameterChanges(node, [123, 'randomize', 30, 5.5, 'euler', 'beta57', 1]);
|
|
assert.deepEqual(changes.map(c => [c.label, c.from, c.to, c.keep]), [
|
|
['seed', 5, 123, 'seed'], ['control_after_generate', 'fixed', 'randomize', 'seed'], ['steps', 20, 30, ''], ['cfg', 7, 5.5, ''], ['scheduler', 'normal', 'beta57', 'missing'],
|
|
]);
|
|
const loader = { id: 4, type: 'CheckpointLoaderSimple', widgets: [widget('ckpt_name', 'a.safetensors', ['a.safetensors', 'b.safetensors'])] };
|
|
assert.equal(parameterChanges(loader, ['b.safetensors'])[0].keep, 'model');
|
|
assert.deepEqual(parameterChanges(node, ['not a number', 'fixed', 20]), []); // a value of another type is not this input
|
|
}
|
|
|
|
// Applying writes only the changes that are made, as one undoable step.
|
|
{
|
|
const node = sampler();
|
|
node.widgets_values = node.widgets.map(w => w.value);
|
|
const graph = { getNodeById: id => (id === 3 ? node : null), beforeChange() {}, afterChange() {}, change() {}, setDirtyCanvas() {} };
|
|
const app = { graph, canvas: { setDirty() {} } };
|
|
const result = applyParameterChanges(app, node, parameterChanges(node, [123, 'fixed', 30, 5.5, 'euler', 'beta57', 1]));
|
|
assert.deepEqual(node.widgets.map(w => w.value), [5, 'fixed', 30, 5.5, 'euler', 'normal', 1]);
|
|
result.undo();
|
|
assert.deepEqual(node.widgets.map(w => w.value), [5, 'fixed', 20, 7, 'euler', 'normal', 1]);
|
|
}
|
|
|
|
console.log('node parameter sets: merged sources, kept values and undoable apply OK');
|