test-sdk.py

"""
Test file for ZeroCarbon Python SDK

Usage:
1. Install SDK: pip install -e .
2. Set API key: $env:ZEROCARBON_API_KEY = "your_key_here"  (PowerShell)
3. Run: python test-sdk.py
"""

import os
from zerocarbon import ZeroCarbon

API_KEY = os.environ.get('ZEROCARBON_API_KEY', 'YOUR_API_KEY_HERE')

client = ZeroCarbon(
    api_key=API_KEY,
    base_url='http://localhost:3000/api/v1',  # Local development
    test_mode=True  # Set to False for production
)

def test_sdk():
    print('🧪 Testing ZeroCarbon Python SDK...\n')
    
    try:
        # Test 1: Flight Emissions Calculation
        print('1️⃣ Testing flight emissions calculation...')
        flight_result = client.calculate.flight(
            origin='DEL',
            destination='BOM',
            cabin_class='economy',
            passengers=1,
            round_trip=False
        )
        print('✅ Flight emissions:', flight_result)
        print('')
        
        # Test 2: Electricity Emissions Calculation
        print('2️⃣ Testing electricity emissions calculation...')
        electricity_result = client.calculate.electricity(
            kwh=1000,
            country_code='IN',
            state='Maharashtra'
        )
        print('✅ Electricity emissions:', electricity_result)
        print('')
        
        # Test 3: Get Offset Recommendations
        print('3️⃣ Testing offset recommendations...')
        offsets_result = client.offsets.get_recommendations(
            emissions_tonnes=5.0,
            budget_usd=100,
            location='India'
        )
        print('✅ Offset recommendations:', offsets_result)
        print('')
        
        print('🎉 All tests passed!')
        
    except Exception as error:
        print(f'❌ Error: {str(error)}')
        print('\n💡 Make sure:')
        print('   1. Your API key is set correctly')
        print('   2. Your API endpoints are deployed and working')
        print('   3. You have proper network connection')
        print('   4. SDK is installed: pip install -e .')

if __name__ == '__main__':
    test_sdk()