basic advanced example

tests/test_light_keys.py

@@ -0,0 +1,58 @@
+"""Validate light keys without an Unreal installation."""
+import importlib.util
+from pathlib import Path
+import sys
+import unittest
+from unittest.mock import MagicMock, patch
+
+
+class LightKeyTests(unittest.TestCase):
+    def setUp(self):
+        self.unreal = MagicMock()
+        self.unreal.FrameNumber.side_effect = lambda frame: frame
+        path = Path(__file__).resolve().parents[1] / "Content/Python/ue_workshop/sequence.py"
+        spec = importlib.util.spec_from_file_location("ue_workshop._light_key_test", path)
+        module = importlib.util.module_from_spec(spec)
+        with patch.dict(sys.modules, {"unreal": self.unreal}):
+            spec.loader.exec_module(module)
+        self.shot = module.Sequence.__new__(module.Sequence)
+        self.shot.frames = 48
+        self.shot._light_tracks = {}
+        self.shot.bind = MagicMock()
+        self.track = self.shot.bind.return_value.add_track.return_value
+        self.channels = [MagicMock() for _ in range(4)]
+        self.track.add_section.return_value.get_all_channels.return_value = self.channels
+        self.actor = MagicMock()
+        self.light = self.actor.get_component_by_class.return_value
+        self.light.get_path_name.return_value = "Light"
+
+    def test_color_keys_rgb_and_leaves_alpha_at_default(self):
+        self.shot.light_key(self.actor, 10, color=(1, 0.5, 0))
+        self.shot.bind.assert_called_once_with(self.light)
+        self.track.set_property_name_and_path.assert_called_once_with("LightColor", "LightColor")
+        self.assertEqual(self.channels[1].add_key.call_args.args, (10, 0.5))
+        self.channels[3].set_default.assert_called_once_with(1.0)
+        self.channels[3].add_key.assert_not_called()
+
+    def test_later_keys_reuse_the_track(self):
+        self.shot.light_key(self.actor, 0, intensity=0)
+        self.shot.light_key(self.actor, 47, intensity=5000)
+        self.track.add_section.assert_called_once()
+        self.track.set_property_name_and_path.assert_called_once_with("Intensity", "Intensity")
+        self.assertEqual(self.channels[0].add_key.call_count, 2)
+
+    def test_color_and_intensity_use_separate_tracks(self):
+        self.shot.light_key(self.actor, 0, color=(1, 1, 1), intensity=100)
+        self.assertEqual(self.track.add_section.call_count, 2)
+
+    def test_invalid_arguments_create_no_track(self):
+        for args in [dict(frame=48, color=(1, 1, 1)), dict(frame=0, color=(1, 1)),
+                     dict(frame=0, intensity=1, interpolation="smooth")]:
+            with self.assertRaises(ValueError):
+                self.shot.light_key(self.actor, **args)
+        self.shot.bind.assert_not_called()
+
+    def test_actor_without_light_is_rejected(self):
+        self.actor.get_component_by_class.return_value = None
+        with self.assertRaises(ValueError):
+            self.shot.light_key(self.actor, 0, intensity=1)