
    <ih                     x    d dl Zd dlmc mZ d dlZej                  j                  d d       d dl	Z	d dl
mZ d Zy)    Nz!/home/jang/Projects/ai-researcher)Tracec            	         t        ddddddddgd	d
dgddi      } | 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}}| 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}}| j                  d   }d}||k(  }|slt        j                  d|fd||f      t        j                  |      t        j                  |      dz  }dd|iz  }	t        t        j                  |	            dx}x}}| j                  }dddg}||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	d
dg}||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   }d}||k(  }|slt        j                  d|fd||f      t        j                  |      t        j                  |      dz  }dd|iz  }	t        t        j                  |	            dx}x}}y)u   测试Trace创建	trace_123task_456query_employeegffffff?)intent
confidence
data_queryz
hr.api.com)skillendpoint   
extraction)steptypeoverallg
ףp=
?)idtask_idquery_analysisdata_sourcesreasoning_stepsr	   )==)z*%(py2)s
{%(py2)s = %(py0)s.id
} == %(py5)strace)py0py2py5zassert %(py7)spy7N)z/%(py2)s
{%(py2)s = %(py0)s.task_id
} == %(py5)sr   )z%(py1)s == %(py4)s)py1py4zassert %(py6)spy6)z4%(py2)s
{%(py2)s = %(py0)s.data_sources
} == %(py5)s)z7%(py2)s
{%(py2)s = %(py0)s.reasoning_steps
} == %(py5)s)r   r   
@pytest_ar_call_reprcompare@py_builtinslocals_should_repr_global_name	_safereprAssertionError_format_explanationr   r   r   r   r	   )
r   @py_assert1@py_assert4@py_assert3@py_format6@py_format8@py_assert0@py_assert2@py_format5@py_format7s
             @/home/jang/Projects/ai-researcher/tests/unit/test_trace_model.pytest_trace_creationr2   
   s   "2$G ,,GH"#\:;t$E 88"{"8{""""8{""""""5"""5"""8"""{"""""""==&J&=J&&&&=J&&&&&&5&&&5&&&=&&&J&&&&&&&)=-==)-=====)-====)===-========TLl"S!TT!TTTTT!TTTTTTT5TTT5TTTTTT!TTTTTTTT  Ga%F$GG $GGGGG $GGGGGGG5GGG5GGG GGG$GGGGGGGGI&.$.&$....&$...&...$.......    )builtinsr"   _pytest.assertion.rewrite	assertionrewriter    syspathinsertpytestsrc.models.tracer   r2    r3   r1   <module>r>      s*      
 6 7  "/r3   