Spaces:
Running on Zero
Running on Zero
| import sys | |
| import unittest | |
| from pathlib import Path | |
| ROOT = Path(__file__).resolve().parent.parent | |
| sys.path.insert(0, str(ROOT)) | |
| from export_profiles import ( # noqa: E402 | |
| balanced_export_settings, | |
| faces_for_texture_size, | |
| simplify_ratio_for_texture_size, | |
| texture_steps_for_texture_size, | |
| ) | |
| class TextureQualityBudget(unittest.TestCase): | |
| def test_face_budget_scales_with_texture_area_and_caps_at_200k(self): | |
| self.assertEqual( | |
| [faces_for_texture_size(size) for size in (512, 1024, 1536, 2048, 4096)], | |
| [6250, 25000, 56250, 100000, 200000], | |
| ) | |
| def test_optimization_steps_scale_with_texture_size_and_cap_at_1000(self): | |
| self.assertEqual( | |
| [texture_steps_for_texture_size(size) for size in (512, 1024, 1536, 2048, 4096)], | |
| [250, 500, 750, 1000, 1000], | |
| ) | |
| def test_simplify_ratio_reaches_the_budget_without_over_decimating(self): | |
| self.assertAlmostEqual(simplify_ratio_for_texture_size(2048, 400000), 0.75) | |
| self.assertAlmostEqual(simplify_ratio_for_texture_size(4096, 400000), 0.5) | |
| self.assertEqual(simplify_ratio_for_texture_size(2048, 50000), 0.0) | |
| self.assertEqual(simplify_ratio_for_texture_size(512, 10000000), 0.999) | |
| def test_balanced_settings_carry_the_derived_steps(self): | |
| settings = balanced_export_settings(0.5, 4096) | |
| self.assertEqual(settings["texture_optimization_steps"], 1000) | |
| self.assertEqual(settings["texture_init"], "zeros") | |
| self.assertEqual(settings["texture_lr"], 1e-2) | |
| self.assertEqual(settings["texture_size"], 4096) | |
| self.assertEqual(settings["simplify"], 0.5) | |
| self.assertEqual(settings["uv_backend"], "cumesh") | |
| self.assertEqual(settings["render_views"], 100) | |
| self.assertEqual(settings["decimate_backend"], "cumesh") | |
| if __name__ == "__main__": | |
| unittest.main() | |