179 lines
6.5 KiB
Python
179 lines
6.5 KiB
Python
import os
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
import qr_bridge as qb
|
|
|
|
|
|
class TestTrisToTargetQuads(unittest.TestCase):
|
|
def test_half(self):
|
|
self.assertEqual(qb.tris_to_target_quads(10000), 5000)
|
|
|
|
def test_odd_floors(self):
|
|
self.assertEqual(qb.tris_to_target_quads(3), 1)
|
|
|
|
def test_at_least_one(self):
|
|
self.assertEqual(qb.tris_to_target_quads(1), 1)
|
|
self.assertEqual(qb.tris_to_target_quads(0), 1)
|
|
|
|
|
|
class TestBuildSettings(unittest.TestCase):
|
|
def _s(self, **kw):
|
|
return qb.build_settings("/i.fbx", "/o.fbx", "/p.txt", 500, **kw)
|
|
|
|
def test_core_lines_present(self):
|
|
s = self._s()
|
|
self.assertIn('FileIn="/i.fbx"', s)
|
|
self.assertIn('FileOut="/o.fbx"', s)
|
|
self.assertIn('ProgressFile="/p.txt"', s)
|
|
self.assertIn("TargetQuadCount=500", s)
|
|
self.assertIn("HostApp=Blender", s)
|
|
self.assertTrue(s.endswith("\n"))
|
|
|
|
def test_hard_edges_toggle(self):
|
|
self.assertIn("AutoDetectHardEdges=1", self._s(hard_edges=True))
|
|
self.assertIn("AutoDetectHardEdges=0", self._s(hard_edges=False))
|
|
|
|
def test_exact_toggle(self):
|
|
self.assertIn("ExactQuadCount=0", self._s(exact=False))
|
|
self.assertIn("ExactQuadCount=1", self._s(exact=True))
|
|
|
|
|
|
class TestParseProgress(unittest.TestCase):
|
|
def test_success(self):
|
|
self.assertEqual(qb.parse_progress("2\n"), ("success", ""))
|
|
|
|
def test_error_with_text(self):
|
|
self.assertEqual(qb.parse_progress("-3\n重拓扑失败"), ("error", "重拓扑失败"))
|
|
|
|
def test_error_no_text(self):
|
|
self.assertEqual(qb.parse_progress("-3"), ("error", ""))
|
|
|
|
def test_running_fraction(self):
|
|
self.assertEqual(qb.parse_progress("0.5"), ("running", ""))
|
|
|
|
def test_running_positive_nonfinal(self):
|
|
# 正整数但非终态标记 2 -> running（引擎进度信号，非成功）
|
|
self.assertEqual(qb.parse_progress("1"), ("running", ""))
|
|
|
|
def test_success_float_form(self):
|
|
self.assertEqual(qb.parse_progress("2.0"), ("success", ""))
|
|
|
|
def test_running_empty(self):
|
|
self.assertEqual(qb.parse_progress(""), ("running", ""))
|
|
|
|
def test_running_garbage(self):
|
|
self.assertEqual(qb.parse_progress("abc"), ("running", ""))
|
|
|
|
|
|
class TestFindQrEngine(unittest.TestCase):
|
|
def setUp(self):
|
|
self._env = dict(os.environ)
|
|
|
|
def tearDown(self):
|
|
os.environ.clear()
|
|
os.environ.update(self._env)
|
|
|
|
def test_env_var_priority(self):
|
|
with tempfile.NamedTemporaryFile(suffix=".exe", delete=False) as f:
|
|
path = f.name
|
|
try:
|
|
os.environ["QR_ENGINE"] = path
|
|
self.assertEqual(qb.find_qr_engine(), path)
|
|
finally:
|
|
os.remove(path)
|
|
|
|
def test_missing_returns_none(self):
|
|
os.environ["QR_ENGINE"] = os.path.join(tempfile.gettempdir(), "nope_xremesh.exe")
|
|
os.environ["APPDATA"] = tempfile.mkdtemp()
|
|
self.assertIsNone(qb.find_qr_engine())
|
|
|
|
def test_invalid_override_does_not_fall_through(self):
|
|
# QR_ENGINE 权威：显式设了但无效，即便 APPDATA 下装了真引擎也返回 None
|
|
root = tempfile.mkdtemp()
|
|
os.environ["QR_ENGINE"] = os.path.join(root, "nope.exe")
|
|
os.environ["APPDATA"] = root
|
|
eng_dir = os.path.join(root, "Blender Foundation", "Blender", "5.0",
|
|
"scripts", "addons", "QuadRemesher", "Engine")
|
|
os.makedirs(eng_dir)
|
|
open(os.path.join(eng_dir, "xremesh.exe"), "w").close()
|
|
self.assertIsNone(qb.find_qr_engine())
|
|
|
|
def test_globs_addon_path_picks_last(self):
|
|
root = tempfile.mkdtemp()
|
|
os.environ.pop("QR_ENGINE", None)
|
|
os.environ["APPDATA"] = root
|
|
base = os.path.join(root, "Blender Foundation", "Blender")
|
|
for ver in ("4.2", "5.0"):
|
|
eng_dir = os.path.join(base, ver, "scripts", "addons",
|
|
"QuadRemesher", "Engine")
|
|
os.makedirs(eng_dir)
|
|
open(os.path.join(eng_dir, "xremesh.exe"), "w").close()
|
|
chosen = qb.find_qr_engine()
|
|
self.assertIsNotNone(chosen)
|
|
self.assertIn(os.path.join("5.0", "scripts"), chosen)
|
|
|
|
|
|
class TestTopologyAcceptable(unittest.TestCase):
|
|
def test_closed_source_clean_qr_ok(self):
|
|
# 封闭输入(0/0) + 水密 QR(0/0) -> 接受
|
|
self.assertTrue(qb.topology_acceptable(0, 0, 0, 0))
|
|
|
|
def test_closed_source_holey_qr_rejected(self):
|
|
# 封闭输入被 QR 打出开边 -> 拒绝
|
|
self.assertFalse(qb.topology_acceptable(0, 0, 8, 0))
|
|
|
|
def test_closed_source_nonmanifold_qr_rejected(self):
|
|
self.assertFalse(qb.topology_acceptable(0, 0, 0, 1))
|
|
|
|
def test_open_shell_source_same_ok(self):
|
|
# 输入本身是开放壳(10 开边)，QR 未新增 -> 接受
|
|
self.assertTrue(qb.topology_acceptable(10, 0, 10, 0))
|
|
|
|
def test_open_shell_source_worse_rejected(self):
|
|
# QR 比输入引入更多开边 -> 拒绝
|
|
self.assertFalse(qb.topology_acceptable(10, 0, 12, 0))
|
|
|
|
def test_qr_fewer_defects_ok(self):
|
|
# QR 输出反而更少缺陷（不劣于输入）-> 接受
|
|
self.assertTrue(qb.topology_acceptable(10, 2, 8, 1))
|
|
|
|
|
|
class TestTopologyTolerance(unittest.TestCase):
|
|
def test_tol_allows_small_increase(self):
|
|
# 封闭输入 + QR 多出 3 开边，容忍 12 内 -> 接受
|
|
self.assertTrue(qb.topology_acceptable(0, 0, 3, 0, tol=12))
|
|
|
|
def test_tol_rejects_large_increase(self):
|
|
self.assertFalse(qb.topology_acceptable(0, 0, 13, 0, tol=12))
|
|
|
|
def test_tol_applies_to_nonmanifold(self):
|
|
self.assertTrue(qb.topology_acceptable(0, 0, 0, 12, tol=12))
|
|
self.assertFalse(qb.topology_acceptable(0, 0, 0, 13, tol=12))
|
|
|
|
def test_default_tol_zero_strict(self):
|
|
# 默认 tol=0：现有严格语义不变
|
|
self.assertFalse(qb.topology_acceptable(0, 0, 1, 0))
|
|
self.assertTrue(qb.topology_acceptable(0, 0, 0, 0))
|
|
|
|
|
|
class TestPrecollapseTarget(unittest.TestCase):
|
|
def test_above_cap_returns_cap(self):
|
|
self.assertEqual(qb.precollapse_target(1000000, 50000), 50000)
|
|
|
|
def test_below_cap_returns_none(self):
|
|
self.assertIsNone(qb.precollapse_target(40000, 50000))
|
|
|
|
def test_at_cap_returns_none(self):
|
|
self.assertIsNone(qb.precollapse_target(50000, 50000))
|
|
|
|
def test_default_cap(self):
|
|
self.assertEqual(qb.precollapse_target(60000), qb.QR_INPUT_CAP)
|
|
self.assertIsNone(qb.precollapse_target(40000))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|