tests/test_fred_api.py

#!/usr/bin/env python3
"""
Simple FRED API test
"""

import requests
import sys
import os
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))

from config.settings import FRED_API_KEY

def test_fred_api_direct():
    """Test FRED API directly using requests."""
    print("Testing FRED API directly...")
    
    # Test URL for GDP series
    url = f"https://api.stlouisfed.org/fred/series/observations"
    params = {
        'series_id': 'GDP',
        'api_key': FRED_API_KEY,
        'file_type': 'json',
        'start_date': '2023-01-01',
        'end_date': '2023-12-31'
    }
    
    try:
        response = requests.get(url, params=params)
        
        if response.status_code == 200:
            data = response.json()
            observations = data.get('observations', [])
            
            if observations:
                print("✓ API connection successful!")
                print(f"✓ Retrieved {len(observations)} GDP observations")
                
                # Get the latest observation
                latest = observations[-1]
                print(f"✓ Latest GDP value: ${float(latest['value']):,.2f} billion")
                print(f"✓ Date: {latest['date']}")
                return True
            else:
                print("✗ No observations found")
                return False
        else:
            print(f"✗ API request failed with status code: {response.status_code}")
            print(f"Response: {response.text}")
            return False
            
    except Exception as e:
        print(f"✗ API connection failed: {e}")
        return False

def test_series_search():
    """Test searching for series."""
    print("\nTesting series search...")
    
    url = "https://api.stlouisfed.org/fred/series/search"
    params = {
        'search_text': 'GDP',
        'api_key': FRED_API_KEY,
        'file_type': 'json'
    }
    
    try:
        response = requests.get(url, params=params)
        
        if response.status_code == 200:
            data = response.json()
            series = data.get('seriess', [])
            
            if series:
                print("✓ Series search successful!")
                print(f"✓ Found {len(series)} series matching 'GDP'")
                
                # Show first few results
                for i, s in enumerate(series[:3]):
                    print(f"  {i+1}. {s['id']}: {s['title']}")
                return True
            else:
                print("✗ No series found")
                return False
        else:
            print(f"✗ Search request failed: {response.status_code}")
            return False
            
    except Exception as e:
        print(f"✗ Search failed: {e}")
        return False

def main():
    """Run simple API tests."""
    print("Simple FRED API Test")
    print("=" * 30)
    print(f"API Key: {FRED_API_KEY[:8]}...")
    print()
    
    # Test direct API access
    api_ok = test_fred_api_direct()
    
    # Test series search
    search_ok = test_series_search()
    
    print("\n" + "=" * 30)
    if api_ok and search_ok:
        print("✓ All tests passed! Your API key is working correctly.")
        print("The issue is with the fredapi library, not your API key.")
    else:
        print("✗ Some tests failed. Please check your API key.")
    
    return api_ok and search_ok

if __name__ == "__main__":
    main()