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	 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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)input_utilswordlukewarmhotcoldZappleZbanana)abZ	australiaalphaZbrazilbeta__main__c                   @  s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )InputUtilsTestc              	   C  s&   |  td td W 5 Q R X d S )NZ
UnknownVar)assertRaisesRegex	NameErrorr   get_inputs_source_from_py_var)self r   Q/tmp/pip-unpacked-wheel-doshhd5e/google/generativeai/notebook/input_utils_test.py/test_get_inputs_source_from_py_var_invalid_name$   s    z>InputUtilsTest.test_get_inputs_source_from_py_var_invalid_namec                 C  s    t d}| }| | d S )N_EMPTY_INPUT_VAR_ONEr   r   to_normalized_inputsZassertEmptyr   sourceresultsr   r   r   ,test_get_inputs_source_from_py_var_empty_one(   s    
z;InputUtilsTest.test_get_inputs_source_from_py_var_empty_onec                 C  s    t d}| }| | d S )N_EMPTY_INPUT_VAR_TWOr   r   r   r   r   ,test_get_inputs_source_from_py_var_empty_two-   s    
z;InputUtilsTest.test_get_inputs_source_from_py_var_empty_twoc                 C  s(   t d}| }| ddig| d S )N_INPUT_VAR_ONEr   r   r   r   r   assertEqualr   r   r   r   3test_get_inputs_source_from_py_var_single_input_one2   s    
zBInputUtilsTest.test_get_inputs_source_from_py_var_single_input_onec                 C  s.   t d}| }| ddiddig| d S )N_INPUT_VAR_TWOr   r   r	   r!   r   r   r   r   3test_get_inputs_source_from_py_var_single_input_two7   s    
zBInputUtilsTest.test_get_inputs_source_from_py_var_single_input_twoN)__name__
__module____qualname__r   r   r   r#   r%   r   r   r   r   r   !   s
   r   )__doc__
__future__r   sysZunittestr   Zabsl.testingr   Zgoogle.generativeai.notebookr   r   r   r    r$   Z_MULTI_INPUTS_VAR_ONEZ_MULTI_INPUTS_VAR_TWOpatchdictmodulesr&   ZTestCaser   mainr   r   r   r   <module>   s   
