
    5P@i                       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                  j                  ej                  dej                  e   i       G d	 d
ej                                Zedk(  r ej$                          yy)zUnittest for py_utils.    )annotationsN)mock)absltest)py_utilshello world__main__c                      e Zd Zd Zd Zy)PyUtilsTestc                    | j                  t        d      5  t        j                  d       d d d        t        j                  d      }| j	                  d|       y # 1 sw Y   1xY w)NIncorrectVar
_INPUT_VARr   )assertRaisesRegex	NameErrorr   
get_py_varassertEqual)selfresultss     t/home/www/lebenam.kofcorporation.com/venv/lib/python3.12/site-packages/google/generativeai/notebook/py_utils_test.pytest_get_py_varzPyUtilsTest.test_get_py_var   sW    ##I~> 	0/	0 %%l30		0 	0s   AA&c                    t        j                  dd       | j                  dt               t        j                  dd       | j                  dt               y )N_OUTPUT_VARzworld hello_NEW_VARzworld hello world)r   
set_py_varr   r   r   )r   s    r   test_set_py_varzPyUtilsTest.test_set_py_var'   sF    M=94 	J(;<,h7    N)__name__
__module____qualname__r   r    r   r   r
   r
      s    18r   r
   )__doc__
__future__r   sysunittestr   absl.testingr   google.generativeai.notebookr   r   r   patchdictmodulesr   TestCaser
   mainr   r   r   <module>r+      s     " 
  ! 1
 z3;;x+@AB8(## 8 C8& zHMMO r   