
    5P@i                        d Z ddlmZ ddlmZ ddlmZ  G d dej                        Ze	dk(  r ej                          yy)	zUnittest for model_registry.    )annotations)absltest)model_registryc                      e Zd Zd Zy)ModelRegistryTestc                   t        j                         }|j                  t         j                  j                        }|j                  d      }| j                  d|j                         | j                  dg|j                         y )Nthis_is_a_test)model_input)	r   ModelRegistry	get_model	ModelName
ECHO_MODEL
call_modelassertEqualr
   text_results)selfregistrymodelresultss       z/home/www/lebenam.kofcorporation.com/venv/lib/python3.12/site-packages/google/generativeai/notebook/model_registry_test.pytest_get_model_echo_modelz+ModelRegistryTest.test_get_model_echo_model   ss    !//1"">#;#;#F#FG""/?"@)7+>+>? 	*+W-A-AB    N)__name__
__module____qualname__r    r   r   r   r      s    Cr   r   __main__N)__doc__
__future__r   absl.testingr   google.generativeai.notebookr   TestCaser   r   mainr   r   r   <module>r$      sA    # " ! 7C)) C zHMMO r   