Function test_basic_functionality

tests/test_enhanced_library.py:11–83

Test basic functionality of the enhanced library.

()

def test_basic_functionality():
    """Test basic functionality of the enhanced library."""
    print("🧪 Testing Enhanced Document Data Extractor Library")
    print("=" * 50)
    
    extractor = DocumentExtractor()
    
    # Test 1: Text file conversion
    print("\n1. Testing text file conversion...")
    try:
        with tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False) as f:
            f.write("This is a test document.\n\nIt has multiple lines.\n\n# This is a heading\n\n- List item 1\n- List item 2")
            temp_file = f.name
        
        result = extractor.extract(temp_file)
        print(f"✅ Text conversion successful: {len(result.content)} characters")
        print(f"   Metadata: {result.metadata}")
        
        # Test different output formats
        markdown = result.extract_markdown()
        html = result.extract_html()
        json_output = result.extract_data()
        
        print(f"   Markdown length: {len(markdown)}")
        print(f"   HTML length: {len(html)}")
        print(f"   JSON keys: {list(json_output.keys())}")
        
        os.unlink(temp_file)
        
    except Exception as e:
        print(f"❌ Text conversion failed: {e}")
    
    # Test 2: URL conversion
    print("\n2. Testing URL conversion...")
    try:
        result = extractor.convert_url("https://httpbin.org/html")
        print(f"✅ URL conversion successful: {len(result.content)} characters")
        print(f"   Status code: {result.metadata.get('status_code')}")
        
    except Exception as e:
        print(f"❌ URL conversion failed: {e}")
    
    # Test 3: Plain text conversion
    print("\n3. Testing plain text conversion...")
    try:
        text = "This is plain text for testing the extractor."
        result = extractor.extract_text(text)
        print(f"✅ Plain text conversion successful: {len(result.content)} characters")
        
    except Exception as e:
        print(f"❌ Plain text conversion failed: {e}")
    
    # Test 4: Supported formats
    print("\n4. Testing supported formats...")
    try:
        formats = extractor.get_supported_formats()
        print(f"✅ Supported formats: {formats}")