Add unit tests for result deletion from images and signals

Author: PierreRaybaut

datalab/tests/features/common/result_deletion_unit_test.py

@@ -0,0 +1,96 @@
+# Copyright (c) DataLab Platform Developers, BSD 3-Clause license, see LICENSE file.
+
+"""
+Result deletion unit test
+--------------------------
+
+Test the deletion of analysis results from objects.
+"""
+
+# pylint: disable=invalid-name  # Allows short reference names like x, y, ...
+# guitest: show
+
+from __future__ import annotations
+
+from sigima.objects import Gauss2DParam, create_image_from_param
+from sigima.tests.data import create_paracetamol_signal
+
+from datalab.adapters_metadata import GeometryAdapter, TableAdapter
+from datalab.config import Conf
+from datalab.env import execenv
+from datalab.objectmodel import get_uuid
+from datalab.tests import datalab_test_app_context
+
+
+def test_delete_results_image():
+    """Test deletion of analysis results from images"""
+    with datalab_test_app_context(console=False) as win:
+        execenv.print("Image result deletion test:")
+        panel = win.imagepanel
+
+        # Create a test image
+        param = Gauss2DParam.create(height=200, width=200, sigma=20)
+        img = create_image_from_param(param)
+        panel.add_object(img)
+
+        # Run centroid analysis to create results
+        execenv.print("  Running centroid analysis...")
+        with Conf.proc.show_result_dialog.temp(False):
+            panel.processor.run_feature("centroid")
+
+        # Verify that results exist
+        img_refreshed = panel.objmodel[get_uuid(img)]
+        adapter_before = GeometryAdapter.from_obj(img_refreshed, "centroid")
+        assert adapter_before is not None, (
+            "Centroid result should exist before deletion"
+        )
+        execenv.print("  ✓ Centroid result created")
+
+        # Delete all results
+        execenv.print("  Deleting all results...")
+        panel.objview.select_objects([get_uuid(img)])
+        panel.delete_results()
+
+        # Verify that results were deleted
+        img_after = panel.objmodel[get_uuid(img)]
+        adapter_after = GeometryAdapter.from_obj(img_after, "centroid")
+        assert adapter_after is None, "Centroid result should be deleted"
+        execenv.print("  ✓ Centroid result deleted")
+
+
+def test_delete_results_signal():
+    """Test deletion of analysis results from signals"""
+    with datalab_test_app_context(console=False) as win:
+        execenv.print("Signal result deletion test:")
+        panel = win.signalpanel
+
+        # Create a test signal
+        sig = create_paracetamol_signal()
+        panel.add_object(sig)
+
+        # Run stats analysis to create table results
+        execenv.print("  Running stats analysis...")
+        with Conf.proc.show_result_dialog.temp(False):
+            panel.processor.run_feature("stats")
+
+        # Verify that results exist
+        sig_refreshed = panel.objmodel[get_uuid(sig)]
+        tables_before = list(TableAdapter.iterate_from_obj(sig_refreshed))
+        assert len(tables_before) > 0, "Stats result should exist before deletion"
+        execenv.print("  ✓ Stats result created")
+
+        # Delete all results
+        execenv.print("  Deleting all results...")
+        panel.objview.select_objects([get_uuid(sig)])
+        panel.delete_results()
+
+        # Verify that results were deleted
+        sig_after = panel.objmodel[get_uuid(sig)]
+        tables_after = list(TableAdapter.iterate_from_obj(sig_after))
+        assert len(tables_after) == 0, "Stats result should be deleted"
+        execenv.print("  ✓ Stats result deleted")
+
+
+if __name__ == "__main__":
+    test_delete_results_image()
+    test_delete_results_signal()