Update test_auto_landing_worker.py

Author: danielquzhao

tests/integration/test_auto_landing_worker.py

@@ -2,7 +2,6 @@
 Test auto-landing worker process.
 """
 
-import datetime
 import multiprocessing as mp
 import pathlib
 import time
@@ -17,29 +16,17 @@
 from modules.common.modules import orientation
 from modules.common.modules import position_local
 from modules.common.modules.mavlink import drone_odometry_local
-from modules.common.modules.logger import logger
 from modules.common.modules.read_yaml import read_yaml
 from utilities.workers import queue_proxy_wrapper
 from utilities.workers import worker_controller
 
 
 # Test-specific parameters
 WORKER_PERIOD = 0.1  # seconds
-LOG_TIMINGS = False  # Disable timing logging for the test
 CONFIG_FILE_PATH = pathlib.Path("config.yaml")
 detection_strategy = auto_landing.DetectionSelectionStrategy.FIRST_DETECTION
 # detection_strategy = auto_landing.DetectionSelectionStrategy.HIGHEST_CONFIDENCE
 
-# Ensure logs directory exists and create timestamped subdirectory
-LOG_DIR = pathlib.Path("logs")
-LOG_DIR.mkdir(parents=True, exist_ok=True)
-
-# Create a timestamped subdirectory for this test session
-TIMESTAMP_FORMAT = "%Y-%m-%d_%H-%M-%S"
-test_session_dir = LOG_DIR / datetime.datetime.now().strftime(TIMESTAMP_FORMAT)
-test_session_dir.mkdir(parents=True, exist_ok=True)
-
-
 def simulate_detection_input(
     input_queue: queue_proxy_wrapper.QueueProxyWrapper,
     detections: list,
@@ -114,20 +101,14 @@ def main() -> int:
     """
     Main function.
     """
-    # Logger
-    test_name = pathlib.Path(__file__).stem
-    result, local_logger = logger.Logger.create(test_name, LOG_TIMINGS)
-    assert result  # Logger initialization should succeed
-    assert local_logger is not None
-
     # Read config file
     result, config = read_yaml.open_config(CONFIG_FILE_PATH)
     assert result and config is not None, "Failed to read config file"
 
     # Check if the feature is enabled in the config
     auto_landing_config = config.get("auto_landing", {})
     if not auto_landing_config.get("enabled", False):
-        local_logger.info("Auto-landing is disabled in config.yaml, skipping test.")
+        print("Auto-landing is disabled in config.yaml, skipping test.")
         return 0
 
     # Extract parameters from config
@@ -137,7 +118,7 @@ def main() -> int:
         image_height = config["video_input"]["height"]
         image_width = config["video_input"]["width"]
     except KeyError as e:
-        local_logger.error(f"Config file missing required key: {e}")
+        print(f"ERROR: Config file missing required key: {e}")
         return -1
 
     # Setup
@@ -170,7 +151,7 @@ def main() -> int:
     time.sleep(0.5)
 
     # Test 1: Send a single detection and verify it's processed
-    local_logger.info("--- Test 1: Processing single detection ---")
+    print("--- Test 1: Processing single detection ---")
     detection1_bbox = [200, 200, 400, 400]
     detection1 = create_test_detection(detection1_bbox, 1, 0.9)
     simulate_detection_input(
@@ -189,10 +170,10 @@ def main() -> int:
     assert hasattr(landing_info, "angle_x")
     assert hasattr(landing_info, "angle_y")
     assert hasattr(landing_info, "target_dist")
-    local_logger.info("--- Test 1 Passed ---")
+    print("--- Test 1 Passed ---")
 
     # Test 2: Verify strategy with multiple detections
-    local_logger.info(f"--- Test 2: Verifying {detection_strategy.value} strategy ---")
+    print(f"--- Test 2: Verifying {detection_strategy.value} strategy ---")
 
     # This is the first detection in the list
     detection_first_bbox = [100, 100, 200, 200]
@@ -243,7 +224,7 @@ def main() -> int:
     assert math.isclose(
         landing_info2.angle_y, expected_angle_y, rel_tol=1e-7
     ), f"Expected angle_y to be {expected_angle_y}, but got {landing_info2.angle_y}"
-    local_logger.info("--- Test 2 Passed ---")
+    print("--- Test 2 Passed ---")
 
     # Cleanup
     controller.request_exit()
@@ -253,9 +234,7 @@ def main() -> int:
 
     # Wait for worker to finish
     worker.join(timeout=5.0)
-    if worker.is_alive():
-        worker.terminate()
-        worker.join()
+    worker.terminate()
 
     return 0