remove empty lines with spaces.

Author: neuromechanist

examples/test_emg_simulation.py

@@ -8,27 +8,27 @@
 async def test_emg():
     """Test EMG simulation."""
     print("\nTesting EMG simulation...")
-    
+
     server = MatlabServer()
     await server.initialize()
-    
+
     try:
         # Add matlab_scripts directory to MATLAB path
         script_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), 'matlab_scripts'))
         print(f"Adding script directory to path: {script_dir}")
         await server.engine.execute(f"addpath(genpath('{script_dir}'))")
-        
+
         # Verify the path was added
         await server.engine.execute("path")
-        
+
         print("Running EMG simulation...")
         # Run the EMG simulation
         result = await server.engine.execute(
             "simulate_emg",
             capture_plots=True
         )
         print(f"Simulation complete. Result: {result}")
-        
+
         # Verify figures were captured
         print(f"Number of figures captured: {len(result.figures)}")
         if len(result.figures) > 0:
@@ -38,7 +38,7 @@ async def test_emg():
                 print(f"- {fig}")
         else:
             print("No figures were captured")
-            
+
     except Exception as e:
         print(f"Test failed: {str(e)}")
         raise

examples/test_engine.py

@@ -13,15 +13,15 @@
 async def test_engine():
     """Test basic MATLAB engine functionality."""
     engine = MatlabEngine()
-    
+
     try:
         # Initialize engine
         await engine.initialize()
-        
+
         # Run a simple MATLAB command
         result = await engine.execute('2 + 2')
         print(f"2 + 2 = {result.output}")
-        
+
         # Try plotting
         script_path = Path(__file__).parent / "matlab_scripts" / "test_plot.m"
         if script_path.exists():
@@ -32,17 +32,17 @@ async def test_engine():
                 capture_plots=True
             )
             print(f"Generated {len(result.figures)} figures")
-            
+
             # Save figures
             output_dir = Path("test_output")
             output_dir.mkdir(exist_ok=True)
-            
+
             for i, fig in enumerate(result.figures):
                 ext = ".png" if fig.format.value == "png" else ".svg"
                 output_file = output_dir / f"figure_{i}{ext}"
                 output_file.write_bytes(fig.data)
                 print(f"Saved {output_file}")
-        
+
     except Exception as e:
         print(f"Error: {str(e)}")
     finally:

examples/test_initialization.py

@@ -2,7 +2,6 @@
 
 import asyncio
 import os
-from pathlib import Path
 
 from matlab_mcp.server import MatlabServer
 from matlab_mcp.models import FigureFormat
@@ -12,25 +11,25 @@ async def test_server_initialization():
     """Test server initialization sequence."""
     print("\nTesting server initialization with detailed logging...")
     print("Current MATLAB_PATH:", os.getenv('MATLAB_PATH', 'Not set'))
-    
+
     server = MatlabServer()
-    
+
     try:
         # Test initial state
         print("Checking initial state...")
         assert not server._initialized
         assert server.engine.eng is None
         print("Initial state verified")
-        
+
         # Test initialization
         print("Attempting server initialization...")
         await server.initialize()
         print("Server initialized")
-        
+
         assert server._initialized
         assert server.engine.eng is not None
         print("Server initialization successful")
-        
+
         # Test MATLAB functionality
         print("Testing MATLAB functionality...")
         ver = server.engine.eng.version()
@@ -46,59 +45,59 @@ async def test_server_initialization():
 async def test_figure_cleanup():
     """Test figure capture and cleanup."""
     print("\nTesting figure capture and cleanup...")
-    
+
     server = MatlabServer()
     await server.initialize()
-    
+
     # Create a test plot
     result = await server.engine.execute(
         "figure; plot(1:10); title('Test Plot')",
         capture_plots=True
     )
-    
+
     # Verify figures were captured
     assert len(result.figures) > 0
     assert any(fig.format == FigureFormat.PNG for fig in result.figures)
     assert any(fig.format == FigureFormat.SVG for fig in result.figures)
     print("Figure capture successful")
-    
+
     # Verify figures were cleaned up in MATLAB
     fig_count = server.engine.eng.eval('length(get(groot, "Children"))', nargout=1)
     assert fig_count == 0
     print("Figure cleanup successful")
-    
+
     # Verify temporary files were cleaned up
     output_dir = server.engine.output_dir
     assert not any(output_dir.glob("figure_*.png"))
     assert not any(output_dir.glob("figure_*.svg"))
     print("Temporary file cleanup successful")
-    
+
     # Clean up
     server.close()
 
 
 async def test_concurrent_requests():
     """Test handling of concurrent protocol requests."""
     print("\nTesting concurrent protocol requests...")
-    
+
     server = MatlabServer()
-    
+
     # Simulate concurrent protocol requests
     results = await asyncio.gather(
         server._list_tools(),
         server._list_resources(),
         server._list_resource_templates()
     )
-    
+
     # Verify all requests completed successfully
     assert all(result is not None for result in results)
     print("Concurrent request handling successful")
-    
+
     # Verify server was initialized only once
     assert server._initialized
     assert server.engine.eng is not None
     print("Single initialization verified")
-    
+
     # Clean up
     server.close()
 
@@ -110,7 +109,7 @@ async def run_tests():
         await test_figure_cleanup()
         await test_concurrent_requests()
         print("\nAll initialization and cleanup tests passed!")
-        
+
     except Exception as e:
         print(f"\nTest failed: {str(e)}")
         raise

examples/test_plot_capture.py

@@ -18,7 +18,7 @@ async def test_plot_capture():
     # Set MATLAB environment
     matlab_path = os.getenv('MATLAB_PATH', '/Applications/MATLAB_R2024a.app')
     print(f"MATLAB_PATH: {matlab_path}")
-    
+
     # Ask user for correct MATLAB path if default doesn't exist
     if not os.path.exists(matlab_path):
         print(f"MATLAB not found at {matlab_path}")
@@ -32,53 +32,53 @@ async def test_plot_capture():
             return
         os.environ['MATLAB_PATH'] = matlab_path
         print(f"Set MATLAB_PATH to: {matlab_path}")
-    
+
     server = MatlabServer()
     print("Created MatlabServer instance")
-    
+
     try:
         print("Initializing MATLAB engine...")
         await server.engine.initialize()
         print("MATLAB engine initialized")
-        
+
         # Execute test plot script
         script_path = Path(__file__).parent / "matlab_scripts" / "test_plot.m"
         if not script_path.exists():
             print(f"Test script not found at {script_path}")
             return
-            
+
         print(f"Executing script: {script_path}")
         result = await server.engine.execute(
             str(script_path),
             is_file=True,
             capture_plots=True
         )
-        
+
         if result.error:
             print(f"Error executing script: {result.error}")
             return
-            
+
         # Verify we got both PNG and SVG versions
         png_figures = [f for f in result.figures if f.format == FigureFormat.PNG]
         svg_figures = [f for f in result.figures if f.format == FigureFormat.SVG]
-        
+
         print(f"Generated {len(png_figures)} PNG figures")
         print(f"Generated {len(svg_figures)} SVG figures")
-        
+
         # Save figures to verify content
         output_dir = Path("test_output")
         output_dir.mkdir(exist_ok=True)
-        
+
         for i, fig in enumerate(result.figures):
             ext = ".png" if fig.format == FigureFormat.PNG else ".svg"
             output_file = output_dir / f"figure_{i}{ext}"
             output_file.write_bytes(fig.data)
             print(f"Saved {output_file}")
-            
+
     except Exception as e:
         print("Error during execution:")
         print(traceback.format_exc())
-        
+
     finally:
         print("Cleaning up MATLAB engine...")
         server.engine.cleanup()

examples/test_sections.py

@@ -10,22 +10,22 @@
 async def test_section_execution():
     """Test section-based MATLAB script execution."""
     print("Testing MATLAB Section Execution...")
-    
+
     server = MatlabServer()
-    
+
     # Test section parsing
     print("\nTesting section parsing...")
     script_path = Path("examples/matlab_scripts/section_test.m")
     if not script_path.exists():
         print(f"Error: Script not found at {script_path}")
         return
-        
+
     sections = get_section_info(script_path)
     print(f"Found {len(sections)} sections:")
     for section in sections:
         print(f"- {section['title']} (lines {section['start_line']}-{section['end_line']})")
         print(f"  Preview: {section['preview']}")
-    
+
     # Test executing each section
     print("\nExecuting sections one by one...")
     for section in sections:
@@ -34,26 +34,26 @@ async def test_section_execution():
             str(script_path),
             (section['start_line'], section['end_line'])
         )
-        
+
         print("Output:")
         print(result.output)
-        
+
         print("Workspace variables:")
         for var, value in result.workspace.items():
             print(f"- {var}: {value}")
-            
+
         if result.figures:
             print(f"Generated {len(result.figures)} figures")
-            
+
             # Save figures
             output_dir = Path("test_output")
             output_dir.mkdir(exist_ok=True)
-            
+
             for i, fig_data in enumerate(result.figures):
                 output_path = output_dir / f"section_{section['start_line']}_figure_{i}.png"
                 output_path.write_bytes(fig_data)
                 print(f"Saved figure to: {output_path}")
-    
+
     # Clean up
     server.engine.cleanup()
     print("\nTest completed successfully!")

examples/test_server.py

@@ -9,41 +9,41 @@
 async def test_basic_execution():
     """Test basic MATLAB script execution."""
     print("Testing MATLAB MCP Server...")
-    
+
     server = MatlabServer()
-    
+
     # Test direct command execution
     print("\nTesting direct command execution...")
     result = await server.engine.execute(
         "a = 5; b = 10; c = a + b; fprintf('Sum: %d\\n', c)"
     )
     print(f"Output: {result.output}")
     print(f"Workspace: {result.workspace}")
-    
+
     # Test script file execution
     print("\nTesting script file execution...")
     script_path = Path("examples/matlab_scripts/test_plot.m")
     if not script_path.exists():
         print(f"Error: Script not found at {script_path}")
         return
-        
+
     result = await server.engine.execute(
         str(script_path),
         is_file=True
     )
     print(f"Output: {result.output}")
     print(f"Workspace: {result.workspace}")
     print(f"Number of figures captured: {len(result.figures)}")
-    
+
     # Save captured figures
     output_dir = Path("test_output")
     output_dir.mkdir(exist_ok=True)
-    
+
     for i, fig_data in enumerate(result.figures):
         output_path = output_dir / f"figure_{i}.png"
         output_path.write_bytes(fig_data)
         print(f"Saved figure to: {output_path}")
-    
+
     # Clean up
     server.engine.cleanup()
     print("\nTest completed successfully!")