tests updated

Author: prathameshnium

tests/test_gui_iv_k2400.py

@@ -23,13 +23,19 @@ def tearDown(self):
         self.root.destroy()
 
     @patch('Keithley_2400.IV_K2400_GUI_v5.Keithley2400_IV_Backend')
-    def test_start_measurement_logic(self, MockBackend):
+    @patch('matplotlib.figure.Figure')
+    def test_start_measurement_logic(self, MockFigure, MockBackend):
         """
         Tests the core logic of the 'Start' button click.
         Verifies that parameters are read from the UI and passed to the backend correctly.
         """
         # --- Setup ---
         # Instantiate the GUI. This also creates all the tk widgets.
+        # --- FIX: Configure the mock for Figure.subplots ---
+        mock_figure_instance = MockFigure.return_value
+        mock_ax_vi = MagicMock()
+        mock_ax_ri = MagicMock()
+        mock_figure_instance.subplots.return_value = (mock_ax_vi, mock_ax_ri)
         app = MeasurementAppGUI(self.root)
 
         # Mock the backend instance that the GUI will create

tests/test_iv_k6517b_simple_backend.py

@@ -7,7 +7,7 @@
 class TestIVK6517BSimpleBackend(unittest.TestCase):
 
     @patch('builtins.input', side_effect=['-10', '10', '5', '0.1', 'test_iv_simple.csv'])
-    @patch('pymeasure.instruments.keithley.Keithley6517B')
+    @patch('Keithley_6517B.High_Resistance.Backends.IV_K6517B_Simple_Backend_v10.Keithley6517B')
     @patch('builtins.open', new_callable=mock_open)
     @patch('matplotlib.pyplot.show')
     def test_full_run(self, mock_show, mock_file, mock_keithley_class, mock_input):

tests/test_t_control_l350_backend.py

@@ -93,7 +93,7 @@ def test_get_user_parameters(self, mock_input):
     @patch('tkinter.Tk')
     @patch('tkinter.filedialog.asksaveasfilename', return_value='test.csv')
     @patch('builtins.input', side_effect=['10', '20', '5', '30'])
-    @patch('Lakeshore_350_340.Backends.T_Control_L350_Simple_Backend_v10.Lakeshore350') # This patch is correct as it targets the class in the script's namespace
+    @patch('Lakeshore_350_340.Backends.T_Control_L350_Simple_Backend_v10.Lakeshore350')
     @patch('matplotlib.pyplot.show')
     @patch('builtins.open', new_callable=mock_open)
     @patch('time.sleep', MagicMock())