DataOps-platform/test_parse_resume_mapping.py

#!/usr/bin/env python3
"""
测试修改后的parse_resume_with_qwen函数的字段映射功能
"""

import os
import sys
import logging
from pathlib import Path

# 添加项目根目录到Python路径
project_root = Path(__file__).parent
sys.path.insert(0, str(project_root))

# 配置日志
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s'
)

def test_field_mapping():
    """测试字段映射功能"""
    try:
        # 导入函数
        from app.core.data_parse.parse_resume import parse_resume_with_qwen
        
        # 测试文件路径
        test_file = "方全.pdf"
        
        # 检查文件是否存在
        if not os.path.exists(test_file):
            print(f"错误: 测试文件 {test_file} 不存在")
            return False
        
        print(f"开始测试字段映射功能...")
        print(f"测试文件: {test_file}")
        print(f"文件大小: {os.path.getsize(test_file)} 字节")
        
        # 调用函数
        result = parse_resume_with_qwen(test_file)
        
        # 输出映射结果
        print("\n=== 字段映射结果 ===")
        print(f"姓名(中文): {result.get('name_zh', 'N/A')}")
        print(f"姓名(英文): {result.get('name_en', 'N/A')}")
        print(f"职位(中文): {result.get('title_zh', 'N/A')}")
        print(f"职位(英文): {result.get('title_en', 'N/A')}")
        print(f"公司(中文): {result.get('hotel_zh', 'N/A')}")
        print(f"公司(英文): {result.get('hotel_en', 'N/A')}")
        print(f"手机号码: {result.get('mobile', 'N/A')}")
        print(f"固定电话: {result.get('phone', 'N/A')}")
        print(f"电子邮箱: {result.get('email', 'N/A')}")
        print(f"地址(中文): {result.get('address_zh', 'N/A')}")
        print(f"地址(英文): {result.get('address_en', 'N/A')}")
        print(f"生日: {result.get('birthday', 'N/A')}")
        print(f"年龄: {result.get('age', 'N/A')}")
        print(f"籍贯: {result.get('native_place', 'N/A')}")
        print(f"居住地: {result.get('residence', 'N/A')}")
        print(f"品牌组合: {result.get('brand_group', 'N/A')}")
        
        # 输出职业轨迹
        career_path = result.get('career_path', [])
        if career_path:
            print(f"\n职业轨迹 (共{len(career_path)}条):")
            for i, career in enumerate(career_path, 1):
                print(f"  {i}. 日期: {career.get('date', 'N/A')}")
                print(f"     公司: {career.get('hotel_zh', 'N/A')} / {career.get('hotel_en', 'N/A')}")
                print(f"     职位: {career.get('title_zh', 'N/A')} / {career.get('title_en', 'N/A')}")
        else:
            print("\n职业轨迹: 无")
        
        # 输出隶属关系
        affiliation = result.get('affiliation', [])
        if affiliation:
            print(f"\n隶属关系 (共{len(affiliation)}条):")
            for i, aff in enumerate(affiliation, 1):
                print(f"  {i}. 公司: {aff.get('company', 'N/A')}")
                print(f"     集团: {aff.get('group', 'N/A')}")
        else:
            print("\n隶属关系: 无")
        
        # 验证字段映射是否正确
        print("\n=== 字段映射验证 ===")
        mapping_verification = {
            'name_zh': '中文姓名',
            'name_en': '英文姓名', 
            'title_zh': '中文头衔',
            'title_en': '英文头衔',
            'hotel_zh': '中文酒店',
            'hotel_en': '英文酒店',
            'mobile': '手机号',
            'email': '邮箱',
            'address_zh': '中文工作地址',
            'address_en': '英文工作地址',
            'birthday': '生日',
            'age': '年龄',
            'native_place': '籍贯',
            'residence': '居住地',
            'brand_group': '品牌组合'
        }
        
        for english_field, chinese_field in mapping_verification.items():
            value = result.get(english_field, '')
            if value:
                print(f"✓ {chinese_field} -> {english_field}: {value}")
            else:
                print(f"✗ {chinese_field} -> {english_field}: 空值")
        
        print("\n=== 测试完成 ===")
        return True
        
    except Exception as e:
        print(f"测试失败: {str(e)}")
        import traceback
        traceback.print_exc()
        return False

if __name__ == "__main__":
    print("=" * 50)
    print("测试字段映射功能")
    print("=" * 50)
    
    # 测试字段映射
    success = test_field_mapping()
    
    print("\n" + "=" * 50)
    print("测试总结:")
    print(f"字段映射测试: {'✓ 成功' if success else '✗ 失败'}")
    print("=" * 50)