
    5P@i	                    J   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	 i Z
dg iZddgiZdd	d
giZdgdgdZddgddg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)input_utilswordlukewarmhotcoldapplebanana)ab	australiaalphabrazilbeta__main__c                  *    e Zd Zd Zd Zd Zd Zd Zy)InputUtilsTestc                    | j                  t        d      5  t        j                  d       d d d        y # 1 sw Y   y xY w)N
UnknownVar)assertRaisesRegex	NameErrorr   get_inputs_source_from_py_var)selfs    w/home/www/lebenam.kofcorporation.com/venv/lib/python3.12/site-packages/google/generativeai/notebook/input_utils_test.py/test_get_inputs_source_from_py_var_invalid_namez>InputUtilsTest.test_get_inputs_source_from_py_var_invalid_name#   s9    ##I|< 	D55lC	D 	D 	Ds   6?c                p    t        j                  d      }|j                         }| j                  |       y )N_EMPTY_INPUT_VAR_ONEr   r   to_normalized_inputsassertEmptyr   sourceresultss      r   ,test_get_inputs_source_from_py_var_empty_onez;InputUtilsTest.test_get_inputs_source_from_py_var_empty_one'   /    ::;QR--/!    c                p    t        j                  d      }|j                         }| j                  |       y )N_EMPTY_INPUT_VAR_TWOr    r#   s      r   ,test_get_inputs_source_from_py_var_empty_twoz;InputUtilsTest.test_get_inputs_source_from_py_var_empty_two,   r'   r(   c                x    t        j                  d      }|j                         }| j                  ddig|       y )N_INPUT_VAR_ONEr   r   r   r   r!   assertEqualr#   s      r   3test_get_inputs_source_from_py_var_single_input_onezBInputUtilsTest.test_get_inputs_source_from_py_var_single_input_one1   s9    ::;KL--/6:./9r(   c                ~    t        j                  d      }|j                         }| j                  ddiddig|       y )N_INPUT_VAR_TWOr   r	   r
   r.   r#   s      r   3test_get_inputs_source_from_py_var_single_input_twozBInputUtilsTest.test_get_inputs_source_from_py_var_single_input_two6   s?    ::;KL--/65/FF+;<gFr(   N)__name__
__module____qualname__r   r&   r+   r0   r3    r(   r   r   r   !   s    D"
"
:
Gr(   r   )__doc__
__future__r   sysunittestr   absl.testingr   google.generativeai.notebookr   r   r*   r-   r2   _MULTI_INPUTS_VAR_ONE_MULTI_INPUTS_VAR_TWOpatchdictmodulesr4   TestCaser   mainr7   r(   r   <module>rE      s     " 
  ! 4 | :,'5&/*&ixj9 *G4Hf;MN  z3;;x+@ABGX&& G CG4 zHMMO r(   