董宏凯andCursor
7c15ca6710
Disable HTML labels so nodes use native SVG text, and rewrite bare edge labels plus invalid class statements before render. Co-authored-by: Cursor <cursoragent@cursor.com>
72 lines
2.3 KiB
TypeScript
72 lines
2.3 KiB
TypeScript
import assert from "node:assert/strict";
|
|
import test from "node:test";
|
|
|
|
import { sanitizeMermaidSource } from "./mermaid";
|
|
|
|
test("inserts a missing arrow before a bare edge label", () => {
|
|
const input = "E --> G|价值输出 V_s| H[PPO 损失]";
|
|
assert.equal(
|
|
sanitizeMermaidSource(input),
|
|
"E --> G -->|价值输出 V_s| H[PPO 损失]",
|
|
);
|
|
});
|
|
|
|
test("leaves already-valid labeled edges unchanged", () => {
|
|
const input = "E -->|价值输出 V_s| H[PPO 损失]\nG -->|状态| I";
|
|
assert.equal(sanitizeMermaidSource(input), input);
|
|
});
|
|
|
|
test("fixes a label attached to a shaped node", () => {
|
|
const input = "E[Actor 网络]|价值输出 V_s| H[PPO 损失]";
|
|
assert.equal(
|
|
sanitizeMermaidSource(input),
|
|
"E[Actor 网络] -->|价值输出 V_s| H[PPO 损失]",
|
|
);
|
|
});
|
|
|
|
test("fixes several bare labels on one line", () => {
|
|
const input = "A|x| B|y| C";
|
|
assert.equal(sanitizeMermaidSource(input), "A -->|x| B -->|y| C");
|
|
});
|
|
|
|
test("inserts a space after class when the model starts the id list with a comma", () => {
|
|
const input = " class,E,F,G,H,I,J,K loss;";
|
|
assert.equal(
|
|
sanitizeMermaidSource(input),
|
|
" class E,F,G,H,I,J,K loss;",
|
|
);
|
|
});
|
|
|
|
test("splits zip-assigned class names onto one statement per class", () => {
|
|
const input = " class A,B,C,D input,module,module,module;";
|
|
assert.equal(
|
|
sanitizeMermaidSource(input),
|
|
" class A input;\n class B,C,D module;",
|
|
);
|
|
});
|
|
|
|
test("leaves a valid class assignment and classDef alone", () => {
|
|
const input = [
|
|
" classDef input fill:#e3f2fd,stroke:#1565c0;",
|
|
" class A,B input;",
|
|
].join("\n");
|
|
assert.equal(sanitizeMermaidSource(input), input);
|
|
});
|
|
|
|
test("repairs the DAC-PPO architecture diagram class block", () => {
|
|
const input = `graph TD
|
|
A[智能体观测 o_t] --> B[个体 Encoder]
|
|
E --> G -->|价值输出 V_s| H[PPO 损失]
|
|
classDef input fill:#e3f2fd,stroke:#1565c0;
|
|
classDef module fill:#fff3e0,stroke:#e65100;
|
|
classDef loss fill:#e8f5e9,stroke:#2e7d32;
|
|
class A,B,C,D input,module,module,module;
|
|
class,E,F,G,H,I,J,K loss;`;
|
|
const output = sanitizeMermaidSource(input);
|
|
assert.match(output, /class A input;/);
|
|
assert.match(output, /class B,C,D module;/);
|
|
assert.match(output, /class E,F,G,H,I,J,K loss;/);
|
|
assert.doesNotMatch(output, /class,/);
|
|
assert.doesNotMatch(output, /input,module/);
|
|
});
|