Skip to content

Commit 654dc6e

Browse files
committed
Fixed test for keyboard controller to reflect angle to key mapping.
1 parent dc71c10 commit 654dc6e

File tree

1 file changed

+17
-21
lines changed

1 file changed

+17
-21
lines changed

tests/control/manual/test_keyboard_controller.py

Lines changed: 17 additions & 21 deletions
Original file line numberDiff line numberDiff 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."""

0 commit comments

Comments
 (0)