RagFlow_Plugs/services/pdf_parser/test_service.py

from .main import PDFParsingService

def test_service_initialization():
    """测试服务初始化"""
    print("测试服务初始化...")
    try:
        service = PDFParsingService()
        print("✓ 服务初始化成功")
        return True
    except Exception as e:
        print(f"✗ 服务初始化失败: {str(e)}")
        return False

def test_module_imports():
    """测试模块导入"""
    print("测试模块导入...")
    
    modules_to_test = [
        ("pdf_splitter", ".pdf_splitter"),
        ("qwen_vl", ".qwen_vl"),
        ("workflow", ".workflow"),
        ("main", ".main")
    ]
    
    all_imported = True
    for module_name, module_path in modules_to_test:
        try:
            __import__(module_path, fromlist=["*"], level=1)
            print(f"✓ {module_name} 模块导入成功")
        except Exception as e:
            print(f"✗ {module_name} 模块导入失败: {str(e)}")
            all_imported = False
    
    return all_imported

def main():
    """测试主函数"""
    print("开始测试PDF解析服务...")
    print("=" * 50)
    
    # 测试模块导入
    module_test = test_module_imports()
    print()
    
    # 测试服务初始化
    init_test = test_service_initialization()
    print()
    
    print("=" * 50)
    if module_test and init_test:
        print("✓ 所有测试通过！服务可以正常使用")
        print("\n使用示例:")
        print("python -m services.pdf_parser.main --pdf_path <pdf文件路径> --output <输出json路径>")
    else:
        print("✗ 测试失败！请检查错误信息")

if __name__ == "__main__":
    main()