Tests added

Author: prathameshnium

Delta_mode_Keithley_6221_2182/Delta_RT_K6221_K2182_L350_T_Control_GUI_v5.py

@@ -944,7 +944,7 @@ def _update_measurement_loop(self):
             else:
                 # Slightly less than 1s to prevent drift
                 self.root.after(900, self._update_measurement_loop)
-        except Exception as e:
+        except Exception:
             self.log(f"RUNTIME ERROR: {traceback.format_exc()}")
             self.stop_measurement()
 

Lock_in_amplifier/BasicTest_S830_Backend_v1.py

@@ -75,7 +75,7 @@ def main():
 
     except pyvisa.errors.VisaIOError as e:
         # Handle VISA-specific errors, like instrument not found or timeout.
-        print(f"VISA Error: Could not connect or communicate with the instrument.")
+        print("VISA Error: Could not connect or communicate with the instrument.")
         print(f"   Details: {e}")
         print("   Troubleshooting tips:")
         print("   - Is the instrument powered on and connected?")

deployment/build.py

@@ -115,7 +115,7 @@ def run_command(command):
     try:
         subprocess.run(command, check=True, text=True, capture_output=True)
     except subprocess.CalledProcessError as e:
-        print(f"--- ERROR ---", file=sys.stderr)
+        print("--- ERROR ---", file=sys.stderr)
         print(e.stdout, file=sys.stderr)
         print(e.stderr, file=sys.stderr)
         raise
@@ -175,9 +175,9 @@ def build():
         final_app_dir)
 
     print("\n--- Build Complete! ---")
-    print(f"The final application folder is located at:")
+    print("The final application folder is located at:")
     print(f"  {final_app_dir}")
-    print(f"\nA distributable ZIP file has been created at:")
+    print("\nA distributable ZIP file has been created at:")
     print(f"  {os.path.join(DIST_DIR, zip_filename + '.zip')}")
 
 

tests/test_deep_simulation.py

@@ -69,7 +69,7 @@ def run_module_safely(self, module_name):
                 print(f"   -> Running {module_name}.main()...", flush=True)
                 mod.main()
             else:
-                print(f"   -> Module loaded (no main function).", flush=True)
+                print("   -> Module loaded (no main function).", flush=True)
         except Exception as e:
             if "Force Test Exit" in str(e) or isinstance(e, SystemExit):
                 print(
@@ -196,7 +196,7 @@ def test_08_combined_2400_2182(self):
         # THIS WAS THE TEST CAUSING THE HANG
         # We suspect input mismatch or resource opening hang.
         with patch('pyvisa.ResourceManager') as MockRM, \
-                patch('pymeasure.instruments.keithley.Keithley2400') as MockK2400, \
+                patch('pymeasure.instruments.keithley.Keithley2400'), \
                 patch('time.sleep', side_effect=self.get_circuit_breaker(10)):
 
             rm = MockRM.return_value

tests/test_iv_k2400_loop_backend.py

@@ -1,7 +1,5 @@
 import unittest
-from unittest.mock import patch, MagicMock, mock_open
-import numpy as np
-import pandas as pd
+from unittest.mock import patch, MagicMock
 import sys
 import os
 
@@ -14,7 +12,7 @@
 class TestIVK2400LoopBackend(unittest.TestCase):
 
     @patch('builtins.input', side_effect=['10', '2', 'test_output'])
-    @patch('pymeasure.instruments.keithley.Keithley2400')
+    @patch('Keithley_2400.Backends.IV_K2400_Loop_Backend_v10.Keithley2400')
     @patch('matplotlib.pyplot.show')
     @patch('pandas.DataFrame.to_csv')
     def test_main_full_run(self, mock_to_csv, mock_plt_show, mock_keithley_class, mock_input):

tests/test_t_control_l350_backend.py

@@ -1,7 +1,6 @@
 import unittest
 from unittest.mock import patch, MagicMock, mock_open
 import pyvisa
-import time
 import sys
 import os
 
@@ -31,7 +30,7 @@ def test_initialization_success(self):
     @patch('pyvisa.ResourceManager')
     def test_initialization_failure(self, mock_rm):
         # Test that a connection error is raised if pyvisa fails
-        mock_rm.return_value.open_resource.side_effect = pyvisa.errors.VisaIOError("Test Error")
+        mock_rm.return_value.open_resource.side_effect = pyvisa.errors.VisaIOError(pyvisa.constants.VI_ERROR_RSRC_NFOUND)
         with self.assertRaises(ConnectionError):
             Lakeshore350("GPIB0::13::INSTR")
 
@@ -79,7 +78,7 @@ def test_close(self):
         self.assertIsNone(self.controller.instrument)
 
 class TestMainFunctionAndUserInput(unittest.TestCase):
-    @patch('builtins.input', side_effect=['100', '200', '300', '10', 'not-a-number', '50', '350', '400'])
+    @patch('builtins.input', side_effect=['100', '200', '300', '10', 'not-a-number', '50', '350', '400', '10'])
     def test_get_user_parameters(self, mock_input):
         # First call: Valid input
         start, end, rate, cutoff = get_user_parameters()
@@ -89,15 +88,17 @@ def test_get_user_parameters(self, mock_input):
         # Second call: Invalid text input, should retry and get the next valid ones
         start, end, rate, cutoff = get_user_parameters()
         self.assertEqual((start, end, cutoff), (50, 350, 400))
-    
+        self.assertEqual(rate, 10)
+
+    @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')
     @patch('matplotlib.pyplot.show')
     @patch('builtins.open', new_callable=mock_open)
     @patch('time.sleep', MagicMock())
     @patch('time.time', side_effect=[1000, 1002, 1004, 1006, 1008, 1010]) # Simulate time passing
-    def test_main_runs_and_completes(self, mock_time, mock_open_file, mock_plt_show, mock_ls_class, mock_input, mock_file_dialog):
+    def test_main_runs_and_completes(self, mock_time, mock_open_file, mock_plt_show, mock_ls_class, mock_input, mock_file_dialog, mock_tk):
         # --- MOCK SETUP ---
         mock_controller = MagicMock()
         mock_ls_class.return_value = mock_controller