test_python.py

#!/usr/bin/env python3

"""
Simple test script to verify Python modules work correctly
"""

def test_imports():
    """Test that all required modules can be imported"""
    print("Testing Python module imports...")
    
    try:
        import click
        print("✅ click imported successfully")
    except ImportError as e:
        print(f"❌ click import failed: {e}")
        return False
    
    try:
        import json
        print("✅ json imported successfully")
    except ImportError as e:
        print(f"❌ json import failed: {e}")
        return False
    
    try:
        import pathlib
        print("✅ pathlib imported successfully")
    except ImportError as e:
        print(f"❌ pathlib import failed: {e}")
        return False
    
    try:
        from loguru import logger
        print("✅ loguru imported successfully")
    except ImportError as e:
        print(f"❌ loguru import failed: {e}")
        return False
    
    try:
        import tiktoken
        print("✅ tiktoken imported successfully")
    except ImportError as e:
        print(f"❌ tiktoken import failed: {e}")
        return False
    
    # Test docx import separately as it seems to be problematic
    try:
        import docx
        print("✅ python-docx imported successfully")
    except ImportError as e:
        print(f"❌ python-docx import failed: {e}")
        return False
    except Exception as e:
        print(f"⚠️ python-docx import had issues: {e}")
        # Don't fail the test for this
    
    return True

def test_basic_functionality():
    """Test basic functionality"""
    print("\nTesting basic functionality...")

    # Test click
    import click

    @click.command()
    @click.option('--test', default='hello')
    def dummy_command(test):
        return f"Test: {test}"

    print("✅ Click command creation works")

    # Test tiktoken
    import tiktoken
    encoding = tiktoken.get_encoding("cl100k_base")
    tokens = encoding.encode("Hello, world!")
    print(f"✅ Tiktoken encoding works: {len(tokens)} tokens")

    return True

if __name__ == "__main__":
    print("🧪 MCP Server Python Module Test\n")
    
    success = True
    success &= test_imports()
    success &= test_basic_functionality()
    
    if success:
        print("\n🎉 All tests passed!")
        exit(0)
    else:
        print("\n❌ Some tests failed!")
        exit(1)