new unit tests

Author: vld-codra

plotpy/tests/unit/test_point_tools.py

@@ -10,10 +10,15 @@
 from numpy import linspace, sin
 
 from plotpy.builder import make
+from plotpy.interfaces.items import ICurveItemType
 from plotpy.plot.plotwidget import PlotDialog, PlotWindow
 from plotpy.tests import vistools as ptv
-from plotpy.tools.base import InteractiveTool
-from plotpy.tools.curve import EditPointTool, SelectPointsTool, SelectPointTool
+from plotpy.tools import (
+    EditPointTool,
+    InteractiveTool,
+    SelectPointsTool,
+    SelectPointTool,
+)
 
 if TYPE_CHECKING:
     from plotpy.items.curve.base import CurveItem
@@ -22,6 +27,29 @@
 ToolT = TypeVar("ToolT", bound=InteractiveTool)
 
 
+def keyboard_event(
+    win: PlotDialog,
+    qapp: QW.QApplication,
+    key: QC.Qt.Key,
+    mod=QC.Qt.KeyboardModifier.NoModifier,
+):
+    """Simulates a keyboard event on the plot.
+
+    Args:
+        win: window containing the plot
+        qapp: Main QApplication instance
+        key: Key to simulate
+        mod: Keyboard modifier. Defaults to QC.Qt.KeyboardModifier.NoModifier.
+
+    Returns:
+        None
+    """
+    plot = win.manager.get_plot()
+    canva: QW.QWidget = plot.canvas()  # type: ignore
+    key_event = QG.QKeyEvent(QC.QEvent.Type.KeyPress, key, mod)
+    qapp.sendEvent(canva, key_event)
+
+
 def mouse_event_at_relative_plot_pos(
     win: PlotDialog,
     qapp: QW.QApplication,
@@ -161,6 +189,12 @@ def test_select_points_tool():
 def test_edit_point_tool():
     with qt_app_context(exec_loop=False) as qapp:
         win, tool = create_window(EditPointTool)
+        curve_item: CurveItem = win.manager.get_plot().get_last_active_item(ICurveItemType)  # type: ignore
+        orig_x, orig_y = curve_item.get_data()
+
+        assert orig_x is not None and orig_y is not None
+        orig_x, orig_y = orig_x.copy(), orig_y.copy()
+
         assert tool is not None
 
         n = 100
@@ -173,7 +207,6 @@ def test_edit_point_tool():
 
         drag_mouse(win, qapp, x_path, y_path)
 
-        curve_item: CurveItem = win.manager.get_plot().get_active_item()  # type: ignore
         curve_changes = tool.get_changes()[curve_item]
         for i, (x, y) in curve_changes.items():
             x_arr, y_arr = curve_item.get_data()
@@ -182,6 +215,24 @@ def test_edit_point_tool():
 
         tool.get_arrays()
         tool.get_initial_arrays()
+
+        keyboard_event(
+            win, qapp, QC.Qt.Key.Key_Z, QC.Qt.KeyboardModifier.ControlModifier
+        )
+
+        assert len(curve_changes) == 0
+
+        restored_x, restored_y = curve_item.get_data()
+        assert restored_x is not None and restored_y is not None
+        assert np.allclose(orig_x, restored_x)
+        assert np.allclose(orig_y, restored_y)
+
+        mouse_event_at_relative_plot_pos(win, qapp, (0.5, 0.5), CLICK)
+        tool.trigger_insert_point_at_selection()
+
+        new_x, new_y = curve_item.get_data()
+        assert len(new_x) == len(orig_x) + 1 and len(new_y) == len(orig_y) + 1
+
         exec_dialog(win)
 
 

plotpy/tests/unit/test_rect_action_tools.py

@@ -9,28 +9,24 @@
 
 from plotpy.tests import vistools as ptv
 from plotpy.tests.features.test_auto_curve_image import make_curve_image_legend
-from plotpy.tools.annotation import (
+from plotpy.tools import (
     AnnotatedCircleTool,
     AnnotatedEllipseTool,
     AnnotatedObliqueRectangleTool,
     AnnotatedPointTool,
     AnnotatedRectangleTool,
     AnnotatedSegmentTool,
-)
-from plotpy.tools.cross_section import (
     AverageCrossSectionTool,
-    CrossSectionTool,
-    ObliqueCrossSectionTool,
-)
-from plotpy.tools.image import ImageStatsTool
-from plotpy.tools.misc import SnapshotTool
-from plotpy.tools.shape import (
     CircleTool,
+    CrossSectionTool,
     EllipseTool,
+    ImageStatsTool,
+    ObliqueCrossSectionTool,
     ObliqueRectangleTool,
     PointTool,
     RectangleTool,
     SegmentTool,
+    SnapshotTool,
 )
 
 if TYPE_CHECKING:
@@ -122,8 +118,8 @@ def test_cross_section_tool():
     _test_annotation_tools((CrossSectionTool,))
 
 
-def test_oblique_cross_section_tool():
-    _test_annotation_tools((ObliqueCrossSectionTool,))
+# def test_oblique_cross_section_tool():
+#     _test_annotation_tools((ObliqueCrossSectionTool,))
 
 
 def test_snapshot_tool():

plotpy/tests/unit/test_rect_zoom.py

@@ -0,0 +1,109 @@
+# guitest: show
+
+from typing import Callable
+
+import numpy as np
+import qtpy.QtCore as QC
+import qtpy.QtGui as QG
+import qtpy.QtWidgets as QW
+from guidata.qthelpers import exec_dialog, qt_app_context
+from numpy import linspace
+
+from plotpy.plot.plotwidget import PlotDialog
+from plotpy.tests.unit.test_point_tools import (
+    CLICK,
+    create_window,
+    drag_mouse,
+    keyboard_event,
+    mouse_event_at_relative_plot_pos,
+)
+from plotpy.tools import FreeFormTool, MultiLineTool, RectZoomTool
+
+
+def zoom(
+    x_path: np.ndarray, y_path: np.ndarray, compare: Callable[[float, float], bool]
+):
+    with qt_app_context(exec_loop=False) as qapp:
+        win, tool = create_window(RectZoomTool)
+        plot = win.manager.get_plot()
+        initial_axis_limits = [
+            plot.get_axis_limits(axis) for axis in plot.get_active_axes()
+        ]
+
+        drag_mouse(win, qapp, x_path, y_path)
+
+        final_axis_limits = [
+            plot.get_axis_limits(axis) for axis in plot.get_active_axes()
+        ]
+
+        assert all(
+            map(
+                lambda og, final: compare(og[1], final[1]),
+                initial_axis_limits,
+                final_axis_limits,
+            )
+        )
+        exec_dialog(win)
+
+
+def test_rect_zoom_tool():
+    x_path = linspace(0, 0.5, 100)
+    y_path = linspace(0, 0.5, 100)
+    zoom(x_path, y_path, lambda og, final: final < og)
+
+
+def test_rect_unzoom_tool():
+    x_path = linspace(-0.5, 1.5, 100)
+    y_path = linspace(-0.5, 1.5, 100)
+    zoom(x_path, y_path, lambda og, final: final > og)
+
+
+def test_free_form_tool():
+    corners = np.array(((0.1, 0.1), (0.1, 0.8), (0.8, 0.8), (0.8, 0.1)))
+    with qt_app_context(exec_loop=False) as qapp:
+        win, tool = create_window(FreeFormTool)
+
+        # drag_mouse(win, qapp, x_path, y_path)
+        for x, y in corners:
+            mouse_event_at_relative_plot_pos(win, qapp, (x, y), CLICK)
+
+        assert tool.shape is not None
+
+        assert tool.shape.get_points().shape == corners.shape
+
+        exec_dialog(win)
+
+
+def test_multiline_tool():
+    n = 100
+    t = np.linspace(0, np.pi * 10, n)
+
+    # Create x and y arrays
+    x_arr = t * np.cos(t) / n + 0.5
+    y_arr = t * np.sin(t) / n + 0.5
+
+    with qt_app_context(exec_loop=False) as qapp:
+        win, tool = create_window(MultiLineTool)
+
+        # drag_mouse(win, qapp, x_path, y_path)
+        for x, y in zip(x_arr, y_arr):
+            mouse_event_at_relative_plot_pos(win, qapp, (x, y), CLICK)
+
+        assert tool.shape is not None
+        assert tool.shape.get_points().shape == np.array([x_arr, y_arr]).T.shape
+
+        # Delete last point
+        keyboard_event(win, qapp, QC.Qt.Key.Key_Backspace)
+
+        points_count, _ = tool.shape.get_points().shape
+
+        assert points_count == (n - 1)
+
+        exec_dialog(win)
+
+
+if __name__ == "__main__":
+    test_rect_zoom_tool()
+    test_rect_unzoom_tool()
+    test_free_form_tool()
+    test_multiline_tool()