#!/usr/bin/env python3
"""
TTRLVR + AZR 시스템 빠른 테스트

timeout 문제를 피하기 위한 최소한의 테스트
"""

import os
import sys

# 환경 설정
os.environ['CUDA_VISIBLE_DEVICES'] = '4'
sys.path.append('/home/ubuntu/RLVR/TestTime-RLVR-v2')

def main():
    print("🚀 TTRLVR + AZR Quick Test")
    print("=" * 50)
    
    try:
        # 1. Import 테스트
        print("1️⃣ Testing imports...")
        from absolute_zero_reasoner.testtime.config import TestTimeConfig, BenchmarkConfig
        from absolute_zero_reasoner.testtime.logger import TestTimeLogger
        from utils.iterative_trainer import IterativeTrainer
        print("✅ All imports successful")
        
        # 2. 기본 객체 생성
        print("\n2️⃣ Creating basic objects...")
        config = TestTimeConfig()
        config.model_name = "Qwen/Qwen2.5-7B"
        config.debug = True
        
        benchmark_config = BenchmarkConfig(
            name='mbpp',
            data_path='/test/path',  # 더미 경로
            problem_prefix='Mbpp_',
            start_index=1,
            max_problems=1
        )
        
        logger = TestTimeLogger()
        print("✅ Basic objects created")
        
        # 3. IterativeTrainer 생성
        print("\n3️⃣ Creating IterativeTrainer...")
        trainer = IterativeTrainer(config, logger)
        print("✅ IterativeTrainer created successfully")
        
        # 4. 설정 확인
        print(f"\n4️⃣ Configuration check:")
        print(f"   - Model: {trainer.current_model_path}")
        print(f"   - Checkpoint dir: {trainer.checkpoint_dir}")
        print(f"   - Config debug: {config.debug}")
        
        # 5. 파일 존재 확인
        print(f"\n5️⃣ File existence check:")
        required_files = [
            '/home/ubuntu/RLVR/TestTime-RLVR-v2/test/configs/ttrlvr_azr_7b_single_gpu.sh',
            '/data/RLVR/checkpoints'
        ]
        
        for file_path in required_files:
            exists = os.path.exists(file_path)
            status = "✅" if exists else "❌"
            print(f"   {status} {file_path}")
        
        print(f"\n🎉 Quick test completed successfully!")
        print(f"System appears ready for full integration test.")
        
        return 0
        
    except Exception as e:
        print(f"\n❌ Quick test failed: {e}")
        import traceback
        traceback.print_exc()
        return 1

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