
    Xi                         d dl Zd dlmc mZ d dlZd dlZej                  j                  d d       d dl
mZmZmZ ej                  d        Zej                   j"                  d        Zy)    Nz!/home/jang/Projects/ai-researcher)SkillExecutorSkillResultSkillContextc                      t               S )N)r        C/home/jang/Projects/ai-researcher/tests/unit/test_skill_executor.pyexecutorr
      s
    ?r   c                   K    G d d      }t        ddd      }| j                   |       di |       d	{   }|j                  }d
di}||k(  }|st        j                  d|fd||f      dt        j                         v st        j                  |      rt        j                  |      ndt        j                  |      t        j                  |      dz  }dd|iz  }t        t        j                  |            d	x}x}}|j                  }d}||k(  }|st        j                  d|fd||f      dt        j                         v st        j                  |      rt        j                  |      ndt        j                  |      t        j                  |      dz  }dd|iz  }t        t        j                  |            d	x}x}}y	7 w)u   测试执行Skillc                       e Zd Zd Zy)%test_execute_skill.<locals>.MockSkillc                 .   K   t        ddidg i d      S w)Ncount   ffffff?d   )data
confidencesourcesmetadataexecution_time_ms)r   )self
capabilityparamscontexts       r	   executez-test_execute_skill.<locals>.MockSkill.execute   s&     r]"% s   N)__name__
__module____qualname__r   r   r   r	   	MockSkillr      s    	r   r    user_1tenant_1	trace_123)user_id	tenant_idtrace_idtest)skillr   r   r   Nr   r   )==)z,%(py2)s
{%(py2)s = %(py0)s.data
} == %(py5)sresult)py0py2py5zassert %(py7)spy7r   )z2%(py2)s
{%(py2)s = %(py0)s.confidence
} == %(py5)s)r   r   r   
@pytest_ar_call_reprcompare@py_builtinslocals_should_repr_global_name	_safereprAssertionError_format_explanationr   )	r
   r    r   r*   @py_assert1@py_assert4@py_assert3@py_format6@py_format8s	            r	   test_execute_skillr<      s9      G ##k	 $  F ;;'7B-';-'''';-''''''6'''6''';'''-'''''''$$$$$$$$$$$$6$$$6$$$$$$$$$$$$$s   6GGFG)builtinsr1   _pytest.assertion.rewrite	assertionrewriter/   pytestsyspathinsertsrc.agent.executorr   r   r   fixturer
   markasyncior<   r   r   r	   <module>rI      s^       
 6 7 G G   % %r   