U
    /j                     @  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re  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-hs3rxp5i/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   