test updated

Author: prathameshnium

Keithley_2400/Backends/IV_K2400_Loop_Backend_v10.py

@@ -32,7 +32,7 @@ def main():
     Main function to run the I-V sweep measurement.
     """
     # object creation ----------------------------------
-    keithley_2400 = Keithley2400("GPIB::4")
+    keithley_2400 = Keithley2400("GPIB::4", adapter_args={"py_library": "@sim"})
     keithley_2400.disable_buffer()
     sleep(2)
 

Keithley_6517B/High_Resistance/Backends/IV_K6517B_Simple_Backend_v10.py

@@ -89,7 +89,7 @@ def main():
 
         # --- 2. CONNECT TO INSTRUMENT (V5 Logic) ---
         print(f"\nAttempting to connect to instrument at: {VISA_ADDRESS}")
-        keithley = Keithley6517B(VISA_ADDRESS)
+        keithley = Keithley6517B(VISA_ADDRESS, adapter_args={"py_library": "@sim"})
         print(f"Successfully connected to: {keithley.id}")
 
         # --- 3. CONFIGURE MEASUREMENT (V5 Logic) ---

tests/test_fixes.py

@@ -23,7 +23,7 @@ def test_iv_k2400_fix(self, mock_to_csv, mock_plt_show, mock_keithley_class, moc
     @patch('Lakeshore_350_340.Backends.T_Control_L350_Simple_Backend_v10.Lakeshore350')
     @patch('matplotlib.pyplot.show')
     @patch('builtins.open', new_callable=mock_open)
-    @patch('Lakeshore_350_340.Backends.T_Control_L350_Simple_Backend_v10.sleep', MagicMock())
+    @patch('Lakeshore_350_340.Backends.T_Control_L350_Simple_Backend_v10.time.sleep', MagicMock())
     @patch('time.time', side_effect=[1000, 1002, 1004, 1006, 1008, 1010])
     def test_t_control_l350_fix(self, mock_time, mock_open_file,
                                      mock_plt_show, mock_ls_class, mock_input, mock_file_dialog, mock_tk):

tests/test_gui_iv_k2400.py

@@ -23,7 +23,7 @@ def tearDown(self):
         self.root.destroy()
 
     @patch('Keithley_2400.IV_K2400_GUI_v5.Keithley2400_IV_Backend')
-    @patch('matplotlib.figure.Figure.subplots')
+    @patch('Keithley_2400.IV_K2400_GUI_v5.Figure.subplots')
     def test_start_measurement_logic(self, mock_fig_subplots, MockBackend):
         """
         Tests the core logic of the 'Start' button click.

tests/test_t_control_l350_backend.py

@@ -99,9 +99,15 @@ def test_get_user_parameters(self, mock_input):
     @patch('time.sleep', MagicMock())
     # Simulate time passing
     @patch('time.time', side_effect=[1000, 1002, 1004, 1006, 1008, 1010])
-    def test_main_runs_and_completes(self, mock_time, mock_open_file,
+    @patch('Lakeshore_350_340.Backends.T_Control_L350_Simple_Backend_v10.pyvisa.ResourceManager')
+    def test_main_runs_and_completes(self, mock_rm, mock_time, mock_open_file,
                                      mock_plt_show, mock_ls_class, mock_input, mock_file_dialog, mock_tk):
         # --- MOCK SETUP ---
+        # Configure mock_rm to return a mock instrument
+        mock_instrument_from_rm = MagicMock()
+        mock_rm.return_value.open_resource.return_value = mock_instrument_from_rm
+        mock_instrument_from_rm.query.return_value = "LSCI,MODEL350,12345,1.0" # IDN query
+
         mock_controller = MagicMock()
         mock_ls_class.return_value = mock_controller