/ 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()