HRFCO Service

#!/usr/bin/env python3 """ Test intelligent water search functions """ import asyncio import json from smart_water_search import SmartWaterSearch async def test_all_functions(): """Test all intelligent search functions""" print("🧪 Testing Intelligent Water Search System") search = SmartWaterSearch() # Test 1: search_water_station_by_name print("\n🔍 Test 1: search_water_station_by_name('한강')") result1 = await search.search_stations_by_name("한강", limit=3) print(f"✅ Found {result1.get('found_stations', 0)} stations") if result1.get('stations'): print(f" First station: {result1['stations'][0]['name']}") # Test 2: get_water_info_by_location print("\n🔍 Test 2: get_water_info_by_location('서울 수위')") result2 = await search.get_water_info_by_location("서울 수위", limit=2) print(f"✅ Status: {result2.get('status', 'unknown')}") if result2.get('data', {}).get('stations'): print(f" Found: {len(result2['data']['stations'])} stations") # Test 3: recommend_nearby_stations print("\n🔍 Test 3: recommend_nearby_stations('부산')") result3 = await search.recommend_nearby_stations("부산", radius=30) print(f"✅ Recommendations: {len(result3.get('recommendations', []))}") # Test response sizes print("\n📊 Response Size Analysis:") print(f" Test 1 size: {len(json.dumps(result1, ensure_ascii=False))} bytes") print(f" Test 2 size: {len(json.dumps(result2, ensure_ascii=False))} bytes") print(f" Test 3 size: {len(json.dumps(result3, ensure_ascii=False))} bytes") print("\n🎉 All tests completed successfully!") if __name__ == "__main__": asyncio.run(test_all_functions())