Skip to content

Commit

Permalink
Add unit test for l shortcut
Browse files Browse the repository at this point in the history
Since I already had a test for `k`, added a similar test for `l`.
  • Loading branch information
GuiMacielPereira committed May 31, 2024
1 parent 0f6fd37 commit c0cdaaf
Showing 1 changed file with 17 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -724,7 +724,7 @@ def test_click_y_axes_tick_label_launches_y_axes_editor(self, mock_y_editor):

mock_y_editor.assert_called_once()

def test_keyboard_shortcuts_switch_axes_scale(self):
def test_keyboard_shortcut_switch_x_scale(self):
key_press_event = self._create_mock_key_press_event("k")
key_press_event.inaxes.get_xscale.return_value = "linear"
key_press_event.inaxes.get_yscale.return_value = "log"
Expand All @@ -740,6 +740,22 @@ def test_keyboard_shortcuts_switch_axes_scale(self):
key_press_event.inaxes.set_xlim.assert_called_once_with((0, 100))
key_press_event.inaxes.set_ylim.assert_called_once_with((5, 10))

def test_keyboard_shortcut_switch_y_scale(self):
key_press_event = self._create_mock_key_press_event("l")
key_press_event.inaxes.get_xscale.return_value = "linear"
key_press_event.inaxes.get_yscale.return_value = "log"
key_press_event.inaxes.get_xlim.return_value = (0, 100)
key_press_event.inaxes.get_ylim.return_value = (5, 10)
key_press_event.inaxes.get_lines.return_value = ["fake_line"]
fig_manager = MagicMock()
fig_manager.canvas = MagicMock()
interactor = FigureInteraction(fig_manager)
interactor.on_key_press(key_press_event)
key_press_event.inaxes.set_xscale.assert_called_once_with("linear")
key_press_event.inaxes.set_yscale.assert_called_once_with("linear")
key_press_event.inaxes.set_xlim.assert_called_once_with((0, 100))
key_press_event.inaxes.set_ylim.assert_called_once_with((5, 10))

# Private methods
def _create_mock_fig_manager_to_accept_right_click(self):
fig_manager = MagicMock()
Expand Down

0 comments on commit c0cdaaf

Please sign in to comment.