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_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()) if __name__ == "__main__": unittest.main()