5P@i>UdZddlmZddlZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZdd lmZddlmZddgiZddgiZddgiZdddgiZdaded<daded<gdZd0dZGddej:Zej>jAejBdejBe"iGddeZ#ej>jAejBdejBe"iGd d!eZ$ej>jAejBdejBe"iGd"d#eZ%ej>jAejBdejBe"iGd$d%eZ&ejNejPd&g'Z)ejNejPd(g'Z*d1d)Z+ej>jAejBdejBe"iGd*d+eZ,ej>jAejBdejBe"iGd,d-eZ-ej>jAejBdejBe"iGd.d/eZ.e"dk(rej^yy)2zUnittests for cmd_line_parser.) annotationsN)mock)absltest)argument_parser)cmd_line_parser)model_registry)parsed_args_lib)post_process_utils) llm_function)llmfn_output_row) llmfn_outputsmodelwordonetwothreenot_wordhelloworldz!llmfn_outputs.LLMFnOutputs | None_OUTPUT_VAR_ONE_OUTPUT_VAR_TWOapplebanana cantaloupec|jtjtjddit j tj jdtj j|itggy)Nr)data result_type) prompt_num input_num prompt_vars output_rows)outputs) write_outputsr LLMFnOutputsLLMFnOutputEntryr LLMFnOutputRow ColumnNames RESULT_NUM TEXT_RESULTstr text_resultsinks {/home/www/lebenam.kofcorporation.com/venv/lib/python3.12/site-packages/google/generativeai/notebook/cmd_line_parser_test.py_set_output_sinkr1*sy""..  "(77 - 9 9 D Da - 9 9 E E "), !  ceZdZfdZxZS)CmdLineParserTestBasec*t|dadayN)supersetUprr)self __class__s r0r8zCmdLineParserTestBase.setUpBs   r2)__name__ __module__ __qualname__r8 __classcell__)r:s@r0r4r4As r2r4__main__c"eZdZdZdZdZdZy)CmdLineParserCommonTestz/For tests that are not specific to any command.cTtj}|jtj5|j dddd|jtj5|j dddd|jtj5|j dddd|jtj5|j ddddy#1swYxYw#1swYxYw#1swY[xYw#1swYyxYw)Nz-hz--helpzrun -hz run --help)r CmdLineParser assertRaisesrParserNormalExit parse_liner9parsers r0test_parse_args_helpz,CmdLineParserCommonTest.test_parse_args_helpQs ..0   ?? @ $   d # $   ?? @ (   h ' (   ?? @ (   h ' (   ?? @ ,   l + , ,  $ $ ( ( ( ( , ,s/C:-D&DD:DDDD'ctj}|jd\}}|jtj j |j|jtjj|j|jg|j|jtj|jy)N)rrCrF assertEqualr CommandNameRUN_CMDcmdr ModelName TEXT_MODEL model_typeinputs model_libModelArguments model_argsr9rHresults_s r0test_parse_args_emptyz-CmdLineParserCommonTest.test_parse_args_empty\s ..0&&r*  44<P>PQ W^^,   $ $ &    r2c tj}|jd\}}|j|jd|j ddig|jdj |jd\}}|j|jd|j ddig|jdj |j dd ig|jdj y) Nz--inputs _INPUT_VAR_ONErrrz(--inputs _INPUT_VAR_TWO _INPUT_VAR_THREErr)rrCrF assertLenrSrLto_normalized_inputsrWs r0test_parse_args_no_reusez0CmdLineParserCommonTest.test_parse_args_no_reusegs ..0&&'@A  w~~q) 65/*GNN1,=,R,R,TU&&'QR  w~~q) 65/*GNN1,=,R,R,TU 67+,gnnQ.?.T.T.VWr2N)r;r<r=__doc__rIrZr`r2r0rArAMs9 ,   Xr2rAc*eZdZdZdZdZdZdZy)CmdLineParserModelFlagsTestcLtj}|jd\}}|jtj j |j|jd\}}|jtj j|jy)N--model_type=echoz--model_type=text) rrCrFrLrrP ECHO_MODELrRrQrWs r0test_parse_args_sets_model_typez;CmdLineParserModelFlagsTest.test_parse_args_sets_model_typews~ ..0&&':;  11<P>PQ&&':;  11<P>PQr2ctj}|jd\}}|jt j d|j y)Nz--model=/ml/testz/ml/testr)rrCrFrLrTrUrVrWs r0test_parse_args_sets_modelz6CmdLineParserModelFlagsTest.test_parse_args_sets_modelsF ..0&&'9:  11 CWEWEWXr2ctj}|jd\}}|jt j d|j |jd\}}|jt j d|j |jtjd5|jddddy#1swYyxYw)Nz--temperature=0r) temperaturez--temperature=0.5g?z`Error with value "-1.0", got ValueError: Value should be greater than or equal to zero, got -1.0z--temperature=-1 rrCrFrLrTrUrVassertRaisesRegexr ParserErrorrWs r0 test_parse_args_sets_temperaturezP>PQr2cJtj}|jd\}}|jtj j |j|jtjj|j|j|jd|jddig|jdj|jddig|jdj|j|jdt!d|jd |j#t$t&j(|jdt$dj*dt&j,j.t!d|jd |j#t0t&j(|jdt0dj*dt&j,j.t!d |jd t2}|j#|t&j(|jd |dj*dt&j,j.y) Nz}run --model_type=echo --inputs _INPUT_VAR_ONE _INPUT_VAR_TWO --outputs _OUTPUT_VAR_ONE _OUTPUT_VAR_TWO _UNDECLARED_OUTPUT_VARr]rrrrr\r- undeclared)rrCrFrLr rMrNrOrrPrgrRr^rSr_r$r1assertIsInstancerr r&r#r)r+r_UNDECLARED_OUTPUT_VAR)r9rHrXrYundeclared_vars r0 test_parse_input_and_output_argsz5CmdLineParserRunTest.test_parse_input_and_output_argss  ..0&& F  44<P>PQ w~~q) 65/*GNN1,=,R,R,TU 65/*GNN1,=,R,R,TU w* U1CD o}/I/IJ   A  * *1 -m.G.G.S.S T  U1CD o}/I/IJ   A  * *1 -m.G.G.S.S T  \8JK/ nm.H.HI   1  ) )! ,]-F-F-R-R S r2ctj}|jtjd5|j dt dhdddy#1swYyxYw)Nsargument --inputs/-i: Error with value "_NOT_WORD_INPUT_VAR", got ValueError: Placeholder "word" not found in inputz run --inputs _NOT_WORD_INPUT_VARr placeholdersrrCrnrrorF frozensetrGs r0test_placeholder_errorz+CmdLineParserRunTest.test_placeholder_errorsd ..0  # #  ' 'E    2&x0      AA%N)r;r<r=rar~rrrbr2r0r{r{s R $ L r2r{c"eZdZdZdZdZdZy)CmdLineParserCompileTestzFor the "compile" command.ctj}|jtjd5|j ddddy#1swYyxYw)Nz7the following arguments are required: compile_save_namecompilerrCrnrrorFrGs r0test_parse_args_needs_save_namez8CmdLineParserCompileTest.test_parse_args_needs_save_namesO ..0  # #  ' ' E  )   i (  ) ) ) AActj}|jtjd5|j ddddy#1swYyxYw)NzInvalid Python variable namez compile 1234rrGs r0test_parse_args_bad_save_namez6CmdLineParserCompileTest.test_parse_args_bad_save_namesJ ..0  # #O$?$?A_ ` .   n - . . .rctj}|jd\}}|jd|jy)Nz compile my_fnmy_fn)rrCrFrLcompile_save_namerWs r0test_parse_args_has_save_namez6CmdLineParserCompileTest.test_parse_args_has_save_name s: ..0&&7  '";";compare _test_lhs_fn _test_rhs_fn --inputs _NOT_WORD_INPUT_VARrrGs r0rz/CmdLineParserCompareTest.test_placeholder_error4sW ..0  # #  ' 'E  `   ^ _ ` ` `rN)r;r<r=rarrrrbr2r0rrs$-W `r2rceZdZdZdZdZy)CmdLineParserEvalTestzFor the "eval" command.ctj}|jd\}}|jgd|jy)Nz=eval --ground_truth _GROUND_TRUTH_VAR --inputs _INPUT_VAR_ONEr)rrCrFrL ground_truthrWs r0 test_evalzCmdLineParserEvalTest.test_evalEs@ ..0&& K  :G? *'9!'<=r2ctj}|jtjd5|j dddd|jtjd5|j dddd|jtjd5|j ddddy#1swY~xYw#1swYPxYw#1swYyxYw)Nz,Cannot have empty post-processing expressionz | | to_upperz| z| add_length |rrCrnr PostProcessParseErrorrFrGs r0test_illformed_expressionz9CmdLineParserPostProcessingTest.test_illformed_expressiongs ..0  # #  4 4 :  .   n -  .  # #  4 4 :  $   d #  $  # #  4 4 :  0   . /  0 0 . .  $ $  0 0s#C/C)CC CC%ctj}|jtjd5|j ddddy#1swYyxYw)Nz3Post-processing expression should be a single tokenz| add_length to_upperrrGs r03test_cannot_parse_multiple_tokens_in_one_expressionzSCmdLineParserPostProcessingTest.test_cannot_parse_multiple_tokens_in_one_expression{sP ..0  # #  4 4 A  7   5 6  7 7 7rN)r;r<r=rarrrrbr2r0rr\s >0(7r2r)r.r,r/zllmfn_outputs.LLMFnOutputsSinkreturnNone)rr,rr,rbool)0ra __future__rsysunittestr absl.testingrgoogle.generativeai.notebookrrrr r google.generativeai.notebook.libr r r rrT_INPUT_VAR_ONE_INPUT_VAR_TWO_INPUT_VAR_THREE_NOT_WORD_INPUT_VARr__annotations__r_GROUND_TRUTH_VARr1TestCaser4patchdictmodulesr;rArdr{rLLMFunctionImpl EchoModelrrrrrrmainrbr2r0rs%" !8878;9=:?5'"5'"WI&!GW#56592959295.H--z3;;x+@AB#X3#XC#XNz3;;x+@AB@("7@(C@(Hz3;;x+@AB;0;C;~z3;;x+@AB=4=C=,,|++ )   *C)D ,|++ )   *C)D  z3;;x+@AB`4`C`Bz3;;x+@AB1C4z3;;x+@AB$7&;$7C$7N zHMMOr2