"""
测试处理器兼容性
"""
from pipeline.handler.advanced_heatmap_processor import AdvancedHeatmapProcessor
from pipeline.handler.draw_heatmap_box_processor import DrawHeatMapBoxProcessor
from pipeline.pipeline_data import PipelineData
import numpy as np
import cv2


def test_compatibility():
    print("测试处理器兼容性...")
    
    # 创建处理器
    heatmap_proc = AdvancedHeatmapProcessor()
    draw_proc = DrawHeatMapBoxProcessor()
    
    print(f"处理器创建成功: {heatmap_proc.name}, {draw_proc.name}")
    
    # 创建模拟数据
    data = PipelineData()
    # 创建一个模拟帧
    data.frame = np.zeros((480, 640, 3), dtype=np.uint8)
    data.frame_idx = 1
    
    # 由于我们无法创建真实的YOLO结果，我们跳过处理步骤，主要验证类定义和依赖
    print("处理器兼容性测试完成!")


if __name__ == "__main__":
    test_compatibility()