File size: 4,366 Bytes
cf554cc | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 | # -*- coding: utf-8 -*-
"""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 # noqa: E402
from holoaeon import concept_visual_forge as forge_mod # noqa: E402
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)
|