| |
| """V21 概念视觉锻造台测试：八步流水线 + --forge 集成。""" |
| from __future__ import annotations |
|
|
| import contextlib |
| import io |
| import json |
| import os |
| import subprocess |
| import sys |
| import unittest |
|
|
| sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) |
|
|
| from holoaeon.concept_anatomy import CONCEPT_MAP |
| from holoaeon import concept_visual_forge as forge_mod |
|
|
|
|
| class TestTextSearch(unittest.TestCase): |
| def test_hit_builtin(self): |
| s = forge_mod.text_search("经络") |
| self.assertTrue(s["hit"]) |
| self.assertEqual(len(s["components"]), 8) |
| self.assertEqual(s["dimension"], "mechanism") |
|
|
| def test_miss_offtable(self): |
| s = forge_mod.text_search("圆珠笔") |
| self.assertFalse(s["hit"]) |
| self.assertIn("圆珠笔", s["description"]) |
| self.assertGreaterEqual(len(s["related"]), 0) |
|
|
| def test_all_builtin_searchable(self): |
| for c in CONCEPT_MAP: |
| self.assertTrue(forge_mod.text_search(c)["hit"], c) |
|
|
|
|
| class TestForgePipeline(unittest.TestCase): |
| def test_image_desc_v1(self): |
| d = forge_mod.image_desc_v1(forge_mod.text_search("经络")) |
| self.assertIn("主体", d) |
|
|
| def test_system_prompt_gen(self): |
| p = forge_mod.system_prompt_gen("主体：测试", "工业蓝图", "爆炸分解", "冷灰蓝") |
| self.assertIn("工业蓝图", p) |
| self.assertIn("爆炸分解", p) |
| self.assertIn("冷灰蓝", p) |
|
|
| def test_matrix_modify(self): |
| mx = forge_mod.matrix_modify("主体：测试") |
| self.assertEqual(mx["variant_count"], len(forge_mod.FORGE_MATRIX["style"])) |
| self.assertEqual(len(mx["variants"]), mx["variant_count"]) |
| self.assertIn("工业蓝图", mx["variants"][0]) |
|
|
| def test_optimize_prompt(self): |
| op = forge_mod.optimize_prompt("主体：测试") |
| self.assertIn("负面", op) |
| self.assertIn("确定性", op) |
| self.assertIn("只画概念映射表内元素", op) |
|
|
| def test_stylize_three_tiers(self): |
| st = forge_mod.stylize("主体：测试") |
| self.assertEqual(set(st.keys()), {"blueprint", "art_direction", "artistic"}) |
| self.assertIn("工业蓝图制图", st["blueprint"]) |
| self.assertIn("美术设定集", st["art_direction"]) |
| self.assertIn("艺术创作", st["artistic"]) |
|
|
| def test_forge_full_pipeline(self): |
| r = forge_mod.forge("经络") |
| self.assertEqual(r["input"], "经络") |
| self.assertTrue(r["step2_text_search"]["hit"]) |
| self.assertEqual(r["step5_matrix"]["variant_count"], 6) |
| for k in ("blueprint", "art_direction", "artistic"): |
| self.assertIn(k, r["step7_9_stylized"]) |
|
|
| def test_forge_offtable(self): |
| r = forge_mod.forge("圆珠笔") |
| self.assertFalse(r["step2_text_search"]["hit"]) |
| self.assertIn("step6_optimized", r) |
|
|
|
|
| class TestForgeCli(unittest.TestCase): |
| @classmethod |
| def setUpClass(cls): |
| cls.repo = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) |
|
|
| def test_module_main_stdout(self): |
| buf = io.StringIO() |
| with contextlib.redirect_stdout(buf): |
| rc = forge_mod.main(["经络"]) |
| self.assertEqual(rc, 0) |
| d = json.loads(buf.getvalue()) |
| self.assertTrue(d["step2_text_search"]["hit"]) |
|
|
| def test_concept_anatomy_forge_integration(self): |
| r = subprocess.run( |
| [sys.executable, "-X", "utf8", "-m", "holoaeon.concept_anatomy", |
| "圆珠笔", "--forge"], |
| cwd=self.repo, capture_output=True, text=True, encoding="utf-8", |
| timeout=60) |
| self.assertEqual(r.returncode, 0) |
| d = json.loads(r.stdout) |
| self.assertIn("step7_9_stylized", d) |
| self.assertEqual(d["input"], "圆珠笔") |
|
|
| def test_forge_json_regression(self): |
| """V19/V20 输出路径不破：--forge 不污染 --explain 与 JSON 解剖。""" |
| r = subprocess.run( |
| [sys.executable, "-X", "utf8", "-m", "holoaeon.concept_anatomy", |
| "经络", "--explain"], |
| cwd=self.repo, capture_output=True, text=True, encoding="utf-8", |
| timeout=60) |
| self.assertEqual(r.returncode, 0) |
| self.assertIn("一句话", r.stdout) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main(verbosity=2) |
|
|
