trying fix before sleep

Author: prathameshnium

Keithley_6517B/High_Resistance/Backends/IV_K6517B_Simple_Backend_v10.py

@@ -49,7 +49,7 @@ def get_sweep_parameters():
     delay = float(input("Enter Settling Delay between points (s): "))
     filename = input(
         "Enter the filename to save data (e.g., SampleA_IV.csv): ")
-
+    
     if not filename.lower().endswith('.csv'):
         filename += '.csv'
     return start_v, stop_v, steps, delay, filename

tests/test_fixes.py

@@ -3,7 +3,8 @@
 
 class TestFixes(unittest.TestCase):
 
-    @patch('builtins.input', side_effect=['10', '2', 'test_output'])
+    @patch('time.sleep', MagicMock())  # Add mock for time.sleep to isolate the test
+    @patch('builtins.input', side_effect=['10', '2', 'test_output']) 
     @patch('pymeasure.instruments.keithley.Keithley2400')
     @patch('matplotlib.pyplot.show')
     @patch('pandas.DataFrame.to_csv')

tests/test_gui_iv_k2400.py

@@ -35,8 +35,7 @@ def test_start_measurement_logic(self, MockFigure, MockBackend):
         mock_figure_instance = MockFigure.return_value
         mock_ax_vi = MagicMock()
         mock_ax_ri = MagicMock()
-        # The GUI script calls subplots(), which needs to return our two mock axes
-        mock_figure_instance.subplots.return_value = mock_ax_vi, mock_ax_ri
+        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_t_control_l350_backend.py

@@ -92,7 +92,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('builtins.input', side_effect=['10', '20', '5', '300'])
     @patch('Lakeshore_350_340.Backends.T_Control_L350_Simple_Backend_v10.Lakeshore350')
     @patch('matplotlib.pyplot.show')
     @patch('builtins.open', new_callable=mock_open)