From e7ba29569d526ca0a00e1e85086708d17bfb5bce Mon Sep 17 00:00:00 2001
From: wyzula-jan <jan.wyzula@psi.ch>
Date: Fri, 3 Oct 2025 12:19:15 +0200
Subject: [PATCH] test(color_utils): cleanup for pyqtgraph

---
 tests/unit_tests/test_color_utils.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/tests/unit_tests/test_color_utils.py b/tests/unit_tests/test_color_utils.py
index 7628e8ae..ac2ef246 100644
--- a/tests/unit_tests/test_color_utils.py
+++ b/tests/unit_tests/test_color_utils.py
@@ -144,6 +144,19 @@ class ExamplePlotWidget(BECWidget, QWidget):
         self.glw.addItem(self.pi)
         self.pi.plot([1, 2, 3, 4, 5], pen="r")
 
+    def cleanup_pyqtgraph(self, item: pg.PlotItem | None = None):
+        """Cleanup pyqtgraph items."""
+        if item is None:
+            item = self.pi
+        item.vb.menu.close()
+        item.vb.menu.deleteLater()
+        item.ctrlMenu.close()
+        item.ctrlMenu.deleteLater()
+
+    def cleanup(self):
+        self.cleanup_pyqtgraph()
+        super().cleanup()
+
 
 def test_apply_theme(qtbot, mocked_client):
     widget = create_widget(qtbot, ExamplePlotWidget, client=mocked_client)