From 08d956940e744987176556c728560ecf5dce1ffb Mon Sep 17 00:00:00 2001
From: wyzula-jan <133381102+wyzula-jan@users.noreply.github.com>
Date: Thu, 28 Sep 2023 17:06:07 +0200
Subject: [PATCH] test: test_eiger_plot.py optimised imports

---
 tests/test_eiger_plot.py | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/tests/test_eiger_plot.py b/tests/test_eiger_plot.py
index 95a9901c..bc76642a 100644
--- a/tests/test_eiger_plot.py
+++ b/tests/test_eiger_plot.py
@@ -1,10 +1,9 @@
-import pytest
-from unittest.mock import MagicMock, patch, Mock
+import json
+from unittest.mock import MagicMock, patch
 import numpy as np
 import pyqtgraph as pg
+import pytest
 import zmq
-import json
-from PyQt5.QtWidgets import QPushButton
 from bec_widgets.examples.eiger_plot.eiger_plot import EigerPlot
 
 
@@ -16,8 +15,6 @@ def eiger_plot_instance(qtbot):
     return widget
 
 
-# Tests for on_image_update method
-# class TestOnImageUpdate:
 @pytest.mark.parametrize(
     "fft_checked, rotation_index, transpose_checked, log_checked, expected_image",
     [