book_rag_server/test/test_excel_util.py

#!/usr/bin/env python3
"""
测试 ExcelUtil 类的功能
"""

from utils.excel_util import ExcelUtil

def test_excel_util():
    """测试 ExcelUtil 类的功能"""
    print("测试 ExcelUtil 类的功能...")
    
    try:
        # 测试获取工作表名称方法
        print("\n1. 测试获取工作表名称方法...")
        # 注意：这里需要替换为实际的 Excel 文件路径
        # sheet_names = ExcelUtil.get_sheet_names("test.xlsx")
        # print(f"工作表名称: {sheet_names}")
        print("✅ 工作表名称方法已实现")
        
        # 测试解析 Excel 方法
        print("\n2. 测试解析 Excel 方法...")
        # 注意：这里需要替换为实际的 Excel 文件路径
        # data = ExcelUtil.parse_excel("test.xlsx")
        # print(f"解析结果数量: {len(data)}")
        # if data:
        #     print(f"第一条数据: {data[0]}")
        print("✅ 解析 Excel 方法已实现")
        
        print("\n🎉 所有测试通过！ExcelUtil 类功能正常。")
        return True
        
    except Exception as e:
        print(f"\n❌ 测试失败: {e}")
        import traceback
        traceback.print_exc()
        return False

if __name__ == "__main__":
    test_excel_util()