Fix more tests

Author: tonyxty

openevolve/process_parallel.py

@@ -262,7 +262,7 @@ def _run_iteration_worker(
 class ProcessParallelController:
     """Controller for process-based parallel evolution"""
 
-    def __init__(self, config: Config, evaluation_file: str, evaluation_object: EvaluationObject, database: ProgramDatabase):
+    def __init__(self, config: Config, evaluation_file: str, evaluation_object: Optional[EvaluationObject], database: ProgramDatabase):
         self.config = config
         self.evaluation_file = evaluation_file
         self.evaluation_object = evaluation_object

tests/test_process_parallel.py

@@ -64,15 +64,15 @@ def tearDown(self):
 
     def test_controller_initialization(self):
         """Test that controller initializes correctly"""
-        controller = ProcessParallelController(self.config, self.eval_file, self.database)
+        controller = ProcessParallelController(self.config, self.eval_file, None, self.database)
 
         self.assertEqual(controller.num_workers, 2)
         self.assertIsNone(controller.executor)
         self.assertIsNotNone(controller.shutdown_event)
 
     def test_controller_start_stop(self):
         """Test starting and stopping the controller"""
-        controller = ProcessParallelController(self.config, self.eval_file, self.database)
+        controller = ProcessParallelController(self.config, self.eval_file, None, self.database)
 
         # Start controller
         controller.start()
@@ -85,7 +85,7 @@ def test_controller_start_stop(self):
 
     def test_database_snapshot_creation(self):
         """Test creating database snapshot for workers"""
-        controller = ProcessParallelController(self.config, self.eval_file, self.database)
+        controller = ProcessParallelController(self.config, self.eval_file, None, self.database)
 
         snapshot = controller._create_database_snapshot()
 
@@ -106,7 +106,7 @@ def test_run_evolution_basic(self):
         """Test basic evolution run"""
 
         async def run_test():
-            controller = ProcessParallelController(self.config, self.eval_file, self.database)
+            controller = ProcessParallelController(self.config, self.eval_file, None, self.database)
 
             # Mock the executor to avoid actually spawning processes
             with patch.object(controller, "_submit_iteration") as mock_submit:
@@ -152,7 +152,7 @@ async def run_test():
 
     def test_request_shutdown(self):
         """Test graceful shutdown request"""
-        controller = ProcessParallelController(self.config, self.eval_file, self.database)
+        controller = ProcessParallelController(self.config, self.eval_file, None, self.database)
 
         # Request shutdown
         controller.request_shutdown()