| |
| """V21 图像渲染器测试：4 渲染器 + 加载协议 + --img CLI。""" |
| from __future__ import annotations |
|
|
| import json |
| import os |
| import subprocess |
| import sys |
| import tempfile |
| import unittest |
|
|
| sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) |
|
|
| from holoaeon.concept_anatomy import ConceptAnatomizer |
| from holoaeon import concept_anatomy_render as rmod |
|
|
|
|
| def _anatomy(): |
| return ConceptAnatomizer().anatomize("经络") |
|
|
|
|
| class TestRenderers(unittest.TestCase): |
| @classmethod |
| def setUpClass(cls): |
| cls.result = _anatomy() |
| cls.tokens = {**rmod.DEFAULT_TOKENS} |
|
|
| def test_extract(self): |
| d = rmod._extract(self.result) |
| self.assertEqual(d["concept"], "经络") |
| self.assertGreaterEqual(len(d["components"]), 6) |
|
|
| def test_explosion_svg(self): |
| svg = rmod.render_explosion(self.result, self.tokens) |
| self.assertTrue(svg.startswith("<svg")) |
| self.assertIn("概念爆炸图", svg) |
| self.assertIn("成品对照", svg) |
| self.assertIn("components:", svg) |
|
|
| def test_explosion_horizontal(self): |
| svg = rmod.render_explosion(self.result, {**self.tokens, "img_axis": "horizontal"}) |
| self.assertIn("axis: horizontal", svg) |
|
|
| def test_annotated_svg(self): |
| svg = rmod.render_annotated(self.result, self.tokens) |
| self.assertIn("横截面标注图", svg) |
| self.assertIn("放大细节", svg) |
|
|
| def test_legend_svg(self): |
| svg = rmod.render_legend(self.result, self.tokens) |
| self.assertIn("图例", svg) |
| self.assertIn("零件清单", svg) |
| self.assertIn("机制", svg) |
|
|
| def test_diagram_svg(self): |
| svg = rmod.render_diagram(self.result, self.tokens) |
| self.assertIn("概念结构图", svg) |
| self.assertIn("isometric", svg) |
|
|
| def test_palette_switch(self): |
| svg = rmod.render(self.result, {**self.tokens, "img_palette": "线稿黑白"}) |
| self.assertIn("#ffffff", svg) |
|
|
|
|
| class TestRenderProtocol(unittest.TestCase): |
| def setUp(self): |
| self.result = _anatomy() |
|
|
| def test_registry_populated(self): |
| for name in ("explosion", "annotated", "legend", "diagram"): |
| self.assertIn(name, rmod.RENDER_REGISTRY) |
|
|
| def test_load_render_fallback(self): |
| fn = rmod.load_render("不存在的风格") |
| self.assertIs(fn, rmod.render_diagram) |
|
|
| def test_register_custom(self): |
| rmod.register_render("custom_test", lambda a, t: "<svg>custom</svg>", "t") |
| fn = rmod.load_render("custom_test") |
| self.assertEqual(fn({}, {}), "<svg>custom</svg>") |
|
|
| def test_render_protocol_entry(self): |
| svg = rmod.render(self.result, {"img_style": "explosion"}) |
| self.assertIn("概念爆炸图", svg) |
|
|
|
|
| class TestRenderCli(unittest.TestCase): |
| @classmethod |
| def setUpClass(cls): |
| cls.repo = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) |
|
|
| def test_module_main_save(self): |
| with tempfile.TemporaryDirectory() as td: |
| out = os.path.join(td, "pen.svg") |
| r = subprocess.run( |
| [sys.executable, "-X", "utf8", "-m", "holoaeon.concept_anatomy", |
| "经络", "--img", "explosion", "--save", out], |
| cwd=self.repo, capture_output=True, text=True, encoding="utf-8", |
| timeout=60) |
| self.assertEqual(r.returncode, 0, r.stderr) |
| d = json.loads(r.stdout) |
| self.assertEqual(d["saved"], out) |
| self.assertEqual(d["style"], "explosion") |
| self.assertTrue(os.path.isfile(out)) |
| with open(out, encoding="utf-8") as f: |
| head = f.read(60) |
| self.assertTrue(head.startswith("<svg")) |
|
|
| def test_all_styles_cli(self): |
| for style in ("explosion", "annotated", "legend", "diagram"): |
| r = subprocess.run( |
| [sys.executable, "-X", "utf8", "-m", "holoaeon.concept_anatomy_render", |
| "经络", "--style", style], |
| cwd=self.repo, capture_output=True, text=True, encoding="utf-8", |
| timeout=60) |
| self.assertEqual(r.returncode, 0, r.stderr) |
| self.assertIn("<svg", r.stdout) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main(verbosity=2) |
|
|
