U
    h                     @  s\   d dl mZ d dlmZ d dlmZ d dlmZ ejZG dd dejZ	e
dkrXe  dS )	    )annotations)absltest)llmfn_output_row)	unique_fnc                   @  s   e Zd Zdd Zdd ZdS )UniqueFntestc                 C  sJ   t dditdt dditdt dditdg}| dddgt| d S )	Ntext_resultreddataZresult_typegreenbluer         LLMFnOutputRowstrassertEqualr   selfZrows r   S/tmp/pip-unpacked-wheel-doshhd5e/google/generativeai/notebook/lib/unique_fn_test.pytest_all_unique   s
    zUniqueFntest.test_all_uniquec              	   C  st   t dditdt dditdt dditdt dditdt dditdt dditdg}| dddgt| d S )	Nr   r   r	   r   r   r   r      r   r   r   r   r   test_some_dupes#   s    zUniqueFntest.test_some_dupesN)__name__
__module____qualname__r   r   r   r   r   r   r      s   r   __main__N)
__future__r   Zabsl.testingr   Z google.generativeai.notebook.libr   r   r   ZTestCaser   r   mainr   r   r   r   <module>   s   