nan
489003c2cc
- Introduced color_image.py for preparing standalone color images for tests. - Added hdr_image.py to create HDR textures for testing. - Implemented test_image_depth_panorama.py to evaluate panorama topology, UVs, and depth discontinuities. - Created test_shadeless.py to validate the opacity and HDR color endpoints of the Shadeless group. - Developed moge3_export.py for exporting dynamic MoGe-3 backbone and ONNX graphs. - Added material.py to inherit material slots onto generated geometry. - Built image_depth_panorama.py to construct a quad panorama with depth separation and sampling.
55 lines
2.7 KiB
Python
55 lines
2.7 KiB
Python
import tempfile
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
from unittest.mock import Mock, patch
|
|
|
|
from tests.support.blender import BlenderTestCase
|
|
|
|
|
|
class InputPreparationTest(BlenderTestCase):
|
|
def test_sequence_format_is_checked_before_copy(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
path = Path(directory) / "frame.0001.exr"
|
|
path.touch()
|
|
image = SimpleNamespace(source="SEQUENCE")
|
|
with (
|
|
patch.object(self.image_data, "image_path", return_value=path),
|
|
patch.object(self.image_data, "image_is_packed", return_value=False),
|
|
patch.object(self.image_data.tempfile, "mkdtemp") as create,
|
|
):
|
|
with self.assertRaisesRegex(RuntimeError, "Unsupported input format: .exr"):
|
|
self.image_data.prepare_image_input(image, None)
|
|
create.assert_not_called()
|
|
self.assertTrue(path.exists())
|
|
|
|
def test_packed_export_failure_removes_temporary_directory(self):
|
|
for failure in (OSError("disk full"), None):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
output = Path(directory) / "export"
|
|
output.mkdir()
|
|
image = SimpleNamespace(file_format="OPEN_EXR", save=Mock(side_effect=failure))
|
|
with patch.object(self.image_data.tempfile, "mkdtemp", return_value=str(output)):
|
|
with self.assertRaises(RuntimeError):
|
|
self.image_data.save_packed_image(image, None)
|
|
self.assertEqual(image.file_format, "OPEN_EXR")
|
|
self.assertFalse(output.exists())
|
|
|
|
def test_sequence_copy_failure_preserves_originals(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
source = root / "frame.0001.png"
|
|
source.write_bytes(b"source")
|
|
output = root / "prepared"
|
|
output.mkdir()
|
|
with (
|
|
patch.object(self.image_data, "image_path", return_value=source),
|
|
patch.object(self.image_data, "image_is_packed", return_value=False),
|
|
patch.object(self.image_data, "collect_sequence_frames", return_value=[source]),
|
|
patch.object(self.image_data.tempfile, "mkdtemp", return_value=str(output)),
|
|
patch.object(self.image_data.shutil, "copy2", side_effect=OSError("disk full")),
|
|
):
|
|
with self.assertRaises(OSError):
|
|
self.image_data.prepare_image_input(SimpleNamespace(source="SEQUENCE"), None)
|
|
self.assertFalse(output.exists())
|
|
self.assertEqual(source.read_bytes(), b"source")
|