some corrections in tests

Author: prathameshnium

tests/test_deep_simulation.py

@@ -201,23 +201,19 @@ def test_08_combined_2400_2182(self):
         mock_sleep = patch('time.sleep', side_effect=self.get_circuit_breaker(10))
         mock_sleep.start()
         self.addCleanup(mock_sleep.stop)
-        with patch('pyvisa.ResourceManager') as MockRM:
-            mock_pymeasure = patch('pymeasure.instruments.keithley.Keithley2400')
-            mock_pymeasure.start()
-
+        with patch('pyvisa.ResourceManager') as MockRM, \
+             patch('pymeasure.instruments.keithley.Keithley2400'):
             rm = MockRM.return_value
             k2182_spy = MagicMock()
             k2182_spy.assert_trigger = MagicMock()
             rm.open_resource.return_value = k2182_spy
 
-            # Add extra inputs just in case the script asks for more than
-            # expected
+            # Add extra inputs just in case the script asks for more than expected
             inputs = ['10', '1', 'test_file', 'y', 'y', 'y', 'y']
             with patch('builtins.input', side_effect=inputs), \
                  patch('pandas.DataFrame.to_csv'):
                 self.run_module_safely(
-                    "pica.keithley.k2400.Keithley_2182.Instrument_Control.IV_K2400_K2182_Instrument_Control", {})
-            mock_pymeasure.stop()
+                    "Keithley_2400_Keithley_2182.Backends.IV_K2400_K2182_Backend_v1", {})
 
     @pytest.mark.usefixtures("mock_tkinter")
     def test_09_poling(self):

tests/test_gui_modules_initialization.py

@@ -58,8 +58,7 @@ def test_gui_module_initialization(module_path):
         elif base_name.endswith('_GUI'):
             class_name = base_name
         else:
-             pytest.fail(f"Could not determine class name from module path: {module_path}")
-
+            pytest.fail(f"Could not determine class name from module path: {module_path}")
 
         # Get the class from the module and instantiate it
         AppClass = getattr(gui_module, class_name)

tests/test_package_integrity.py

@@ -74,9 +74,16 @@ def test_has_docstring(file_path):
 
     try:
         tree = ast.parse(source)
-        if not (tree.body and isinstance(tree.body[0], ast.Expr) and isinstance(tree.body[0].value, (ast.Str, ast.Constant))):
+        # A docstring is the first statement in a module and must be an expression
+        # containing a string.
+        # For Python < 3.8, this is ast.Str. For >= 3.8, it's ast.Constant holding a str.
+        docstring_node = ast.get_docstring(tree, clean=False)
+
+        # ast.get_docstring returns the string value or None if not found.
+        # We just need to know if it exists.
+        if docstring_node is None:
             warnings.warn(f"JOSS STANDARD MISSING: {os.path.basename(file_path)} has no top-level docstring.", UserWarning)
-            
+        
     except SyntaxError:
         # Syntax errors are handled by the other test
         pass