tests/test_fredapi_library.py

#!/usr/bin/env python3
"""
Test script to verify FRED API key functionality
"""

import os
import sys

from fredapi import Fred

sys.path.append(os.path.join(os.path.dirname(__file__), ".."))

from config.settings import FRED_API_KEY


def test_api_connection():
    """Test the FRED API connection with the provided key."""
    print("Testing FRED API connection...")

    try:
        # Initialize FRED client
        fred = Fred(api_key=FRED_API_KEY)

        # Test with a simple series (GDP)
        print("Fetching GDP data as a test...")
        gdp_data = fred.get_series("GDP", start="2023-01-01", end="2023-12-31")

        if not gdp_data.empty:
            print("✓ API connection successful!")
            print(f"✓ Retrieved {len(gdp_data)} GDP observations")
            print(f"✓ Latest GDP value: ${gdp_data.iloc[-1]:,.2f} billion")
            print(f"✓ Date range: {gdp_data.index.min()} to {gdp_data.index.max()}")
            return True
        else:
            print("✗ No data retrieved")
            return False

    except Exception as e:
        print(f"✗ API connection failed: {e}")
        return False


def test_series_info():
    """Test getting series information."""
    print("\nTesting series information retrieval...")

    try:
        fred = Fred(api_key=FRED_API_KEY)

        # Test getting info for GDP
        series_info = fred.get_series_info("GDP")

        print("✓ Series information retrieved successfully!")
        print(f"  Title: {series_info.title}")
        print(f"  Units: {series_info.units}")
        print(f"  Frequency: {series_info.frequency}")
        print(f"  Last Updated: {series_info.last_updated}")

        return True

    except Exception as e:
        print(f"✗ Failed to get series info: {e}")
        return False


def main():
    """Run API tests."""
    print("FRED API Key Test")
    print("=" * 30)
    print(f"API Key: {FRED_API_KEY[:8]}...")
    print()

    # Test connection
    connection_ok = test_api_connection()

    # Test series info
    info_ok = test_series_info()

    print("\n" + "=" * 30)
    if connection_ok and info_ok:
        print("✓ All tests passed! Your API key is working correctly.")
        print("You can now use the FRED data collector tool.")
    else:
        print("✗ Some tests failed. Please check your API key.")

    return connection_ok and info_ok


if __name__ == "__main__":
    main()