U
    “öhâ  ã                   @  sJ   d dl mZ d dlmZ d dlmZ G dd„ dejƒZedkrFe 	¡  dS )é    )Úannotations)Úabsltest)Úprompt_utilsc                   @  s   e Zd Zdd„ Zdd„ ZdS )ÚPromptUtilsTestc                 C  s,   t  d¡}|  |¡ t  d¡}|  |¡ d S )NÚ zThere are no placeholders here)r   Úget_placeholdersZassertEmpty©ÚselfZplaceholders© r
   úV/tmp/pip-unpacked-wheel-doshhd5e/google/generativeai/notebook/lib/prompt_utils_test.pyÚtest_get_placeholders_empty   s    

ÿz+PromptUtilsTest.test_get_placeholders_emptyc                 C  sZ   t  d¡}|  tdhƒ|¡ t  d¡}|  tddhƒ|¡ t  d¡}|  tdhƒ|¡ d S )Nztoday {hello} worldZhelloz{hello} {world}Zworldz{hello} {hello})r   r   ÚassertEqualÚ	frozensetr   r
   r
   r   Útest_get_placeholders    s    


z%PromptUtilsTest.test_get_placeholdersN)Ú__name__Ú
__module__Ú__qualname__r   r   r
   r
   r
   r   r      s   	r   Ú__main__N)
Ú
__future__r   Zabsl.testingr   Z google.generativeai.notebook.libr   ZTestCaser   r   Úmainr
   r
   r
   r   Ú<module>   s
   