modules/title_styler/test_cases.py

"""
Test script for specific problematic SVG cases
Tests the following cases that were reported to have issues:
1. tourism_cities_3_financial_centers.svg
2. tech_companies_1_rounded_label_big_title.svg
3. renewable_power_3_inline_contrast_triple.svg
4. life_expectancy_2_comic_bouncy_emphasis.svg
5. life_expectancy_1_stacked_badge_headline.svg
6. europe_cars_2_comic_bouncy_emphasis.svg
7. education_cost_1_comic_bold_announcement.svg
8. arctic_ice_decline_1_highlight_stripe.svg
"""

from modules.title_styler.infographic_title_generator import generate_title

# Problematic cases with their corresponding test data and templates
# Each case includes manually segmented title text to match template segments
PROBLEMATIC_CASES = [
    {
        "test_id": "tourism_cities",
        # financial_centers: 3 segments (black + primary bg + black)
        "title_segments": "Top 10 Most\nVisited Cities\nin 2024",
        "description": "Annual tourism statistics reveal which urban destinations attracted the most international visitors last year",
        "primary_color": "#E67E22",
        "background_color": "#FFFFFF",
        "template_name": "financial_centers",
        "output_file": "tourism_cities_3_financial_centers.svg"
    },
    {
        "test_id": "tech_companies",
        # rounded_label_big_title: 2 segments (label badge + headline)
        "title_segments": "TECH\nAmerica's Largest Technology Companies",
        "description": "Market capitalization comparison of leading tech firms headquartered in the United States",
        "primary_color": "#8E44AD",
        "background_color": "#F5F5F5",
        "template_name": "rounded_label_big_title",
        "output_file": "tech_companies_1_rounded_label_big_title.svg"
    },
    {
        "test_id": "renewable_power",
        # inline_contrast_triple: 3 inline segments
        "title_segments": "Renewable Energy\nShare\nof Total Electricity Generation",
        "description": "Tracking the percentage of power generated from solar, wind, hydro, and other renewable sources by country",
        "primary_color": "#16A085",
        "background_color": "#ECF0F1",
        "template_name": "inline_contrast_triple",
        "output_file": "renewable_power_3_inline_contrast_triple.svg"
    },
    {
        "test_id": "life_expectancy",
        # comic_bouncy_emphasis: 3 inline segments
        "title_segments": "Average\nLife Expectancy\nGains Since 1950",
        "description": "How longevity has improved across different regions over seven decades of global development",
        "primary_color": "#9B59B6",
        "background_color": "#FFFFFF",
        "template_name": "comic_bouncy_emphasis",
        "output_file": "life_expectancy_2_comic_bouncy_emphasis.svg"
    },
    {
        "test_id": "life_expectancy",
        # stacked_badge_headline: 3 segments (badge + headline + tagline)
        "title_segments": "LONGEVITY\nLife Expectancy Gains\nSince 1950",
        "description": "How longevity has improved across different regions over seven decades of global development",
        "primary_color": "#9B59B6",
        "background_color": "#FFFFFF",
        "template_name": "stacked_badge_headline",
        "output_file": "life_expectancy_1_stacked_badge_headline.svg"
    },
    {
        "test_id": "europe_cars",
        # comic_bouncy_emphasis: 3 inline segments
        "title_segments": "Europe's\nBest-Selling\nCar Brands in H1 2025",
        "description": "This visualization shows the best-selling car brands in Europe",
        "primary_color": "#1E5BBD",
        "background_color": "#E8E8E8",
        "template_name": "comic_bouncy_emphasis",
        "output_file": "europe_cars_2_comic_bouncy_emphasis.svg"
    },
    {
        "test_id": "education_cost",
        # comic_bold_announcement: 1 segment (single huge title)
        "title_segments": "The Rising Cost of Higher Education",
        "description": "How tuition fees have increased across different types of universities over the past two decades",
        "primary_color": "#C0392B",
        "background_color": "#F9F9F9",
        "template_name": "comic_bold_announcement",
        "output_file": "education_cost_1_comic_bold_announcement.svg"
    },
    {
        "test_id": "arctic_ice_decline",
        # highlight_stripe: 2 segments (lead-in + highlighted core)
        "title_segments": "The Dramatic Decline of\nArctic Sea Ice Coverage Over the Past Four Decades",
        "description": "Satellite imagery and climate data reveal unprecedented melting patterns affecting polar ecosystems and global weather",
        "primary_color": "#2C3E50",
        "background_color": "#FFFFFF",
        "template_name": "highlight_stripe",
        "output_file": "arctic_ice_decline_1_highlight_stripe.svg"
    }
]


# Comprehensive test cases for all 48 templates
FULL_TEMPLATE_TEST_CASES = [
    {
        "test_id": "three_line_emphasis",
        "title_segments": "Average\nLife Expectancy\nGains Since 1950",
        "description": "How longevity has improved across different regions over seven decades",
        "primary_color": "#E67E22",
        "background_color": "#FFFFFF",
        "template_name": "three_line_emphasis",
        "output_file": "three_line_emphasis_test.svg"
    },
    {
        "test_id": "two_line_hierarchy",
        "title_segments": "The Evolution of\nArtificial Intelligence",
        "description": "From early neural networks to modern large language models",
        "primary_color": "#F5F5F5",
        "background_color": "#8E44AD",
        "template_name": "two_line_hierarchy",
        "output_file": "two_line_hierarchy_test.svg"
    },
    {
        "test_id": "lithium_countries",
        "title_segments": "World's Largest\nLithium Producers",
        "description": "Countries leading the global lithium extraction industry",
        "primary_color": "#16A085",
        "background_color": "#ECF0F1",
        "template_name": "lithium_countries",
        "output_file": "lithium_countries_test.svg"
    },
    {
        "test_id": "european_adults",
        "title_segments": "Among European Adults\nSmoking Rates Decline",
        "description": "Public health campaigns and regulations drive behavioral change",
        "primary_color": "#9B59B6",
        "background_color": "#FFFFFF",
        "template_name": "european_adults",
        "output_file": "european_adults_test.svg"
    },
    {
        "test_id": "millionaires_green",
        "title_segments": "Number of\nMillionaires by Country",
        "description": "Wealth distribution across major economies worldwide",
        "primary_color": "#1E5BBD",
        "background_color": "#E8E8E8",
        "template_name": "millionaires_green",
        "output_file": "millionaires_green_test.svg"
    },
    {
        "test_id": "america_valuable",
        "title_segments": "America's\nMost Valuable\nTech\nCompanies",
        "description": "Market capitalization of leading technology firms",
        "primary_color": "#C0392B",
        "background_color": "#F9F9F9",
        "template_name": "america_valuable",
        "output_file": "america_valuable_test.svg"
    },
    {
        "test_id": "dentist_question",
        "title_segments": "How Many People\nVisit the Dentist Annually?",
        "description": "Dental care access and frequency across different demographics",
        "primary_color": "#2C3E50",
        "background_color": "#FFFFFF",
        "template_name": "dentist_question",
        "output_file": "dentist_question_test.svg"
    },
    {
        "test_id": "infant_mortality",
        "title_segments": "HEALTH\nInfant Mortality Rates by Region",
        "description": "Global health indicators tracking child survival rates",
        "primary_color": "#FF00FF",
        "background_color": "#333333",
        "template_name": "infant_mortality",
        "output_file": "infant_mortality_test.svg"
    },
    {
        "test_id": "financial_centers",
        "title_segments": "Top 10\nGlobal Financial Centers\nRanked by GDP",
        "description": "Economic powerhouses driving international finance",
        "primary_color": "#E74C3C",
        "background_color": "#F0F0F0",
        "template_name": "financial_centers",
        "output_file": "financial_centers_test.svg"
    },
    {
        "test_id": "innovative_countries",
        "title_segments": "Innovation Index\nMost Innovative Countries\nand Their Scores",
        "description": "R&D investment and patent output rankings",
        "primary_color": "#3498DB",
        "background_color": "#FFFFFF",
        "template_name": "innovative_countries",
        "output_file": "innovative_countries_test.svg"
    },
    {
        "test_id": "americas_wealth",
        "title_segments": "Total Household Wealth in North America",
        "description": "Combined assets of families across the United States and Canada",
        "primary_color": "#E67E22",
        "background_color": "#2C3E50",
        "template_name": "americas_wealth",
        "output_file": "americas_wealth_test.svg"
    },
    {
        "test_id": "china_dominance",
        "title_segments": "China's\nGrowing\nEconomic Influence",
        "description": "Trade relationships and investment patterns worldwide",
        "primary_color": "#16A085",
        "background_color": "#ECF0F1",
        "template_name": "china_dominance",
        "output_file": "china_dominance_test.svg"
    },
    {
        "test_id": "magazine_cover",
        "title_segments": "FEATURE STORY\nThe Future of Space Exploration\nNew Frontiers Ahead",
        "description": "Innovations in rocket technology and planetary missions",
        "primary_color": "#9B59B6",
        "background_color": "#FFFFFF",
        "template_name": "magazine_cover",
        "output_file": "magazine_cover_test.svg"
    },
    {
        "test_id": "newspaper_headline",
        "title_segments": "Climate Agreement Reached at International Summit",
        "description": "World leaders commit to ambitious carbon reduction targets",
        "primary_color": "#1E5BBD",
        "background_color": "#E8E8E8",
        "template_name": "newspaper_headline",
        "output_file": "newspaper_headline_test.svg"
    },
    {
        "test_id": "elegant_serif",
        "title_segments": "The Philosophy of\nSustainable Development",
        "description": "Balancing economic growth with environmental preservation",
        "primary_color": "#C0392B",
        "background_color": "#F9F9F9",
        "template_name": "elegant_serif",
        "output_file": "elegant_serif_test.svg"
    },
    {
        "test_id": "data_report",
        "title_segments": "Q3 2024\n+18% Revenue Growth",
        "description": "Quarterly financial performance exceeds analyst expectations",
        "primary_color": "#2C3E50",
        "background_color": "#FFFFFF",
        "template_name": "data_report",
        "output_file": "data_report_test.svg"
    },
    {
        "test_id": "corporate_brand",
        "title_segments": "INNOVATION THROUGH COLLABORATION",
        "description": "Our commitment to advancing technology for everyone",
        "primary_color": "#27AE60",
        "background_color": "#FAFAFA",
        "template_name": "corporate_brand",
        "output_file": "corporate_brand_test.svg"
    },
    {
        "test_id": "corporate_brand_desc_gap",
        "title_segments": "Full-Day Kindergarten Outperforms Half-Day on Reading Scores",
        "description": "ReadingScores for Half-Day and Full-Day kindergarten students (from 2015 to 2023) consistently indicate higher performance in Full-Day.",
        "primary_color": "#f8dd3d",
        "background_color": "#E2F1F6",
        "template_name": "corporate_brand",
        "output_file": "corporate_brand_desc_gap_test.svg"
    },
    {
        "test_id": "academic_paper",
        "title_segments": "Machine Learning Applications\nin Climate Science",
        "description": "Improving weather prediction models with AI techniques",
        "primary_color": "#E74C3C",
        "background_color": "#F0F0F0",
        "template_name": "academic_paper",
        "output_file": "academic_paper_test.svg"
    },
    {
        "test_id": "tech_startup",
        "title_segments": "Move Fast\nBreak Barriers",
        "description": "Our mission to disrupt traditional industries",
        "primary_color": "#3498DB",
        "background_color": "#FFFFFF",
        "template_name": "tech_startup",
        "output_file": "tech_startup_test.svg"
    },
    {
        "test_id": "poster_impact",
        "title_segments": "JOIN THE REVOLUTION\nTogether We Rise",
        "description": "Community-driven social change movement",
        "primary_color": "#E67E22",
        "background_color": "#FFFFFF",
        "template_name": "poster_impact",
        "output_file": "poster_impact_test.svg"
    },
    {
        "test_id": "vintage_style",
        "title_segments": "Est. 1887\nHeritage Coffee Co.\nAuthentic Roasters",
        "description": "Traditional craftsmanship meets modern taste",
        "primary_color": "#8E44AD",
        "background_color": "#F5F5F5",
        "template_name": "vintage_style",
        "output_file": "vintage_style_test.svg"
    },
    {
        "test_id": "europe_brands",
        "title_segments": "EUROPE\nBest-Selling Car Brands in 2024",
        "description": "Consumer preferences across major European markets",
        "primary_color": "#16A085",
        "background_color": "#ECF0F1",
        "template_name": "europe_brands",
        "output_file": "europe_brands_test.svg"
    },
    {
        "test_id": "international_students",
        "title_segments": "Asia's\nTop Universities\nfor International Students",
        "description": "Rankings based on diversity and academic excellence",
        "primary_color": "#9B59B6",
        "background_color": "#FFFFFF",
        "template_name": "international_students",
        "output_file": "international_students_test.svg"
    },
    {
        "test_id": "kpi_number_pill",
        "title_segments": "QUARTERLY REVENUE\n$2.4 Billion",
        "description": "Record-breaking sales driven by product innovation",
        "primary_color": "#1E5BBD",
        "background_color": "#E8E8E8",
        "template_name": "kpi_number_pill",
        "output_file": "kpi_number_pill_test.svg"
    },
    {
        "test_id": "timeline_year_span",
        "title_segments": "Looking Back at\n1990-2020\nTechnology Evolution",
        "description": "Three decades of digital transformation",
        "primary_color": "#C0392B",
        "background_color": "#F9F9F9",
        "template_name": "timeline_year_span",
        "output_file": "timeline_year_span_test.svg"
    },
    {
        "test_id": "stacked_badge_headline",
        "title_segments": "SUSTAINABILITY\nCarbon Emissions Reduction\nSince 2010",
        "description": "Progress toward net-zero goals across industries",
        "primary_color": "#2C3E50",
        "background_color": "#FFFFFF",
        "template_name": "stacked_badge_headline",
        "output_file": "stacked_badge_headline_test.svg"
    },
    {
        "test_id": "right_align_elegant_duo",
        "title_segments": "Across Major Cities\nUrban Population Density",
        "description": "Comparing metropolitan areas worldwide",
        "primary_color": "#27AE60",
        "background_color": "#FAFAFA",
        "template_name": "right_align_elegant_duo",
        "output_file": "right_align_elegant_duo_test.svg"
    },
    {
        "test_id": "inline_contrast_triple",
        "title_segments": "The Rise of\nElectric Vehicles\nin Global Markets",
        "description": "Adoption rates accelerate with policy support",
        "primary_color": "#E74C3C",
        "background_color": "#F0F0F0",
        "template_name": "inline_contrast_triple",
        "output_file": "inline_contrast_triple_test.svg"
    },
    {
        "test_id": "split_colon_emphasis",
        "title_segments": "INNOVATION REPORT:\nBreakthrough Technologies of 2024",
        "description": "Scientific advances transforming daily life",
        "primary_color": "#3498DB",
        "background_color": "#FFFFFF",
        "template_name": "split_colon_emphasis",
        "output_file": "split_colon_emphasis_test.svg"
    },
    {
        "test_id": "minimal_right_caption",
        "title_segments": "Global Trade Volumes",
        "description": "International commerce statistics",
        "primary_color": "#E67E22",
        "background_color": "#FFFFFF",
        "template_name": "minimal_right_caption",
        "output_file": "minimal_right_caption_test.svg"
    },
    {
        "test_id": "two_tone_upper",
        "title_segments": "MARKET ANALYSIS\nCRYPTOCURRENCY TRENDS",
        "description": "Digital asset valuations and regulatory developments",
        "primary_color": "#8E44AD",
        "background_color": "#F5F5F5",
        "template_name": "two_tone_upper",
        "output_file": "two_tone_upper_test.svg"
    },
    {
        "test_id": "rounded_label_big_title",
        "title_segments": "TECH\nArtificial Intelligence Market Growth",
        "description": "Industry valuation surpasses $500 billion",
        "primary_color": "#16A085",
        "background_color": "#ECF0F1",
        "template_name": "rounded_label_big_title",
        "output_file": "rounded_label_big_title_test.svg"
    },
    {
        "test_id": "underlined_statement",
        "title_segments": "The Future Belongs to Renewable Energy",
        "description": "Sustainable power generation becomes economically dominant",
        "primary_color": "#9B59B6",
        "background_color": "#FFFFFF",
        "template_name": "underlined_statement",
        "output_file": "underlined_statement_test.svg"
    },
    {
        "test_id": "soft_shadow_duo",
        "title_segments": "Throughout History\nRevolutionary Inventions",
        "description": "Innovations that changed human civilization",
        "primary_color": "#1E5BBD",
        "background_color": "#E8E8E8",
        "template_name": "soft_shadow_duo",
        "output_file": "soft_shadow_duo_test.svg"
    },
    {
        "test_id": "comic_fun_title",
        "title_segments": "Super Amazing Facts About Space!",
        "description": "Mind-blowing discoveries about our universe",
        "primary_color": "#C0392B",
        "background_color": "#F9F9F9",
        "template_name": "comic_fun_title",
        "output_file": "comic_fun_title_test.svg"
    },
    {
        "test_id": "comic_bubble_speech",
        "title_segments": "Guess What?\nPandas Eat 40kg Bamboo Daily!",
        "description": "Incredible animal eating habits revealed",
        "primary_color": "#2C3E50",
        "background_color": "#FFFFFF",
        "template_name": "comic_bubble_speech",
        "output_file": "comic_bubble_speech_test.svg"
    },
    {
        "test_id": "comic_bouncy_emphasis",
        "title_segments": "World's\nCoolest\nGadgets of 2024",
        "description": "The most exciting tech toys released this year",
        "primary_color": "#E74C3C",
        "background_color": "#F0F0F0",
        "template_name": "comic_bouncy_emphasis",
        "output_file": "comic_bouncy_emphasis_test.svg"
    },
    {
        "test_id": "comic_playful_label",
        "title_segments": "FUN FACTS\nOceans Cover 71% of Earth!",
        "description": "Amazing things you never knew about our planet",
        "primary_color": "#3498DB",
        "background_color": "#FFFFFF",
        "template_name": "comic_playful_label",
        "output_file": "comic_playful_label_test.svg"
    },
    {
        "test_id": "comic_question_fun",
        "title_segments": "Did You Know?\nBananas Are Berries!",
        "description": "Surprising botanical classification facts",
        "primary_color": "#E67E22",
        "background_color": "#FFFFFF",
        "template_name": "comic_question_fun",
        "output_file": "comic_question_fun_test.svg"
    },
    {
        "test_id": "comic_bold_announcement",
        "title_segments": "The Most Incredible Animal Superpowers!",
        "description": "Nature's amazing adaptations and abilities",
        "primary_color": "#8E44AD",
        "background_color": "#F5F5F5",
        "template_name": "comic_bold_announcement",
        "output_file": "comic_bold_announcement_test.svg"
    },
    {
        "test_id": "comic_wavy_text",
        "title_segments": "Get Ready for\nAdventure Time!",
        "description": "Exciting outdoor activities for families",
        "primary_color": "#9B59B6",
        "background_color": "#FFFFFF",
        "template_name": "comic_wavy_text",
        "output_file": "comic_wavy_text_test.svg"
    },
    {
        "test_id": "comic_mega_title",
        "title_segments": "Ultimate Gaming Champions Tournament!",
        "description": "Epic esports competition draws global audience",
        "primary_color": "#2C3E50",
        "background_color": "#FFFFFF",
        "template_name": "comic_mega_title",
        "output_file": "comic_mega_title_test.svg"
    },
    {
        "test_id": "comic_sticker_style",
        "title_segments": "AWESOME\nSpace Mission Success!",
        "description": "Satellite deployment achieves perfect orbit",
        "primary_color": "#27AE60",
        "background_color": "#FAFAFA",
        "template_name": "comic_sticker_style",
        "output_file": "comic_sticker_style_test.svg"
    },
]


def test_single_case(test_case):
    """
    Test a single problematic case
    
    Args:
        test_case: Dictionary containing title_segments, description, colors, template_name, etc.
    
    Returns:
        Result of the generation
    """
    print("\n" + "=" * 70)
    print(f"Testing: {test_case['output_file']}")
    print("=" * 70)
    print(f"Title segments: {test_case['title_segments']}")
    if len(test_case['description']) > 60:
        print(f"Description: {test_case['description'][:60]}...")
    else:
        print(f"Description: {test_case['description']}")
    print(f"Template: {test_case['template_name']}")
    print(f"Primary Color: {test_case['primary_color']}")
    print(f"Background Color: {test_case['background_color']}")
    print("=" * 70)
    
    try:
        results = generate_title(
            title=test_case['title_segments'],  # Pass list of segments
            description=test_case['description'],
            primary_color=test_case['primary_color'],
            background_color=test_case['background_color'],
            template_name=test_case['template_name'],
            max_width=900,
            use_llm=False,  # Use specified template directly
            save_to=f"output/{test_case['test_id']}"
        )
        
        if results:
            result = results[0]
            print(f"✅ Successfully generated:")
            print(f"   Template: {result['template_name']}")
            print(f"   Size: {result['width']:.0f}px × {result['height']:.0f}px")
            print(f"   Saved as: output/{test_case['output_file']}")
            return True
        else:
            print(f"❌ Failed to generate")
            return False
            
    except Exception as e:
        print(f"❌ Error: {e}")
        import traceback
        traceback.print_exc()
        return False


if __name__ == "__main__":
    import sys
    
    # Determine which test suite to run
    if len(sys.argv) > 1 and sys.argv[1] == '--all':
        test_suite = FULL_TEMPLATE_TEST_CASES
        suite_name = "ALL TEMPLATES"
    elif len(sys.argv) > 1 and sys.argv[1] == '--problematic':
        test_suite = PROBLEMATIC_CASES
        suite_name = "PROBLEMATIC CASES"
    else:
        # Default: run full template tests
        test_suite = FULL_TEMPLATE_TEST_CASES
        suite_name = "ALL TEMPLATES (DEFAULT)"
        print("💡 Tip: Use '--problematic' to run only problematic cases")
        print("💡 Tip: Use '--all' to explicitly run all template tests")
        print()
    
    print("=" * 70)
    print(f"TESTING {suite_name}")
    print(f"Total cases to test: {len(test_suite)}")
    print("=" * 70)
    
    success_count = 0
    failed_count = 0
    
    for i, test_case in enumerate(test_suite, 1):
        print(f"\n[{i}/{len(test_suite)}]")
        
        if test_single_case(test_case):
            success_count += 1
        else:
            failed_count += 1
            print(f"Failed: {test_case['output_file']}")
    
    # Summary
    print("\n" + "=" * 70)
    print("TEST SUMMARY")
    print("=" * 70)
    print(f"Total test cases: {len(test_suite)}")
    print(f"✅ Successful: {success_count}")
    print(f"❌ Failed: {failed_count}")
    print("=" * 70)
    
    if failed_count > 0:
        print("\n⚠️  Some tests failed. Please review the errors above.")
    else:
        print("\n✅ All tests passed!")
    
    print("=" * 70)