test lock trimage tool

Author: vld-codra

plotpy/tests/unit/test_plot_image.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 #
 # Licensed under the terms of the BSD 3-Clause
 # (see plotpy/LICENSE for details)
@@ -14,6 +13,7 @@
 from plotpy.builder import make
 from plotpy.plot import BasePlot
 from plotpy.tools import ColormapTool, ReverseColormapTool
+from plotpy.tools.image import LockTrImageTool
 
 
 def compute_image():
@@ -104,6 +104,24 @@ def test_plot_tr_image():
             assert bounds3.height() == bounds2.height() * 3.0
 
 
+def test_lock_tr_image():
+    """Test LockTrImageTool"""
+    with qt_app_context(exec_loop=True):
+        win = make.dialog(type="image", toolbar=True)
+        win.show()
+        item = make.trimage(compute_image())
+        # with plot_image(item) as plot:
+        plot = win.manager.get_plot()
+        plot.add_item(item)
+        plot.select_item(item)
+        tool = win.manager.add_tool(LockTrImageTool)
+        tool.activate()
+        assert item.is_locked()
+        tool.activate(False)
+        assert not item.is_locked()
+        exec_dialog(win)
+
+
 @pytest.mark.parametrize("ratio", [1.0, 0.75, 1.5, 2.0, 3.0])
 def test_set_aspect_ratio(ratio):
     """Test BasePlot.set_aspect_ratio method()