
    5P@i                        d Z ddlmZ ddl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 ArgumentParser.    )annotationsN)absltest)argument_parserc                      e Zd Zd Zd Zy)ArgumentParserTestc                    t        j                         }| j                  t         j                  d      5  |j	                  dg       ddd       y# 1 sw Y   yxY w)z1Verify that help messages raise ParserNormalExit.zshow this help message and exitz-hN)
parser_libArgumentParserassertRaisesRegexParserNormalExit
parse_args)selfparsers     {/home/www/lebenam.kofcorporation.com/venv/lib/python3.12/site-packages/google/generativeai/notebook/argument_parser_test.py	test_helpzArgumentParserTest.test_help   sL    **,##J$?$?Abc 	&tf%	& 	& 	&s   AAc                <   d	d} |       j                  ddg      }| j                  d|j                         | j                  t        j
                  d      5   |       j                  ddg       d d d        | j                  t        j
                  d      5   |       j                  g        d d d        | j                  t        j
                  d      5   |       j                  dg       d d d        y # 1 sw Y   xY w# 1 sw Y   VxY w# 1 sw Y   y xY w)
Nc                 ^    t        j                         } | j                  dt        d       | S )N--valueT)typerequired)r	   r
   add_argumentint)r   s    r   
new_parserz<ArgumentParserTest.test_parse_arg_errors.<locals>.new_parser   s*    ..0F	dCM    r   42*   zinvalid int valuez	forty-twoz$the following arguments are requiredzexpected one argument)returnzargparse.ArgumentParser)r   assertEqualvaluer   r	   ParserError)r   r   resultss      r   test_parse_arg_errorsz(ArgumentParserTest.test_parse_arg_errors   s    	 ,))9d*;<W]]+##J$:$:<OP 	>L##Y$<=	> ##J$:$:<bc 	(L##B'	( ##J$:$:<ST 	1L##YK0	1 	1	> 	>	( 	(	1 	1s$   C:DD:DDDN)__name__
__module____qualname__r   r"    r   r   r   r      s    &1r   r   __main__)__doc__
__future__r   argparseabsl.testingr   google.generativeai.notebookr   r	   TestCaser   r#   mainr&   r   r   <module>r/      sB    # "  ! F1** 16 zHMMO r   