resumebuilder / test_pdf_generation.py
sakthi07's picture
pushing to hugging face
ab028ac
#!/usr/bin/env python3
"""
Test script for PDF generation using xhtml2pdf.
"""
from pdf_generator import create_pdf_resume
from utils import log_info
# Test data
test_data = {
'name': 'John Doe',
'email': 'john.doe@example.com',
'phone': '+1 (555) 123-4567',
'linkedin': 'johndoe',
'github': 'johndoe',
'website': 'https://johndoe.com',
'summary': 'Experienced software developer with 5+ years of expertise in full-stack development, cloud technologies, and team leadership. Passionate about building scalable solutions and mentoring junior developers.',
'work_experience': [
{
'title': 'Senior Software Engineer',
'organization': 'Tech Corp',
'start_date': 'Jan 2022',
'end_date': 'Present',
'remarks': 'Leading development of microservices architecture serving 1M+ users. Mentoring team of 5 developers.'
},
{
'title': 'Software Developer',
'organization': 'StartupXYZ',
'start_date': 'Jun 2020',
'end_date': 'Dec 2021',
'remarks': 'Developed RESTful APIs and implemented CI/CD pipelines. Reduced deployment time by 60%.'
}
],
'projects': [
{
'title': 'E-commerce Platform',
'organization': 'Personal Project',
'start_date': 'Mar 2023',
'end_date': 'May 2023',
'remarks': 'Built full-stack e-commerce platform using React, Node.js, and PostgreSQL. Implemented payment processing and inventory management.'
}
],
'education': [
{
'title': 'Bachelor of Science in Computer Science',
'organization': 'University of Technology',
'start_date': 'Sep 2016',
'end_date': 'May 2020',
'remarks': 'Graduated Magna Cum Laude. President of Computer Science Club.'
}
],
'skills': 'Python, JavaScript, React, Node.js, PostgreSQL, MongoDB, AWS, Docker, Kubernetes, Git, CI/CD',
'achievements': 'Employee of the Year 2023, Best Project Award at University Hackathon 2019',
'sections_order': ['work_experience', 'projects', 'education', 'skills', 'achievements']
}
def test_pdf_generation():
"""Test PDF generation for both templates."""
try:
log_info("Testing PDF generation...")
# Test standard template
log_info("Generating standard template PDF...")
pdf_bytes_standard = create_pdf_resume(test_data, "standard")
if pdf_bytes_standard:
with open('test_resume_standard.pdf', 'wb') as f:
f.write(pdf_bytes_standard)
log_info("Standard PDF generated successfully: test_resume_standard.pdf")
else:
log_info("Failed to generate standard PDF")
# Test modern template
log_info("Generating modern template PDF...")
pdf_bytes_modern = create_pdf_resume(test_data, "modern")
if pdf_bytes_modern:
with open('test_resume_modern.pdf', 'wb') as f:
f.write(pdf_bytes_modern)
log_info("Modern PDF generated successfully: test_resume_modern.pdf")
else:
log_info("Failed to generate modern PDF")
except Exception as e:
log_error(f"Test failed: {str(e)}", e)
if __name__ == "__main__":
test_pdf_generation()