
    NSi                     l    d dl Zd dlmc mZ d dlZd dlmZm	Z	 ej                  d        Zd Zd Zd Zy)    N)QueryRouterModec                      t               S )N)r        A/home/jang/Projects/ai-researcher/tests/unit/test_query_router.pyrouterr	      s
    =r   c                 |   | j                  ddgdd      \  }}t        j                  }||k(  }|st        j                  d|fd||f      dt        j                         v st        j                  |      rt        j                  |      nddt        j                         v st        j                  t              rt        j                  t              ndt        j                  |      d	z  }d
d|iz  }t        t        j                  |            dx}}d}||k  }|st        j                  d|fd||f      dt        j                         v st        j                  |      rt        j                  |      ndt        j                  |      dz  }dd|iz  }	t        t        j                  |	            dx}}y)u$   测试简单查询使用同步模式u   你好singleuser_1required_sourcesuser_id==)z,%(py0)s == %(py4)s
{%(py4)s = %(py2)s.SYNC
}moder   py0py2py4assert %(py6)spy6N333333?)<)z%(py0)s < %(py3)sscorer   py3assert %(py5)spy5)assess_complexityr   SYNC
@pytest_ar_call_reprcompare@py_builtinslocals_should_repr_global_name	_safereprAssertionError_format_explanation
r	   r   r   @py_assert3@py_assert1@py_format5@py_format7@py_assert2@py_format4@py_format6s
             r   test_simple_query_is_syncr2   
   s    **&ZH=KD% 9949494444953;53553r   c                 ~   | j                  dddgdd      \  }}t        j                  }||k(  }|st        j                  d|fd||f      dt        j                         v st        j                  |      rt        j                  |      ndd	t        j                         v st        j                  t              rt        j                  t              nd	t        j                  |      d
z  }dd|iz  }t        t        j                  |            dx}}d}||k\  }|st        j                  d|fd||f      dt        j                         v st        j                  |      rt        j                  |      ndt        j                  |      dz  }dd|iz  }	t        t        j                  |	            dx}}y)u$   测试复杂查询使用异步模式uE   对比技术部和产品部的员工数量，分析人员配置建议hrfinancer   r   r   )z-%(py0)s == %(py4)s
{%(py4)s = %(py2)s.ASYNC
}r   r   r   r   r   Nr   )>=)z%(py0)s >= %(py3)sr   r   r   r   )r    r   ASYNCr"   r#   r$   r%   r&   r'   r(   r)   r*   s
             r   test_complex_query_is_asyncr8      s    **O"I.8DKD% ::4:4:4444:5C<5C55Cr   c                    | j                   }ddgi} ||      }d}||k(  }|st        j                  d|fd||f      dt        j                         v st        j
                  |       rt        j                  |       ndt        j                  |      t        j                  |      t        j                  |      t        j                  |      dz  }dd	|iz  }t        t        j                  |            d
x}x}x}x}}| j                   }dddgi} ||      }d}||k(  }|st        j                  d|fd||f      dt        j                         v st        j
                  |       rt        j                  |       ndt        j                  |      t        j                  |      t        j                  |      t        j                  |      dz  }dd	|iz  }t        t        j                  |            d
x}x}x}x}}| j                   }dg di} ||      }d}||k(  }|st        j                  d|fd||f      dt        j                         v st        j
                  |       rt        j                  |       ndt        j                  |      t        j                  |      t        j                  |      t        j                  |      dz  }dd	|iz  }t        t        j                  |            d
x}x}x}x}}y
)u   测试数据源评分r   r   g        r   )zY%(py6)s
{%(py6)s = %(py2)s
{%(py2)s = %(py0)s._score_data_sources
}(%(py4)s)
} == %(py9)sr	   )r   r   r   r   py9zassert %(py11)spy11Nabg      ?)r<   r=   cg?)	_score_data_sourcesr"   r#   r$   r%   r&   r'   r(   r)   )r	   r,   r+   @py_assert5@py_assert8@py_assert7@py_format10@py_format12s           r   test_score_data_sourcesrE      s   %%N'9H:&FN%&FGN3NG3NNNNG3NNNNNN6NNN6NNN%NNN&FNNNGNNN3NNNNNNN%%N'9C:&FN%&FGN3NG3NNNNG3NNNNNN6NNN6NNN%NNN&FNNNGNNN3NNNNNNN%%S'9?&KS%&KLSPSSLPSSSSSLPSSSSSSS6SSS6SSS%SSS&KSSSLSSSPSSSSSSSSSr   )builtinsr$   _pytest.assertion.rewrite	assertionrewriter"   pytestsrc.core.query_routerr   r   fixturer	   r2   r8   rE   r   r   r   <module>rM      s7       3  Tr   