simple_test.py

#!/usr/bin/env python3
"""
Simple test for SEC EDGAR integration
"""

from gradio_stock_dashboard import CompanyInfo, SECEdgarAPI

def test_basic_functionality():
    print("Testing basic SEC integration...")
    
    # Test CompanyInfo
    ci = CompanyInfo('AAPL')
    print(f"Created CompanyInfo for AAPL")
    
    # Test fallback CIK lookup
    cik = ci._get_cik_from_fallback()
    print(f"CIK: {cik}")
    print(f"Company Name: {ci.company_name}")
    
    if cik:
        print("✓ Fallback CIK lookup successful")
        
        # Test SEC API
        sec_api = SECEdgarAPI()
        print("✓ SEC API created")
        
        # Test company submissions
        try:
            submissions = sec_api.get_company_submissions(cik)
            if submissions:
                print("✓ Company submissions retrieved")
                print(f"  - Response keys: {list(submissions.keys())}")
            else:
                print("⚠ No company submissions found")
        except Exception as e:
            print(f"✗ Error getting submissions: {e}")
            
    else:
        print("✗ Fallback CIK lookup failed")

if __name__ == "__main__":
    test_basic_functionality()