import sys
import os

# Přidání adresáře src do sys.path, aby bylo možné importovat moduly
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'src'))

import holidays

def test_holidays():
    """Test funkcionality modulu holidays.py."""
    print("Načítám státní svátky...")
    holiday_data = holidays.load_holidays()
    assert "2025" in holiday_data, "Data pro rok 2025 nebyla nalezena"
    print("Test načtení svátků: OK")
    
    # Test, zda 1. ledna 2025 je svátek
    assert holidays.is_holiday("01.01.2025"), "1. ledna 2025 by měl být svátek"
    print("Test svátku 1.1.2025: OK")
    
    # Test, zda 15. ledna 2025 není svátek
    assert not holidays.is_holiday("15.01.2025"), "15. ledna 2025 by neměl být svátek"
    print("Test nesvátku 15.1.2025: OK")
    
    # Test víkendu
    assert holidays.is_weekend("04.01.2025"), "4. ledna 2025 je sobota"
    print("Test víkendu 4.1.2025 (sobota): OK")
    
    assert not holidays.is_weekend("06.01.2025"), "6. ledna 2025 je pondělí"
    print("Test pracovního dne 6.1.2025: OK")
    
    # Test předchozího pracovního dne
    # 1. ledna 2025 je svátek (středa), předchozí pracovní den by měl být 31.12.2024 (úterý)
    prev_workday = holidays.get_previous_working_day("01.01.2025")
    assert prev_workday == "31.12.2024", f"Očekáváno 31.12.2024, ale vráceno {prev_workday}"
    print("Test předchozího pracovního dne před 1.1.2025: OK")
    
    print("\nVšechny testy modulu holidays.py byly úspěšné!")

if __name__ == "__main__":
    test_holidays()