Spaces:
Sleeping
Sleeping
File size: 1,425 Bytes
4c7be84 | 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 |
import os
from report_generator import ReportGenerator
def test_pdf_generation():
generator = ReportGenerator()
deals = [
{
"name": "Test Hotel 1",
"location": "Berlin",
"price_per_night": 100,
"rating": 4.5,
"reviews": 120,
"source": "booking",
"pet_friendly": True,
"url": "http://example.com/1"
},
{
"name": "Test Apartment 2",
"location": "Paris",
"price_per_night": 200,
"rating": 4.8,
"reviews": 50,
"source": "airbnb",
"pet_friendly": True,
"url": "http://example.com/2"
}
]
search_params = {
"cities": ["Berlin", "Paris"],
"checkin": "2026-06-01",
"checkout": "2026-06-07",
"nights": 6,
"group_size": 2,
"pets": 1,
"budget_range": "EUR 50-250"
}
filename = "test_report.pdf"
if os.path.exists(filename):
os.remove(filename)
success = generator.generate_report(deals, search_params, filename)
assert success
assert os.path.exists(filename)
# Check file size is not empty
assert os.path.getsize(filename) > 0
# Cleanup
os.remove(filename)
print("PDF generation test passed!")
if __name__ == "__main__":
test_pdf_generation()
|