test_server.cpython-312-pytest-7.4.4.pyc•9.79 kB
�
���h� � �l � d dl Zd dlmc mZ d dlZd dlmZ d dl m
Z
ee
� Z d� Z d� Z
d� Zd� Zd� Zy) � N)�
TestClient)�appc � � t j ddddddd�d�� � } | j }d
}||k( }|s�t j d |fd ||f� d
t j � v st j | � rt j | � nd
t j |� t j |� d�z }dd|iz }t t j |� � �d x}x}}d}| j } |� }||v }|s�t j d|fd||f� t j |� d
t j � v st j | � rt j | � nd
t j |� t j |� d�z }dd|iz } t t j | � � �d x}x}x}}y )N�/mcp�get_weather_forecast�^K�=[D@���Mb�R�� F)�latitude� longitude�days�include_hourly��tool�
parameters��json�� ��==�z3%(py2)s
{%(py2)s = %(py0)s.status_code
} == %(py5)s�response��py0�py2�py5�assert %(py7)s�py7�daily��in�zC%(py1)s in %(py7)s
{%(py7)s = %(py5)s
{%(py5)s = %(py3)s.json
}()
}��py1�py3r r �assert %(py9)s�py9� �client�post� status_code�
@pytest_ar�_call_reprcompare� @py_builtins�locals�_should_repr_global_name� _saferepr�AssertionError�_format_explanationr �
r � @py_assert1� @py_assert4� @py_assert3� @py_format6� @py_format8� @py_assert0� @py_assert6� @py_assert2� @py_format10s
�/app/test_server.py�test_get_weather_forecastr? s� � ��{�{�6�&� �!�� #�
�)�{� �H� '�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%� c � � t j ddddd�d��� } | j }d}||k( }|s�t j d |fd
||f� d t j � v st j | � rt j | � nd t j |� t j |� d �z }d
d|iz }t t j |� � �d x}x}}d}| j } |� }||v }|s�t j d|fd||f� t j |� d t j � v st j | � rt j | � nd t j |� t j |� d�z }dd|iz } t t j | � � �d x}x}x}}y )Nr �get_current_weather���(\�BJ@�R� ���*@)r r r r r r r r r r r �current_weatherr r" r# r& r' r( r4 s
r> �test_get_current_weatherrF s� � ��{�{�6�%� � �
�)�{� �H� '�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/�/r@ c � � t j ddddddd�d�� � } | j }d
}||k( }|s�t j d |fd ||f� d
t j � v st j | � rt j | � nd
t j |� t j |� d�z }dd|iz }t t j |� � �d x}x}}d}| j } |� }||v }|s�t j d|fd||f� t j |� d
t j � v st j | � rt j | � nd
t j |� t j |� d�z }dd|iz } t t j | � � �d x}x}x}}y )Nr �get_historical_weatherrC rD z
2023-01-01z
2023-01-02)r r �
start_date�end_dater r r r r r r r r r r r" r# r& r' r( r4 s
r> �test_get_historical_weatherrK s� � ��{�{�6�(� � �&�$�
�)�{� �H� '�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%r@ c � � t j ddi d��� } | j }d}||k( }|s�t j d|fd||f� dt j � v st j | � rt j | � ndt j |� t j |� d �z }d
d |iz }t t j |� � �d x}x}}| j } |� }d d
i}||k( }|s�t j d|fd||f� dt j � v st j | � rt j | � ndt j |� t j |� t j |� d�z }dd|iz }t t j |� � �d x}x}x}}y )Nr � invalid_toolr r i� r r r r r r �detailzTool not found)zC%(py4)s
{%(py4)s = %(py2)s
{%(py2)s = %(py0)s.json
}()
} == %(py7)s)r r �py4r r& r' r( ) r r5 r6 r7 r8 r9 r; � @py_assert5r= s r> �test_tool_not_foundrQ , s� � ��{�{�6� �r�(R�{�S�H�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:�:r@ c �: ��� G d� d� � G �fd�d� ��fd�}| j d|� | j dd� t j d d
d d d
dgd�d��� }|j }d}||k( }|s�t j d|fd||f� dt j � v st j |� rt j |� ndt j |� t j |� d�z }dd|iz }t t j |� � �d x}x}}d}|j } |� } || v }
|
s�t j d|
fd|| f� t j |� dt j � v st j |� rt j |� ndt j |� t j | � d �z }d d |iz } t t j | � � �d x}x}
x}} |j � d }d }||k( }
|
slt j d|
fd ||f� t j |� t j |� d!�z } d"d#| iz }
t t j |
� � �d x}x}
}y )$Nc � � e Zd Zd� Zy)�.test_predict_weather_alert.<locals>.MockChoicec � � | | _ || _ y �N)�message�content)�selfrX s r> �__init__z7test_predict_weather_alert.<locals>.MockChoice.__init__3 s � � �D�L�"�D�Lr@ N��__name__�
__module__� __qualname__rZ � r@ r> �
MockChoicerT 2 s � � #r@ r` c � �� e Zd Z� fd�Zy)�2test_predict_weather_alert.<locals>.MockCompletionc �" �� �|� g| _ y rV )�choices)rY rX r` s �r> rZ z;test_predict_weather_alert.<locals>.MockCompletion.__init__8 s �� � &�w� /�0�D�Lr@ Nr[ )r` s �r> �MockCompletionrb 7 s �� � 1r@ re c � �� �d� S )N�!Sunny with a chance of meatballs.r_ )�args�kwargsre s �r> � mock_createz/test_predict_weather_alert.<locals>.mock_create; s �� � � A�B�Br@ z4openai.resources.chat.completions.Completions.create�OPENAI_API_KEY�test_keyr �predict_weather_alertr r �wheat�corn)r r �cropsr r r r r r r r r �alertr r" r# r& r' rg )z%(py1)s == %(py4)s)r$ rO zassert %(py6)s�py6)�setattr�setenvr) r* r+ r, r- r. r/ r0 r1 r2 r3 r )� monkeypatchrj r r5 r6 r7 r8 r9 r: r; r< r= � @py_format5� @py_format7r` re s @@r> �test_predict_weather_alertrx 1 sI �� �#� #�
1� 1�C� ���N�P[�\����'��4��{�{�6�'� �!� �v�&�
�)�{� �H� '�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�&�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�%�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�Jr@ )�builtinsr. �_pytest.assertion.rewrite� assertion�rewriter, �pytest�fastapi.testclientr � src.mcp_weather_server.serverr r) r? rF rK rQ rx r_ r@ r> �<module>r� s5 �� � �
� )� -� �C� �� &� 0� &�;�
Kr@