U
    “öhµ  ã                   @  s„   d Z ddlmZ ddlZddlmZ ddlmZ ddlm	Z	 dZ
dZej ejdeje i¡G d	d
„ d
ejƒƒZedkr€e ¡  dS )zUnittest for py_utils.é    )ÚannotationsN)Úmock)Úabsltest)Úpy_utilsúhello worldÚ__main__c                   @  s   e Zd Zdd„ Zdd„ ZdS )ÚPyUtilsTestc              	   C  s<   |   td¡ t d¡ W 5 Q R X t d¡}|  d|¡ d S )NZIncorrectVarÚ
_INPUT_VARr   )ÚassertRaisesRegexÚ	NameErrorr   Z
get_py_varÚassertEqual)ÚselfÚresults© r   úN/tmp/pip-unpacked-wheel-doshhd5e/google/generativeai/notebook/py_utils_test.pyÚtest_get_py_var    s    
zPyUtilsTest.test_get_py_varc                 C  s4   t  dd¡ |  dt¡ t  dd¡ |  dt¡ d S )NÚ_OUTPUT_VARzworld helloÚ_NEW_VARzworld hello world)r   Z
set_py_varr   r   r   )r   r   r   r   Útest_set_py_var(   s    zPyUtilsTest.test_set_py_varN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r      s   r   )Ú__doc__Ú
__future__r   ÚsysZunittestr   Zabsl.testingr   Zgoogle.generativeai.notebookr   r	   r   ÚpatchÚdictÚmodulesr   ZTestCaser   Úmainr   r   r   r   Ú<module>   s   