Fix for test_deep_simulation.py

Author: prathameshnium

Keithley_6517B/High_Resistance/Backends/IV_K6517B_Simple_Backend_v10.py

@@ -106,7 +106,7 @@ def main():
         keithley.write(':SYSTem:ZCHeck ON')  # type: ignore
         time.sleep(5)
         print("Step 2: Acquiring zero correction value...")
-        # keithley.write(':SYSTem:ZCORrect:ACQuire')
+        keithley.write(':SYSTem:ZCORrect:ACQuire')
         time.sleep(5)
         print("Step 3: Disabling Zero Check mode...")
         keithley.write(':SYSTem:ZCHeck OFF')

tests/test_deep_simulation.py

@@ -45,6 +45,13 @@ def setUp(self):
     def tearDown(self):
         print(f"[TEST END]   {self._testMethodName}\n", flush=True)
 
+    # -------------------------------------------------------------------------
+    # HELPER: The "Watchdog" Timer
+    # -------------------------------------------------------------------------
+    def _timeout_handler(self, signum, frame):
+        raise TimeoutError(
+            f"Test {self._testMethodName} took longer than 30s! Infinite Loop suspected.")
+
     def run_module_safely(self, module_name):
         """Imports and runs a module with a strict 30-second timeout."""
         # Set an alarm for 30 seconds (Works on Linux/GitHub Actions)

tests/test_fixes.py

@@ -5,7 +5,7 @@ class TestFixes(unittest.TestCase):
 
     @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('Keithley_2400.Backends.IV_K2400_Loop_Backend_v10.Keithley2400')
     @patch('matplotlib.pyplot.show')
     @patch('pandas.DataFrame.to_csv')
     def test_iv_k2400_fix(self, mock_to_csv, mock_plt_show, mock_keithley_class, mock_input):