#!/usr/bin/env python3
"""
Utility Functions Tests
=======================

Author: RSK World (https://rskworld.in)
Year: 2026

Description: Unit tests for utility functions
"""

import pytest
import sys
import os

sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

from utils.weather_utils import (
    format_temperature,
    format_wind_speed,
    get_weather_emoji,
    validate_city_name,
    extract_city_from_query
)

def test_format_temperature():
    """Test temperature formatting."""
    assert format_temperature(20, 'metric') == '20°C'
    assert format_temperature(20, 'imperial') == '68°F'
    assert format_temperature(20, 'kelvin') == '293K'

def test_format_wind_speed():
    """Test wind speed formatting."""
    assert 'm/s' in format_wind_speed(10, 'metric')
    assert 'mph' in format_wind_speed(10, 'imperial')

def test_get_weather_emoji():
    """Test weather emoji selection."""
    assert get_weather_emoji('clear sky') == '☀️'
    assert get_weather_emoji('cloudy') == '☁️'
    assert get_weather_emoji('rain') == '🌧️'
    assert get_weather_emoji('snow') == '❄️'

def test_validate_city_name():
    """Test city name validation."""
    assert validate_city_name('London') == True
    assert validate_city_name('New York') == True
    assert validate_city_name('') == False
    assert validate_city_name('A') == False
    assert validate_city_name('123') == False

def test_extract_city_from_query():
    """Test city extraction from queries."""
    assert extract_city_from_query('weather in London') == 'London'
    assert extract_city_from_query('What is the weather in New York?') == 'New York'
    assert extract_city_from_query('forecast for Paris') == 'Paris'