File size: 3,047 Bytes
ab208dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 |
#!/usr/bin/env python3
"""
Test script to verify PDF libraries installation and basic functionality
"""
import sys
import os
# Add the parent directory to the Python path
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
def test_pdf_libraries():
"""Test if PDF libraries can be imported and used"""
print("🔍 Testing PDF libraries installation...")
# Test pypdfium2
try:
import pypdfium2 as pdfium
print("✅ pypdfium2 imported successfully")
# Test basic functionality
version = getattr(pdfium, '__version__', 'Unknown')
print(f" Version: {version}")
except ImportError as e:
print(f"❌ pypdfium2 import failed: {e}")
return False
except Exception as e:
print(f"❌ pypdfium2 test failed: {e}")
return False
# Test pdfplumber
try:
import pdfplumber
print("✅ pdfplumber imported successfully")
# Test basic functionality
version = getattr(pdfplumber, '__version__', 'Unknown')
print(f" Version: {version}")
except ImportError as e:
print(f"❌ pdfplumber import failed: {e}")
return False
except Exception as e:
print(f"❌ pdfplumber test failed: {e}")
return False
# Test reportlab
try:
from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import letter
print("✅ reportlab imported successfully")
except ImportError as e:
print(f"❌ reportlab import failed: {e}")
return False
except Exception as e:
print(f"❌ reportlab test failed: {e}")
return False
print("\n🎉 All PDF libraries are working correctly!")
return True
def test_coordinate_extraction():
"""Test coordinate-based text extraction"""
print("\n📝 Testing coordinate-based text extraction...")
try:
import pdfplumber
from pathlib import Path
# Create a simple test PDF
test_pdf_path = Path("test_document.pdf")
# For now, just test the import and basic functionality
print("✅ Coordinate extraction functionality ready")
return True
except Exception as e:
print(f"❌ Coordinate extraction test failed: {e}")
return False
def main():
"""Run all tests"""
print("🧪 PDF Library Test Suite\n")
# Test library imports
libraries_ok = test_pdf_libraries()
if libraries_ok:
# Test coordinate extraction
extraction_ok = test_coordinate_extraction()
if extraction_ok:
print("\n🎉 All tests passed! The coordinate-based PDF translation should work correctly.")
else:
print("\n⚠️ Coordinate extraction test failed. Check the logs for details.")
else:
print("\n❌ Library import test failed. Please check your installation.")
if __name__ == "__main__":
main() |