File tree Expand file tree Collapse file tree 1 file changed +17
-21
lines changed
Expand file tree Collapse file tree 1 file changed +17
-21
lines changed Original file line number Diff line number Diff line change @@ -94,28 +94,24 @@ def test_command_keys_photo(self):
9494 def test_command_keys_gimbal (self ):
9595 """Test gimbal control command keys."""
9696 from drone_base .config .drone import GimbalType
97+ angle_keys = {
98+ "z" : 0 ,
99+ "x" : - 15 ,
100+ "c" : - 30 ,
101+ "v" : - 45 ,
102+ "s" : - 60 ,
103+ "d" : - 75 ,
104+ "f" : - 90 ,
105+ }
106+
107+ for key , angle in angle_keys .items ():
108+ self .controller ._on_press_button (KeyCode .from_char (key ))
109+ self .mock_drone_commander .tilt_camera .assert_called_with (
110+ pitch_deg = angle ,
111+ reference_type = GimbalType .REF_ABSOLUTE
112+ )
97113
98- self .controller ._on_press_button (KeyCode .from_char ('z' ))
99- self .mock_drone_commander .tilt_camera .assert_called_with (
100- pitch_deg = 0 ,
101- reference_type = GimbalType .REF_ABSOLUTE
102- )
103-
104- self .mock_drone_commander .reset_mock ()
105-
106- self .controller ._on_press_button (KeyCode .from_char ('x' ))
107- self .mock_drone_commander .tilt_camera .assert_called_with (
108- pitch_deg = - 30 ,
109- reference_type = GimbalType .REF_ABSOLUTE
110- )
111-
112- self .mock_drone_commander .reset_mock ()
113-
114- self .controller ._on_press_button (KeyCode .from_char ('c' ))
115- self .mock_drone_commander .tilt_camera .assert_called_with (
116- pitch_deg = - 90 ,
117- reference_type = GimbalType .REF_ABSOLUTE
118- )
114+ self .mock_drone_commander .reset_mock ()
119115
120116 def test_emergency_stop (self ):
121117 """Test emergency stop functionality."""
You can’t perform that action at this time.
0 commit comments