llm-in-text/backend/tests/quick_verify.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
快速验证脚本
验证TTS/ASR模块修复是否正确应用

运行方式:
    python backend/tests/quick_verify.py
"""

import os
import sys
from pathlib import Path

# 设置控制台编码
if sys.platform == 'win32':
    import io
    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
    sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding='utf-8')

# 确保可以导入backend模块
script_path = Path(__file__).resolve()
project_root = script_path.parent.parent.parent
sys.path.insert(0, str(project_root))

print(f"项目根目录: {project_root}")
print(f"脚本路径: {script_path}")


def check_file_exists(filepath: str, description: str) -> bool:
    """检查文件是否存在"""
    full_path = project_root / filepath
    exists = full_path.exists()
    status = "[OK]" if exists else "[FAIL]"
    print(f"{status} {description}: {filepath} (完整路径: {full_path})")
    return exists


def check_function_exists(module_name: str, function_name: str) -> bool:
    """检查函数是否存在"""
    try:
        module = __import__(module_name, fromlist=[function_name])
        exists = hasattr(module, function_name)
        status = "[OK]" if exists else "[FAIL]"
        print(f"{status} 函数存在: {module_name}.{function_name}")
        return exists
    except Exception as e:
        print(f"[FAIL] 导入失败: {module_name} - {e}")
        return False


def check_environment_variable(var_name: str, expected_default: str) -> bool:
    """检查环境变量默认值"""
    try:
        # 清除可能存在的环境变量
        original_value = os.environ.get(var_name)
        if var_name in os.environ:
            del os.environ[var_name]
        
        # 重新导入模块
        if 'backend.tts_asr' in sys.modules:
            del sys.modules['backend.tts_asr']
        
        from backend.tts_asr import (
            TTS_ASR_DEVICE, TTS_ASR_MODEL_SIZE, TTS_ASR_QUANTIZE,
            TTS_ASR_OFFLINE_MODE, TTS_ASR_WARMUP, TTS_ASR_WARMUP_TIMEOUT,
            TTS_ASR_IDLE_TIMEOUT, TTS_ASR_MPS_MEMORY_LIMIT_MB
        )
        
        var_map = {
            'TTS_ASR_DEVICE': TTS_ASR_DEVICE,
            'TTS_ASR_MODEL_SIZE': TTS_ASR_MODEL_SIZE,
            'TTS_ASR_QUANTIZE': TTS_ASR_QUANTIZE,
            'TTS_ASR_OFFLINE_MODE': TTS_ASR_OFFLINE_MODE,
            'TTS_ASR_WARMUP': TTS_ASR_WARMUP,
            'TTS_ASR_WARMUP_TIMEOUT': TTS_ASR_WARMUP_TIMEOUT,
            'TTS_ASR_IDLE_TIMEOUT': TTS_ASR_IDLE_TIMEOUT,
            'TTS_ASR_MPS_MEMORY_LIMIT_MB': TTS_ASR_MPS_MEMORY_LIMIT_MB,
        }
        
        actual_value = var_map.get(var_name)
        if var_name == 'TTS_ASR_MODEL_SIZE':
            expected = 'auto'
        elif var_name == 'TTS_ASR_QUANTIZE':
            expected = False
        elif var_name == 'TTS_ASR_OFFLINE_MODE':
            expected = False
        elif var_name == 'TTS_ASR_WARMUP':
            expected = True
        elif var_name == 'TTS_ASR_WARMUP_TIMEOUT':
            expected = 120
        elif var_name == 'TTS_ASR_IDLE_TIMEOUT':
            expected = 0
        elif var_name == 'TTS_ASR_MPS_MEMORY_LIMIT_MB':
            expected = 8192
        else:
            expected = expected_default
        
        matches = actual_value == expected
        status = "[OK]" if matches else "[FAIL]"
        print(f"{status} 环境变量默认值: {var_name} = {actual_value} (预期: {expected})")
        return matches
        
    except Exception as e:
        print(f"[FAIL] 检查环境变量失败: {var_name} - {e}")
        return False


def main():
    print("="*70)
    print("TTS/ASR模块快速验证")
    print("="*70)
    
    checks = []
    
    # 1. 检查文件
    print("\n[1] 文件检查")
    print("-"*70)
    checks.append(check_file_exists("backend/tts_asr.py", "主模块文件"))
    checks.append(check_file_exists("backend/tests/test_tts_asr_unit.py", "单元测试"))
    checks.append(check_file_exists("backend/tests/test_tts_asr_integration.py", "集成测试"))
    checks.append(check_file_exists("backend/tests/simulate_macos.py", "macOS模拟工具"))
    checks.append(check_file_exists("backend/tests/TESTING_GUIDE.md", "测试指南"))
    checks.append(check_file_exists("backend/TTS_ASR_MACOS_FIX.md", "修复文档"))
    
    # 2. 检查核心函数
    print("\n[2] 核心函数检查")
    print("-"*70)
    checks.append(check_function_exists("backend.tts_asr", "_is_apple_silicon"))
    checks.append(check_function_exists("backend.tts_asr", "_detect_device_capabilities"))
    checks.append(check_function_exists("backend.tts_asr", "_get_recommended_model_size"))
    checks.append(check_function_exists("backend.tts_asr", "_validate_audio_data"))
    checks.append(check_function_exists("backend.tts_asr", "_resample_audio_robust"))
    checks.append(check_function_exists("backend.tts_asr", "_check_model_cached"))
    
    # 3. 检查数据类
    print("\n[3] 数据类检查")
    print("-"*70)
    checks.append(check_function_exists("backend.tts_asr", "DeviceCapabilities"))
    checks.append(check_function_exists("backend.tts_asr", "ModelStatus"))
    
    # 4. 检查环境变量
    print("\n[4] 环境变量默认值检查")
    print("-"*70)
    checks.append(check_environment_variable("TTS_ASR_DEVICE", "auto"))
    checks.append(check_environment_variable("TTS_ASR_MODEL_SIZE", "auto"))
    checks.append(check_environment_variable("TTS_ASR_QUANTIZE", "false"))
    checks.append(check_environment_variable("TTS_ASR_OFFLINE_MODE", "false"))
    
    # 5. 检查常量
    print("\n[5] 常量检查")
    print("-"*70)
    try:
        from backend.tts_asr import WHISPER_MODEL_SIZES, APPLE_SILICON_DEFAULT_SIZE
        expected_sizes = ['tiny', 'base', 'small', 'medium', 'large', 'turbo']
        sizes_match = list(WHISPER_MODEL_SIZES.keys()) == expected_sizes
        status = "[OK]" if sizes_match else "[FAIL]"
        print(f"{status} WHISPER_MODEL_SIZES: {list(WHISPER_MODEL_SIZES.keys())}")
        checks.append(sizes_match)
        
        size_match = APPLE_SILICON_DEFAULT_SIZE == 'small'
        status = "[OK]" if size_match else "[FAIL]"
        print(f"{status} APPLE_SILICON_DEFAULT_SIZE: {APPLE_SILICON_DEFAULT_SIZE}")
        checks.append(size_match)
    except Exception as e:
        print(f"[FAIL] 常量检查失败: {e}")
        checks.extend([False, False])
    
    # 汇总结果
    print("\n" + "="*70)
    print("验证结果")
    print("="*70)
    
    total = len(checks)
    passed = sum(checks)
    
    print(f"通过: {passed}/{total}")
    
    if all(checks):
        print("\n[SUCCESS] 所有验证通过！TTS/ASR模块修复已正确应用。")
        return 0
    else:
        print("\n[FAILED] 部分验证失败，请检查上述错误。")
        return 1


if __name__ == '__main__':
    sys.exit(main())