/ test_camera_generator.py
test_camera_generator.py
 1  """Test ThreeDCamera with generator_mode=True"""
 2  # Completely purge DreamTalk modules from cache before reimporting
 3  import sys
 4  modules_to_delete = [key for key in sys.modules.keys() if 'DreamTalk' in key]
 5  for mod in modules_to_delete:
 6      del sys.modules[mod]
 7  
 8  from DreamTalk.imports import *
 9  
10  class TestScene(ThreeDScene):
11      def __init__(self):
12          super().__init__(sketch_mode=True, generator_mode=True)
13  
14      def construct(self):
15          # Create a simple cube to have something to look at
16          cube = FoldableCube(color=BLUE)
17  
18          print(f"Camera type: {self.camera.obj.GetType()}")
19          print(f"Is Python Generator: {self.camera.obj.GetType() == 1023866}")
20          print(f"Camera obj name: {self.camera.obj.GetName()}")
21  
22          # Test camera orbiting
23          print("Testing camera orbit movement...")
24  
25  if __name__ == "__main__":
26      scene = TestScene()