"""
Test script to verify retrosheet_data.py works with the new transformer.
"""

import sys
import logging
from retrosheet_data import get_base_batting_df, get_base_pitching_df, RETRO_FILE_PATH, EVENTS_FILENAME

# Set up logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)

print(f"\n{'='*60}")
print(f"Testing Retrosheet Data Integration")
print(f"{'='*60}\n")

file_path = f'{RETRO_FILE_PATH}{EVENTS_FILENAME}'
print(f"Loading from: {file_path}")

# Test date range (full 2005 season)
start_date = 20050403  # Opening Day 2005
end_date = 20051003    # End of 2005 regular season

print(f"\nTest 1: Loading Batting Data")
print(f"Date range: {start_date} to {end_date}")

try:
    all_plays_batting, batting_stats = get_base_batting_df(file_path, start_date, end_date)
    print(f"✓ Success!")
    print(f"  - Total plays loaded: {len(all_plays_batting)}")
    print(f"  - Qualified batters: {len(batting_stats)}")
    print(f"\nSample batting stats:")
    print(batting_stats.head(3))
except Exception as e:
    print(f"✗ Failed: {e}")
    sys.exit(1)

print(f"\n{'-'*60}\n")
print(f"Test 2: Loading Pitching Data")

try:
    all_plays_pitching, pitching_stats = get_base_pitching_df(file_path, start_date, end_date)
    print(f"✓ Success!")
    print(f"  - Total plays loaded: {len(all_plays_pitching)}")
    print(f"  - Qualified pitchers: {len(pitching_stats)}")
    print(f"\nSample pitching stats:")
    print(pitching_stats.head(3))
except Exception as e:
    print(f"✗ Failed: {e}")
    sys.exit(1)

print(f"\n{'='*60}")
print(f"All tests passed! ✓")
print(f"{'='*60}\n")