B u9a@s:ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZGddde ZGdddejZGd d d eZGd d d eZGd ddeZGdddeZddZGddde jZGdddeZefZedeiZGdddeZGdddeZGdddeZ GdddeZ!Gd d!d!eZ"Gd"d#d#eZ#Gd$d%d%eZ$Gd&d'd'eZ%Gd(d)d)eZ&Gd*d+d+eZ'Gd,d-d-eZ(Gd.d/d/eZ)Gd0d1d1eZ*Gd2d3d3eZ+Gd4d5d5eZ,Gd6d7d7eZ-Gd8d9d9eZ.Gd:d;d;eZ/Gdd?d?eZ1Gd@dAdAeZ2GdBdCdCeZ3GdDdEdEeZ4GdFdGdGeZ5GdHdIdIeZ6GdJdKdKeZ7GdLdMdMeZ8GdNdOdOeZ9GdPdQdQeZ:GdRdSdSeZ;GdTdUdUeZGdZd[d[eZ?Gd\d]d]eZ@Gd^d_d_eZAGd`dadaeZBGdbdcdceZCGdddedeeZDGdfdgdgeZEGdhdidieZFGdjdkdkeZGGdldmdmeZHGdndodoeZIGdpdqdqeZJGdrdsdseZKGdtdudueZLGdvdwdweZMGdxdydyeZNGdzd{d{eZOGd|d}d}eZPGd~ddeZQGdddeZRGdddeZSGdddeZTGdddeZUGdddeZVGdddeZWGdddeZXGdddeZYGdddeZZGdddeZ[GdddeZ\GdddeZ]GdddeZ^GdddeZ_GdddeZ`GdddeZaGdddeZbGdddeZcGdddeZdGdddeZeGdddeZfGdddeZgGdddeZhGdddeZiGdddeZjGdddeZkGdddeeZlGdddeeZmGdddeZnGdddZoeodZpeodZqeodZrGdddeZsGdd„deeZtGddĄdeeZuGddƄdeeZvGddȄdeZwexeyedɃoezdkdʡGdd̄deeZ{Gdd΄deeZ|GddЄdeZ}Gdd҄deZ~GddԄdeZGddքdeZGdd؄deZGddڄdeZGdd܄deZGddބdeZGdddeZGdddeZGdddeZGdddeZGdddeeZGdddeeZGdddeeZGdddeeZGdddeeZGdddeeZGdddeeZGdddeeZGdddeZGdddeeZGdddeeZGdddeeZGdddeeZGdddeeZGdddeeZGdddeeZGdddeeZGd d d eZGd d d eZGd ddeZGdddeZefZedeiZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGd d!d!eZGd"d#d#eZGd$d%d%eZGd&d'd'eZGd(d)d)eZGd*d+d+eZGd,d-d-eZGd.d/d/eZGd0d1d1eZGd2d3d3eZGd4d5d5eZGd6d7d7eZGd8d9d9eZGd:d;d;eZGd<d=d=eZGd>d?d?eZGd@dAdAeZGdBdCdCeZGdDdEdEeZGdFdGdGeZGdHdIdIeZGdJdKdKeZGdLdMdMeZGdNdOdOeZGdPdQdQeZGdRdSdSeZGdTdUdUeZGdVdWdWeZGdXdYdYeZGdZd[d[eZGd\d]d]eZGd^d_d_eZGd`dadaeZGdbdcdceZGdddedeeZGdfdgdgeZGdhdidieZGdjdkdkeZGdldmdmeZGdndodoeZGdpdqdqeZGdrdsdseZGdtdudueZGdvdwdweZGdxdydyeZGdzd{d{eZGd|d}d}eZGd~ddeZՐddZeאdkr6eփdS(N)StringIO)support)mockc@s eZdZdS) StdIOBufferN)__name__ __module__ __qualname__r r #/usr/lib/python3.7/test_argparse.pyrsrc@seZdZddZdS)TestCasecCs"t}|d||jdS)NCOLUMNS)rEnvironmentVarGuardZunset addCleanup__exit__)selfenvr r r setUps zTestCase.setUpN)rrrrr r r r r sr c@s$eZdZddZddZddZdS) TempDirMixincCs$t|_t|_t|jdS)N)tempfileZmkdtemptemp_dirosgetcwdold_dirchdir)rr r r r"s  zTempDirMixin.setUpcCsft|jxFtj|jddD]2\}}}x&|D]}ttj|j|tj q.WqWt |jddS)NF)topdownT) rrrwalkrchmodpathjoinstatS_IWRITEshutilZrmtree)rrootdirsfilesnamer r r tearDown's   "zTempDirMixin.tearDownc CsBtj|j|}t|d}||WdQRXt|tjdS)Nw) rrrropenwriterrS_IREAD)rfilenameZ file_pathfiler r r create_readonly_file.s z!TempDirMixin.create_readonly_fileN)rrrrr&r-r r r r r src@seZdZddZdS)SigcOs||_||_dS)N)argskwargs)rr/r0r r r __init__6sz Sig.__init__N)rrrr1r r r r r.4sr.c@s$eZdZddZddZddZdS)NScKs|j|dS)N)__dict__update)rr0r r r r1=sz NS.__init__cCs4t|j}ddd|D}dt|j|fS)Nz, cSsg|] }d|qS)z%s=%rr ).0tupr r r BszNS.__repr__..z%s(%s))sortedr3itemsrtyper)rZ sorted_itemsZ kwarg_strr r r __repr__@sz NS.__repr__cCst|t|kS)N)vars)rotherr r r __eq__Esz NS.__eq__N)rrrr1r;r>r r r r r2;sr2c@seZdZdddZdS)ArgumentParserErrorNcCs,t||||||_||_||_||_dS)N) Exceptionr1messagestdoutstderr error_code)rrArBrCrDr r r r1Ks zArgumentParserError.__init__)NNN)rrrr1r r r r r?Isr?c Osttjtsttjtr"|||Stj}tj}tt_tt_zy\|||}xLtt|D]<}t||tjkrzt|||t||tjkrZt|||qZW|St k rt dj }tj }tj } t d|| |YnXWd|t_|t_XdS)N SystemExit) isinstancesysrCrrBlistr<getattrsetattrrFexc_infocodegetvaluer?) parse_argsr/r0Z old_stdoutZ old_stderrresultkeyrMrBrCr r r stderr_to_parser_errorSs,     rRcs4eZdZfddZfddZfddZZS)ErrorRaisingArgumentParsercstt|j}t|f||S)N)superrSrOrR)rr/r0rO) __class__r r rOus z%ErrorRaisingArgumentParser.parse_argscstt|j}t|f||S)N)rTrSexitrR)rr/r0rV)rUr r rVys zErrorRaisingArgumentParser.exitcstt|j}t|f||S)N)rTrSerrorrR)rr/r0rW)rUr r rW}s z ErrorRaisingArgumentParser.error)rrrrOrVrW __classcell__r r )rUr rSss  rSc@seZdZdZddZdS)ParserTesterMetaclassa}Adds parser tests using the class attributes. Classes of this type should specify the following attributes: argument_signatures -- a list of Sig objects which specify the signatures of Argument objects to be created failures -- a list of args lists that should cause the parser to fail successes -- a list of (initial_args, options, remaining_args) tuples where initial_args specifies the string args to be parsed, options is a dict that should match the vars() of the options parsed out of initial_args, and remaining_args should be any remaining unparsed arguments c Cs|dkr dSt|dst|_t|ds.t|_dd}dd}dd }d d }d d }Gdddt} x0|||gD]"} x||gD]} | || | qWqrWdS)NParserTestCaseparser_signature parser_classcSs"x|D]}|j|j|jqWdS)z(Add all arguments directly to the parserN) add_argumentr/r0)parserargument_signaturessigr r r no_groupss z1ParserTesterMetaclass.__init__..no_groupscSs,|d}x|D]}|j|j|jqWdS)z4Add all arguments under a single group in the parserfooN)add_argument_groupr]r/r0)r^r_groupr`r r r one_groups  z1ParserTesterMetaclass.__init__..one_groupcSs8x2t|D]&\}}|d|}|j|j|jq WdS)z0Add each argument in its own group to the parserzfoo:%iN) enumeratercr]r/r0)r^r_ir`rdr r r many_groupssz3ParserTesterMetaclass.__init__..many_groupscSs ||S)z#Parse the args by passing in a list)rO)r^r/r r r listargssz0ParserTesterMetaclass.__init__..listargscSs,tj}|dg|t_z|S|t_XdS)z(Parse the args by defaulting to sys.argvrN)rHargvrO)r^r/Z old_sys_argvr r r sysargss z/ParserTesterMetaclass.__init__..sysargsc@s,eZdZddZddZddZddZd S) z0ParserTesterMetaclass.__init__..AddTestsc Ss||_||_|jj}|jj}xf|j|jgD]V}|j}|||f}d|} |fdd} y | | _Wntk rrYnXt|| | q*WdS)N_cSs ||dS)Nr )r test_funcr r r wrapperszJParserTesterMetaclass.__init__..AddTests.__init__..wrapper)_add_arguments _parse_argsr test_failurestest_successesr TypeErrorrK) rZ tester_cls add_argumentsrOZadd_arguments_nameZparse_args_namermZ func_namenames test_namernr r r r1s    z9ParserTesterMetaclass.__init__..AddTests.__init__cSs.|jj}|jj}|j||}|||j|S)N)r[r/r0r\ror_)rtesterr/r0r^r r r _get_parsers  z.AddTests._get_parserc SsJ||}x:|jD]0}|}|jt|d||WdQRXqWdS)N)msg)rxfailuressplit assertRaisesr?rO)rrwr^args_strr/r r r rqs   z>ParserTesterMetaclass.__init__..AddTests.test_failurescSsL||}x<|jD]2\}}t|tr,|}|||}|||qWdS)N)rx successesrGstrr{rp assertEqual)rrwr^r/ expected_nsZ result_nsr r r rrs    z?ParserTesterMetaclass.__init__..AddTests.test_successesN)rrrr1rxrqrrr r r r AddTestssr)hasattrr.r[rSr\object) clsr%basesbodydictrarerhrirkrrtrOr r r r1s    -zParserTesterMetaclass.__init__N)rrr__doc__r1r r r r rYsrYrZc@sheZdZdZedgZdddddgZdedd fd edd fd edd fd ed d fded d fgZdS)TestOptionalsSingleDashz1Test an Optional with a single-dash option stringz-xaz--fooz-x --fooz-x -yN)xz-x az-xaz-x -1z-1z-x-1) rrrrr.r_rzr2r~r r r r rs     rc@seZdZdZedddedddded gZd d d d dddddddg Zdeddddfdeddddfdeddd dfdeddd dfdeddd dfdeddd dfdeddd dfdeddd dfd eddddfdeddddfd!eddd dfd"eddd dfg ZdS)#TestOptionalsSingleDashCombinedz1Test an Optional with a single-dash option stringz-x store_true)actionz-yyy store_const*)rconstz-zrz--fooz-xaz-x --fooz-x -zz-z -xz-yxz-yz az-yyyxz-yyyzaz-xyzarFN)ryyyzTz-zaz-z az-xzaz-xz az-x -zaz-x -z az-yz -x -yyy -zaz -x -yyy -z a) rrrrr.r_rzr2r~r r r r rs$    rc@sjeZdZdZedgZddddddgZded d fd edd fd ed d fdedd fdedd fgZd S)TestOptionalsSingleDashLongzATest an Optional with a multi-character single-dash option stringz-foorz--fooz -foo --fooz-foo -yz-fooarN)rbz-foo az-foo -1z-1z-fo az-f a) rrrrr.r_rzr2r~r r r r r"s     rc @seZdZdZedededgZdddddd d gZd ed d d d fdedd d d fdedd d d fdedd d d fdedd d d fded dd d fded d dd fgZd S)&TestOptionalsSingleDashSubsetAmbiguousz=Test Optionals where option strings are subsets of each otherz-fz-foobarz-foorabz-fooz-foz-foo bz-foobz-foobaz-foorarN)ffoobarfoorabz-f arz-faz-foaZoaz-fooaZooaz -foobar az -foorab a) rrrrr.r_rzr2r~r r r r r0src @seZdZdZededgZdddddd d gZd ed d d fdedd d fded dd fdedd d fded dd fdedd d fded dd fgZd S) TestOptionalsSingleDashAmbiguousz7Test Optionals that partially match but are not subsetsz-foobarz-foorabz-fz-f az-faz-foaz-fooz-foz-foo brN)rrz-foob arz-foor az-fooba az-foora az -foobar az -foorab a) rrrrr.r_rzr2r~r r r r r@src@sbeZdZdZedddgZdddddd gZd ed d fd edd fdedd fdedd fgZd S)TestOptionalsNumericz(Test an Optional with a short opt stringz-1one)destrz-1 --fooz-1 -yz-1 -1z-1 -2rN)rz-1 az-1az-1-2z-2) rrrrr.r_rzr2r~r r r r rPs   rc@sjeZdZdZedgZddddddgZded d fd edd fd edd fd edd fdedd fgZd S)TestOptionalsDoubleDashz1Test an Optional with a double-dash option stringz--fooz-fz-f arz--foo -xz --foo --barrN)rbz--foo az--foo=az --foo -2.5z-2.5z --foo=-2.5) rrrrr.r_rzr2r~r r r r r]s     rc @seZdZdZedddedgZdddd d d gZd ed ddfded ddfdedddfdedddfdedddfdedddfgZdS)#TestOptionalsDoubleDashPartialMatchz7Tests partial matching with a double-dash option stringz--badgerr)rz--batz--barz--bz--baz--b=2z--ba=4z --badge 5rFN)badgerZbatz--bat XXz--badTz--badgz--badge) rrrrr.r_rzr2r~r r r r rks  rc @seZdZdZedddedgZddddd gZd ed d d fded dd fded dd fdedd d fdedd d fdedd d fdedd d fgZd S)"TestOptionalsDoubleDashPrefixMatchz?Tests when one double-dash option string is a prefix of anotherz--badgerr)rz--baz--barz--bz--b=2z --badge 5rFN)rZbaz--ba Xrz--ba=Xz--badTz--badgz--badge) rrrrr.r_rzr2r~r r r r r}s  rc @seZdZdZedddededddd gZdd d d d dgZdeddddfdeddddfdeddddfdeddddfdeddddfdeddddfgZdS)TestOptionalsSingleDoubleDashzeZdZdZedgZddgZdeddfdeddfgZdS)TestOptionalsNargsDefaultz7Tests not specifying the number of args for an Optionalz-xrrN)rz-x a) rrrrr.r_rzr2r~r r r r r s   rc@sDeZdZdZedddgZddgZdeddfd edgdfgZdS) TestOptionalsNargs1z&Tests specifying 1 arg for an Optionalz-xrE)nargsrrN)rz-x a) rrrrr.r_rzr2r~r r r r rs  rc@sPeZdZdZedddgZdddddd gZd ed d fd edddgd fgZd S)TestOptionalsNargs3z'Tests specifying 3 args for an Optionalz-x)rrz-x az-x a bza -xza -x brN)rz-x a b cbc) rrrrr.r_rzr2r~r r r r r#s  rc@seZdZdZedddeddddeddd d ed ded d dgZdgZdeddd ddfdeddd ddfdeddd ddfdeddd ddfdeddd ddfdedddddfdedddddfd eddd ddfdeddd ddfg Z dS)TestOptionalsNargsOptionalz0Tests specifying an Optional arg for an Optionalz-w?)rz-xr)rrz-yspam)rrz-z42Z84)rr:rr2rNT)r'rrrz-w 2z-x 2z-y 2z-z 2) rrrrr.intr_rzr2r~r r r r r.s   rc @seZdZdZedddeddddgZdgZd ed dd fdegdd fd edgdd fd eddgdd fded gd fded dgd fded ddgd fgZd S)TestOptionalsNargsZeroOrMorez?Tests specifying args for an Optional that accepts zero or morez-x*)rz-yr)rrrrN)rrz-x az-x a brz-y az-y a b) rrrrr.r_rzr2r~r r r r rEs rc @seZdZdZedddeddddgZdddd d gZd ed dd fdedgdd fdeddgdd fded dgd fded ddgd fgZd S)TestOptionalsNargsOneOrMorez>Tests specifying args for an Optional that accepts one or morez-x+)rz-yr)rrrza -xza -y brN)rrz-x az-x a brz-y az-y a b) rrrrr.r_rzr2r~r r r r rXs rc @seZdZdZedddedeeddgZdd d d d gZd e dddfde dddfde dddfde dddfde dddfde dddfgZ dS)TestOptionalsChoicesz,Tests specifying the choices for an Optionalz-fabc)choicesz-g)r:rrz-f dz-fadz-gaz-g 6rN)rrz-f az-f crz-g 0rz-g 03rz-fb -g4r) rrrrr.rranger_rzr2r~r r r r ris rc@sDeZdZdZededdgZddgZdedd fd ed d fgZ d S) TestOptionalsRequiredz)Tests an optional action that is requiredz-xT)r:requiredrrz-x 1rE)rz-x42rN) rrrrr.rr_rzr2r~r r r r rzs  rc@sBeZdZdZedddgZddgZdedd fd ed d fgZdS) TestOptionalsActionStorez&Tests the store action for an Optionalz-xstore)rrza -xrN)rz-xfoorb) rrrrr.r_rzr2r~r r r r rs  rc@sBeZdZdZeddedgZdgZdeddfdeedfgZ dS) TestOptionalsActionStoreConstz,Tests the store_const action for an Optionalz-yr)rrrrN)r) rrrrr.rr_rzr2r~r r r r rs  rc@sDeZdZdZedddgZdddgZded d fded d fgZd S) TestOptionalsActionStoreFalsez,Tests the store_false action for an Optionalz-z store_false)rrz-zaz-z arT)rFN) rrrrr.r_rzr2r~r r r r rs   rc@sDeZdZdZedddgZdddgZded d fded d fgZd S) TestOptionalsActionStoreTruez+Tests the store_true action for an Optionalz--appler)rrz --apple=bz --apple brF)ZappleTN) rrrrr.r_rzr2r~r r r r rs   rc@sXeZdZdZedddgZddddgZded d fd edgd fd edd gd fgZd S)TestOptionalsActionAppendz'Tests the append action for an Optionalz--bazappend)rrza --bazz --baz a brN)rz--baz az--baz a --baz br) rrrrr.r_rzr2r~r r r r rs   rc@sbeZdZdZedddgdgZddddgZd edgd fd eddgd fd eddd gd fgZdS)$TestOptionalsActionAppendWithDefaultz'Tests the append action for an Optionalz--bazrr)rrrza --bazz --baz a br)rz--baz az--baz a --baz brN) rrrrr.r_rzr2r~r r r r rs  rc@sleZdZdZeddededdddgZd dd d d gZd eddfdeegdfdeededgdfgZ dS)TestOptionalsActionAppendConstz-Tests the append_const action for an Optionalz-b append_const)rrz-crr)rrrza -cz-bxz-b xrN)rz-b -cx -b -cyzryz) rrrrr.r@r_rzr2r~r r r r rs  rc @sveZdZdZeddedgdedddd gZd dd d d gZdedgdfdedegdfdedededgdfgZ dS))TestOptionalsActionAppendConstWithDefaultz-Tests the append_const action for an Optionalz-brr)rrrz-crr)rrrza -cz-bxz-b xr)rz-b -cx -b -cyzrrN) rrrrr.r@r_rzr2r~r r r r rsrc@sFeZdZdZedddgZddddgZd ed d fded d fgZd S) TestOptionalsActionCountz&Tests the count action for an Optionalz-xcount)rrz-x az-x bz -x a -x brN)rrE) rrrrr.r_rzr2r~r r r r rs   rc @sveZdZdZedededddgZddgZd ed d d d fd edd d d fded dd d fdedd dd fgZd S)"TestOptionalsAllowLongAbbreviationz2Allow long options to be abbreviated unambiguouslyz--fooz--foobazz--foobler)rz--foob 5z--foobrNF)rbZfoobazZfooblez--foo 77z --fooba arz--foobl --foo grT) rrrrr.r_rzr2r~r r r r rsrc@steZdZdZeddZededddedgZd d d d gZd eddddfdeddddfdeddddfgZ dS)%TestOptionalsDisallowLongAbbreviationz1Do not allow abbreviations of long options at allF) allow_abbrevz--fooz--foodler)rz--foonlyz-foon 3z--foon 3z--foodz--food --foo 2rN)rbZfoodleZfoonlyz--foo 33z--foonly 7 --foodle --foo 2rTr) rrrrr.r[r_rzr2r~r r r r rs    rc@s4eZdZdZedgZdddgZdeddfgZdS) TestPositionalsNargsNonez,Test a Positional that doesn't specify nargsrbrz-xza br)rbN) rrrrr.r_rzr2r~r r r r rs  rc@s:eZdZdZedddgZdddgZdedgd fgZd S) TestPositionalsNargs1z.Test a Positional that specifies an nargs of 1rbrE)rrz-xza br)rbN) rrrrr.r_rzr2r~r r r r r#s rc@s>eZdZdZedddgZddddgZd edd gd fgZd S) TestPositionalsNargs2z.Test a Positional that specifies an nargs of 2rbr)rrrz-xza b cza br)rbN) rrrrr.r_rzr2r~r r r r r-s rc@sReZdZdZedddgZdgZdegdfdedgdfd edd gdfgZd S) TestPositionalsNargsZeroOrMorez0Test a Positional that specifies unlimited nargsrbr)rz-xr)rbrza brN) rrrrr.r_rzr2r~r r r r r7s  rc@sTeZdZdZeddddgZdgZdeddfd ed gdfd ed d gdfgZd S) %TestPositionalsNargsZeroOrMoreDefaultz>Test a Positional that specifies unlimited nargs and a defaultrbrr)rrz-xr)rbrza brN) rrrrr.r_rzr2r~r r r r rCs  rc@sHeZdZdZedddgZddgZdedgdfd edd gdfgZd S) TestPositionalsNargsOneOrMorez2Test a Positional that specifies one or more nargsrbr)rrz-xr)rbza brN) rrrrr.r_rzr2r~r r r r rOs rc@sBeZdZdZedddgZddgZdedd fd ed d fgZdS) TestPositionalsNargsOptionalzTests an Optional Positionalrbr)rz-xza brN)rbr) rrrrr.r_rzr2r~r r r r rZs  rc@sDeZdZdZeddddgZddgZdedd fd ed d fgZd S) #TestPositionalsNargsOptionalDefaultz1Tests an Optional Positional with a default valuerbrr)rrz-xza br)rbrN) rrrrr.r_rzr2r~r r r r res  rc@sHeZdZdZeddeddgZdddgZd ed d fd ed d fgZ dS),TestPositionalsNargsOptionalConvertedDefaultznTests an Optional Positional with a default value that needs to be converted to the appropriate type. rbrr)rr:rz-xza bz1 2rr)rb1rEN) rrrrr.rr_rzr2r~r r r r rps   rc@s>eZdZdZededgZddddgZdedd d fgZd S) TestPositionalsNargsNoneNonez-Test two Positionals that don't specify nargsrbrrz-xrza b cza br)rbrN) rrrrr.r_rzr2r~r r r r rs rc@sDeZdZdZededddgZdddd gZd edd gd fgZd S)TestPositionalsNargsNone1z6Test a Positional with no nargs followed by one with 1rbrrE)rrz--foorza b cza br)rbrN) rrrrr.r_rzr2r~r r r r rs rc@sHeZdZdZedddedgZdddd d gZd edd gd dfgZdS)TestPositionalsNargs2Nonez8Test a Positional with 2 nargs followed by one with nonerbr)rrrz--foorza bza b c dza b crr)rbrN) rrrrr.r_rzr2r~r r r r rsrc@s`eZdZdZededddgZddgZdedgd fd edd gd fd edd d gd fgZdS)"TestPositionalsNargsNoneZeroOrMorez>Test a Positional with no nargs followed by one with unlimitedrbrr)rrz--foor)rbrza brza b crN) rrrrr.r_rzr2r~r r r r rs rc@sTeZdZdZededddgZdddgZd edd gd fd edd d gd fgZdS)!TestPositionalsNargsNoneOneOrMorez@Test a Positional with no nargs followed by one with one or morerbrr)rrz--foorza br)rbrza b crN) rrrrr.r_rzr2r~r r r r rs  rc@sNeZdZdZededddgZdddgZd ed d d fd ed d d fgZd S) TestPositionalsNargsNoneOptionalz@Test a Positional with no nargs followed by one with an Optionalrbrr)rrz--fooza b crN)rbrza br) rrrrr.r_rzr2r~r r r r rs  rc@s`eZdZdZedddedgZddgZdegdd fd edgd d fd edd gd d fgZdS)"TestPositionalsNargsZeroOrMoreNonez@Test a Positional with unlimited nargs followed by one with nonerbr)rrrz--foor)rbrza brza b crN) rrrrr.r_rzr2r~r r r r rs rc@sTeZdZdZedddedgZdddgZd edgd d fd edd gd d fgZdS)!TestPositionalsNargsOneOrMoreNonezBTest a Positional with one or more nargs followed by one with nonerbr)rrrz--foorza br)rbrza b crN) rrrrr.r_rzr2r~r r r r rs  rc@sPeZdZdZeddddedgZddd gZd edd d fd ed d d fgZdS) TestPositionalsNargsOptionalNonezBTest a Positional with an Optional nargs followed by one with nonerbrr)rrrrz--fooza b cr)rbrza brN) rrrrr.r_rzr2r~r r r r rs  rc@s\eZdZdZedddedddgZddd gZd ed d ggd fd ed d gdgd fgZdS)TestPositionalsNargs2ZeroOrMorez=Test a Positional with 2 nargs followed by one with unlimitedrbr)rrrrz--foorza br)rbrza b crN) rrrrr.r_rzr2r~r r r r rs  rc@sLeZdZdZedddedddgZddd d gZd ed d gd gdfgZdS)TestPositionalsNargs2OneOrMorez?Test a Positional with 2 nargs followed by one with one or morerbr)rrrrz--foorza bza b crr)rbrN) rrrrr.r_rzr2r~r r r r rs rc@s\eZdZdZedddedddgZddd d gZd ed d gd dfded d gddfgZd S)TestPositionalsNargs2Optionalz7Test a Positional with 2 nargs followed by one optionalrbr)rrrrz--foorza b c dza brN)rbrza b cr) rrrrr.r_rzr2r~r r r r rs  rc@sjeZdZdZedddedddgZddgZd egd gd fd ed gd gd fd ed d gdgd fgZdS)TestPositionalsNargsZeroOrMore1z=Test a Positional with unlimited nargs followed by one with 1rbr)rrrErz--foor)rbrza brza b crN) rrrrr.r_rzr2r~r r r r rs rc@s\eZdZdZedddedddgZddd gZd ed gd gd fd ed d gdgd fgZdS)TestPositionalsNargsOneOrMore1z?Test a Positional with one or more nargs followed by one with 1rbr)rrrErz--foorza br)rbrza b crN) rrrrr.r_rzr2r~r r r r r s  rc@sVeZdZdZedddedddgZddd gZd ed d gd fd ed dgd fgZd S)TestPositionalsNargsOptional1z?Test a Positional with an Optional nargs followed by one with 1rbr)rrrErz--fooza b crN)rbrza br) rrrrr.r_rzr2r~r r r r rs  rc@sbeZdZdZededddedddgZdd d gZd ed gd gd fded d gdgd fgZdS)#TestPositionalsNargsNoneZeroOrMore1z=Test three Positionals: no nargs, unlimited nargs and 1 nargsrbrr)rrrErz--foorza br)rbrrza b crN) rrrrr.r_rzr2r~r r r r r#s  rc@sheZdZdZededddedddgZdd d d gZd ed d gd gdfded d d gdgdfgZdS)"TestPositionalsNargsNoneOneOrMore1z?Test three Positionals: no nargs, one or more nargs and 1 nargsrbrr)rrrErz--foorrza b cr)rbrrza b c ddN) rrrrr.r_rzr2r~r r r r r2s  rc@sbeZdZdZededdddeddd gZd d d gZd ed ddgdfded ddgdfgZdS)!TestPositionalsNargsNoneOptional1z;Test three Positionals: no nargs, optional narg and 1 nargsrbrrg?)rrrrE)rrz--foorza br)rbrrza b crN) rrrrr.r_rzr2r~r r r r rAs  rc@s`eZdZdZedddeddddgZdd gZd ed dd fd ed dd fded dd fgZd S)$TestPositionalsNargsOptionalOptionalzTest two optional nargsrbr)rrr)rrz--fooza b crN)rbrrza br) rrrrr.r_rzr2r~r r r r rPs rc@speZdZdZedddedddgZdgZded gd fd ed gd fd ed d gd fded d dgd fgZd S)&TestPositionalsNargsOptionalZeroOrMorez1Test an Optional narg followed by unlimited nargsrbr)rrrz--foorN)rbrrza brza b cr) rrrrr.r_rzr2r~r r r r r_src@sfeZdZdZedddedddgZddgZd ed d gd fd ed d gd fded d dgd fgZd S)%TestPositionalsNargsOptionalOneOrMorez3Test an Optional narg followed by one or more nargsrbr)rrrrz--foorN)rbrza brza b cr) rrrrr.r_rzr2r~r r r r rls rc@sLeZdZdZededdgZddddd gZd ed d fd ed d fgZ d S)TestPositionalsChoicesStringz&Test a set of single-character choicesrZabcdefg)rrz--foorrefr)rrN) rrrrr.setr_rzr2r~r r r r rxs  rc@sNeZdZdZedeeddgZddddd gZd e d d fd e dd fgZ dS)TestPositionalsChoicesIntzTest a set of integer choicesr)r:rrz--foorrr4r)r15N) rrrrr.rrr_rzr2r~r r r r r s  r c@sPeZdZdZedddeddddgZddd d d gZd ed d dggdfgZdS)TestPositionalsActionAppendzTest the 'append' actionrr)rr)rrrz--foorza bza b c dza b crr)rN) rrrrr.r_rzr2r~r r r r rs  rc@sneZdZdZedddeddddgZd d gZd ed d dfdedd dfded ddfdedddfgZd S)"TestOptionalsNumericAndPositionalsz;Tests negative number args when numeric options are presentrr)rz-4rr)rrz-2z-315rNF)rrrTz-4 a) rrrrr.r_rzr2r~r r r r rs rc @szeZdZdZedddeddddgZd gZd ed d d fdedd d fdedd d fded dd fdeddd fgZd S)(TestOptionalsAlmostNumericAndPositionalszBTests negative number args when almost numeric options are presentrr)rz-k4rr)rrz-k3rNF)rrz-2rTz-k4 a) rrrrr.r_rzr2r~r r r r rs rc @seZdZedddeddddgZdgZdgedd d fd ged d d fd ged d d fddged dd fdd ged d d fdd ged d d fd ged d d fdged d d fgZd S)$TestEmptyAndSpaceContainingArgumentsrr)rz-yz--yyyr)rrN)rrza badgerz -a badgerz--yyy=a badgerz--yyy=-a badger)rrrr.r_rzr2r~r r r r rs rc @seZdZeddZedddddedd ed d ed d ddgZddgZdedd ddfdedd ddfdedd ddfdeddddfd edd ddfdedd ddfgZ dS) TestPrefixCharacterOnlyArgumentsz-+)r-rrr)rrrrrr)rr:rz-+-rr)rrz-yz+ -rNF)rrrz- Xrz+ -3Tz- ===z===) rrrr.r[rr_rzr2r~r r r r rs rc @seZdZdZedddedddgZgZdedgdfdeggdfd ed ggdfd ed gd gdfd edd gdfd egd gdfded gd gdfgZdS)TestNargsZeroOrMorez?Tests specifying args for an Optional that accepts zero or morez-xr)rrrN)rrz-x arz -x a -- brza -xza -x b) rrrrr.r_rzr2r~r r r r rsrc @seZdZdZededddedgZdddgZd ed gd d fd ed gd d fded dddd gd d fded ddgd d fgZd S)TestNargsRemainderz2Tests specifying a positional with nargs=REMAINDERrrz...)rz-zrz-z ZrN)rrrz-z Z XZz X A B -z ZArz X Y --fooYz--foo) rrrrr.r_rzr2r~r r r r rs rc@seZdZdZededededdeddd gZdd d d d ddddddddg Zdeddgdfdeddgdfdedddgdfdeddgdfdedd gdfd!eddgdfd"edddgdfdedddgdfd#ed$ddgdfd%ed$ddgdfd&edd$dgdfg Z dS)'TestOptionLikez.Tests options that may or may not be argumentsz-x)r:z-3r)r:rrr)rz-y2.5z-xaz-x -az-x -3z-x -3.5z-3 -3.5z-x -2.5z -x -2.5 az-3 -.5za x -1z-x -1 az-3 -1 arN)rrrz-x 2.5g@z-x 2.5 arz-3.5g?z-3-.5gz-3 .5za -3.5za -x 1g?z-x 1 az-3 1 a) rrrrr.floatr_rzr2r~r r r r rs&   rc @seZdZdZeddejdeddejdeddejd gZd gZd e fd e d d fde d dgdfde ddfde d ddfde d dgddfgZ dS)TestDefaultSuppressz%Test actions with suppressed defaultsrbr)rrrrz--bazr)rrz-xrr)rbza br)rbrT)rza --baz)rbrz --baz a b)rbrrN) rrrrr.argparseSUPPRESSr_rzr2r~r r r r rs  rc @seZdZdZeejdZedddedddedd d gZd gZ d e fd e d dfde d dgdfde ddfde d ddfde d dgddfgZ dS)TestParserDefaultSuppressz4Test actions with a parser-level default of SUPPRESS)argument_defaultrbr)rrrz--bazr)rz-xrr)rbza br)rbrT)rza --baz)rbrz --baz a b)rbrrN) rrrrr.rrr[r_rzr2r~r r r r r0s     rc @seZdZdZeddZeddddedd d ed d d ed ddgZdgZdedddddfdedddddfdeddgdddfd edddddfdedddddfdeddgdddfgZ dS)TestParserDefault42z.Test actions with a parser-level default of 42r)r z --versionversionz1.0)rr"rbr)rrrz--bazr)rz-xr)rbrrr"rza brTza --bazz --baz a bN) rrrrr.r[r_rzr2r~r r r r r!Ds    r!c seZdZdZfddZeddZedededd d gZd d d ddgZde dd dgdfde dd ddgdfde ddd gdfde dd dgdfde ddddgdfde dd ddgdfdd d dge d d dgdfgZ Z S)TestArgumentsFromFilez"Test reading arguments from a filecsJtt|dddg}x,|D]$\}}t|d}|||qWdS)N)helloz hello world! ) recursivez -a A @hello)Zinvalidz@no-such-path r')rTr#rr(r)close)r file_textsrtextr,)rUr r r\s  zTestArgumentsFromFile.setUp@)fromfile_prefix_charsz-arrr)rrz-brz@invalidz@missingzX YNr)rrrz X -a A Y Zrrz@hello Xz hello world!zX @helloz-a B @recursive Y ZzX @recursive Z -a Br) rrrrrr.r[r_rzr2r~rXr r )rUr r#Ys  r#csfeZdZdZfddZGdddeZeZeddZ edd d gZ gZ d e d d dgdfgZ ZS)TestArgumentsFromFileConverterz"Test reading arguments from a filecsFtt|dg}x,|D]$\}}t|d}|||qWdS)N)r$z hello world! r')rTr+rr(r)r&)rr'rr(r,)rUr r rs   z$TestArgumentsFromFileConverter.setUpc@seZdZddZdS)z>TestArgumentsFromFileConverter.FromFileConverterArgumentParserccs&x |D]}|sq |Vq WdS)N)r{strip)rarg_lineargr r r convert_arg_line_to_argsszWTestArgumentsFromFileConverter.FromFileConverterArgumentParser.convert_arg_line_to_argsN)rrrr/r r r r FromFileConverterArgumentParsersr0r))r*rr)rz@hello Xr$zworld!r)r)rrrrrrSr0r\r.r[r_rzr2r~rXr r )rUr r+|s  r+c@s4eZdZddZddZddZddZd d Zd S) TestFileTypeReprcCstd}|dt|dS)Nrz FileType('r'))rFileTyperrepr)rr:r r r test_rs zTestFileTypeRepr.test_rcCs tdd}|dt|dS)NwbrEzFileType('wb', 1))rr3rr4)rr:r r r test_wb_1s zTestFileTypeRepr.test_wb_1cCs"tjddd}|dt|dS)Nr2latin_1)encodingz!FileType('r', encoding='latin_1'))rr3rr4)rr:r r r test_r_latinszTestFileTypeRepr.test_r_latincCs$tjdddd}|dt|dS)Nr'big5ignore)r9errorsz/FileType('w', encoding='big5', errors='ignore'))rr3rr4)rr:r r r test_w_big5_ignoresz#TestFileTypeRepr.test_w_big5_ignorecCs$tjdddd}|dt|dS)Nr2rEreplace)r=z"FileType('r', 1, errors='replace'))rr3rr4)rr:r r r test_r_1_replacesz!TestFileTypeRepr.test_r_1_replaceN)rrrr5r7r:r>r@r r r r r1s r1c@seZdZddZddZdS)StdStreamComparercCs ||_dS)N)attr)rrBr r r r1szStdStreamComparer.__init__cCs|tt|jkS)N)rJrHrB)rr=r r r r>szStdStreamComparer.__eq__N)rrrr1r>r r r r rAsrAstdinrBrCc@s eZdZiZddZddZdS)RFilecCs ||_dS)N)r%)rr%r r r r1szRFile.__init__cCs`||jkr|j|}n|}|j|<|t|tsD|d}|j|jkoZ|kSS)Nascii)seenreadr&rGrdecoder%)rr=r(r r r r>s    z RFile.__eq__N)rrrrFr1r>r r r r rDsrDc seZdZdZfddZededededdgZddd gZ d e d e d d fd e e d e dd fde e d e dd fde e e d fde d e dd fgZ ZS) TestFileTypeRz8Test the FileType option/argument type for reading filescsRtt|x4dD],}ttj|j|d}||| qW| ddS)N)rbrr'readonly) rTrIrr(rrrrr)r&r-)r file_namer,)rUr r rs    zTestFileTypeR.setUpz-x)r:rr2rznon-existent-file.txtrbN)rrz -x foo barrz bar -x fooz-x - -rJ)rrrrrr.rr3r_rzr2rDeq_stdinr~rXr r )rUr rIs  rIcsPeZdZdZfddZededddgZdgZ d e e d d fgZ Z S) TestFileTypeDefaultszs     z WFile.__eq__N)rrrrrFr1r>r r r r rQsrQgeteuidznon-root user requiredcseZdZdZfddZededdededdgZddd gZ d e d e d d fd e e d e dd fde e d e dd fde e e d fgZ ZS) TestFileTypeWz8Test the FileType option/argument type for writing filescstt||ddS)NrJ)rTrVrr-)r)rUr r r(szTestFileTypeW.setUpz-xr')r:rrrJrbN)rrz -x foo barrz bar -x fooz-x - -)rrrrrr.rr3r_rzr2rQ eq_stdoutr~rXr r )rUr rV#s  rVc@seZdZededdededdgZddgZdede ddfd ee de d dfd ee de d dfd ee e dfgZ dS) TestFileTypeWBz-xr6)r:rrrbN)rrz -x foo barrz bar -x fooz-x - -) rrrr.rr3r_rzr2rQrWr~r r r r rX9srXc@seZdZdZddZdS)TestFileTypeOpenArgsz0Test that open (the builtin) is correctly calledc Cstj}|ddf|dddf|ddddf|d d d d f|dd dddfg}td.}x&|D]\}}|d|jd|q^WWdQRXdS)NrP)rPNNr'rE)r'rENNr?)r=)r'rZNr?r6r;)r9)r6rZr;Nrl1strict)r'rr[r\z builtins.openrb)rb)rr3rZpatchZassert_called_with)rZFTZcasesmr:r/r r r test_open_argsKs   z#TestFileTypeOpenArgs.test_open_argsN)rrrrr^r r r r rYHsrYc@sbeZdZdZededededgZddddgZd e d d d fd e ddd fde ddd fgZ dS)TestTypeCallablez,Test some callables as option/argument typesz--eggs)r:rrZ42jz--eggs az --eggs 2iz --eggs=42 42rgE@)eggsrz--eggs 2j -- -1.5y@gz1024.675Ng3333@) rrrrr.complexrr_rzr2r~r r r r r_Zs  r_c@sleZdZdZGdddeZededededgZgZde eded d fd e ed ed d fgZ dS)TestTypeUserDefinedz(Test a user-defined option/argument typec@seZdZddZddZdS)zTestTypeUserDefined.MyTypecCs ||_dS)N)value)rrcr r r r1nsz#TestTypeUserDefined.MyType.__init__cCst||jft||jfkS)N)r:rc)rr=r r r r>qsz!TestTypeUserDefined.MyType.__eq__N)rrrr1r>r r r r MyTypelsrdz-x)r:rza -x brr)rrz-xf grrN) rrrrr rdr.r_rzr2r~r r r r rbis rbc@sjeZdZdZGdddZededededgZgZdeeded d fd eed ed d fgZ dS)TestTypeClassicClasszTest a classic class typec@seZdZddZddZdS)zTestTypeClassicClass.CcCs ||_dS)N)rc)rrcr r r r1szTestTypeClassicClass.C.__init__cCst||jft||jfkS)N)r:rc)rr=r r r r>szTestTypeClassicClass.C.__eq__N)rrrr1r>r r r r Csrfz-x)r:rza -x brr)rrz-xf grrN) rrrrrfr.r_rzr2r~r r r r res rec@seZdZdZddZdS)TestTypeRegistrationz*Test a user-defined type by registering itcCszdd}t}|dd||jddd|jddd||dtdd d ||d td d d dS) NcSsd|S)Nz my_type{%s}r )stringr r r get_my_typesz.TestTypeRegistration.test..get_my_typer:Zmy_typez-x)r:rrz my_type{1})rrz-x 1 42z my_type{42})rArgumentParserregisterr]rrOr{r2)rrir^r r r testszTestTypeRegistration.testN)rrrrrlr r r r rgsrgc@seZdZdZGdddejZGdddejZeddee dd ed ee d d d gZ gZ de dd dfde dddfde dddfde dddfgZdS)TestActionUserDefinedz*Test a user-defined option/argument actionc@seZdZdddZdS)z$TestActionUserDefined.OptionalActionNcCsy~|jdkstd|j|dks.td|tdd}|dkrHd|_n|d krXd |_n td |||ks|td ||fWn,tk rtd }td|YnXt|d|dS)Nrzdest: %sz-szflag: %sg?)r)g?g?r)g@rz value: %szexpected %s, got %srEzopt_action failed: %s)rAssertionErrorr2rrHrLr?rK)rr^ namespacerc option_stringrer r r __call__s    z-TestActionUserDefined.OptionalAction.__call__)N)rrrrrr r r r OptionalActionsrsc@seZdZdddZdS)z&TestActionUserDefined.PositionalActionNcCsy|dkstd||jdks.td|jtdd}|dkrHd|_n,|dkrXd |_n|d krhd |_n td |||kstd ||fWn,tk rtd}td|YnXt|d|dS)Nzoption_string: %srzdest: %sr)r)rrg?)rEg?)rg?z value: %szexpected %s, got %srEzarg_action failed: %s)rnrr2rrHrLr?rK)rr^rorcrprrqr r r rrs$     z/TestActionUserDefined.PositionalAction.__call__)N)rrrrrr r r r PositionalActionsrtz-srg?)rrr:rrrr)rr:rrz-s0.125g?)rrrrz -s 0.625 1g?rEz84 -s2g@rN)rrrrrActionrsrtr.rrr_rzr2r~r r r r rmsrmc@s*eZdZdZGdddejZddZdS)TestActionRegistrationz5Test a user-defined action supplied by registering itc@seZdZdddZdS)zTestActionRegistration.MyActionNcCst||jd|dS)Nzfoo[%s])rKr)rr^rovaluesrpr r r rrsz(TestActionRegistration.MyAction.__call__)N)rrrrrr r r r MyActionsrxcCs^t}|dd|j|jddd||dgtdd||dgtd ddS) NrZ my_actionr)rrzfoo[1])rrzfoo[42])rrjrkrxr]rrOr2)rr^r r r rls zTestActionRegistration.testN)rrrrrrurxrlr r r r rvsrvcseZdZdZddZd4ddZfdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3ZZS)5TestAddSubparserszTest the add_subparsers methodcOs|jtf||dS)N)r|r?)rr/r0r r r assertArgumentParserErrorsz+TestAddSubparsers.assertArgumentParserErrorFNc Csz|r0tdd|d}|j|dddddd ntddd }|jd ddd |jd td dddi}|rzd|d<d|d<nd|d<|jf|}||jtdd}|rd|d<|rddg|d<|jd7|}|jdtdd|jd d!d"d#td$d} |rd%| d<|jd8| } | jd'd(d)d#| jd*td+d,d-td.d} |rJd/| d<|jd9| } | jd1td2d| jd3d4d5d6|S):NPROGzmain description)prog descriptionrrrrbrzfoo help)rhelp)r|r}z--foorzbar help)r:r~rFZCOMMANDmetavarZcommandstitlez command helpr~z 1 description)r}z1 helpZ1alias1Z1alias2aliasesrz-wzw helprrzx help)rr~z 2 descriptionz2 helprz-yZ123zy helprrzz help)r:rr~z 3 descriptionz3 helprtzt helpuz...zu help)rr~)r)r)r) rSr]radd_subparsersrzdict add_parserrra) rsubparser_helprrr^Zsubparsers_kwargs subparsersZparser1_kwargsparser1Zparser2_kwargsparser2Zparser3_kwargsZparser3r r r rx sN              zTestAddSubparsers._get_parsercs&t||_|jdd|_dS)NT)r)rTrrxr^command_help_parser)r)rUr r rBs  zTestAddSubparsers.setUpcCs*x$dD]}|}||jj|qWdS)N)rrza az0.5 az0.5 1z0.5 1 -yz0.5 2 -w)r{rzr^rO)rr}r/r r r test_parse_args_failuresGs z*TestAddSubparsers.test_parse_args_failuresc Cs||jdtddddd||jdtdd d d d gd ||jdtddddd||jdtddddddddgddS)Nz 0.5 1 b -w 7Fg?r)rbrr'rz0.25 --foo 2 -y 2 3j -- -1jTg?ry@y)rbrrrz--foo 0.125 1 cg?rz-1.5 3 11 -- a --foo 7 -- bg rz--foorz--)rbrrr)rr^rOr{r2)rr r r test_parse_argsNsz!TestAddSubparsers.test_parse_argsc Cs||jdtdddddgf||jdtddddddgf||jd tddddddgf||jd tdddddd d gf||jd tdddddddddgfdS)Nz 0.5 1 b -w 7Fg?rr)rbrr'rz0.5 -p 1 b -w 7z-pz0.5 1 b -w 7 -pz0.5 1 b -q -rs -w 7z-qz-rsz0.5 -W 1 b -X Y -w 7 Zz-Wz-Xrr)rr^parse_known_argsr{r2)rr r r test_parse_known_argsasz'TestAddSubparsers.test_parse_known_argscCsXt}|jddd|jdd}|d}|d|tddd d |d dS) Nz--foor)rr)rrrFr)rbrrz1 2)rSr]rrrr2rOr{)rr^rrr r r test_destws   zTestAddSubparsers.test_destcCs,|dg}||jd||jddS)Nrunr )rOrcommandrz)rr^retr r r _test_required_subparserss z+TestAddSubparsers._test_required_subparserscCs0t}|jdd}d|_|d||dS)Nr)rTr)rSrrrr)rr^rr r r &test_required_subparsers_via_attributes   z8TestAddSubparsers.test_required_subparsers_via_attributecCs,t}|jddd}|d||dS)NrT)rrr)rSrrr)rr^rr r r "test_required_subparsers_via_kwargs z4TestAddSubparsers.test_required_subparsers_via_kwargcCs6t}|jdd}|d|d}||jdS)Nr)rrr )rSrrrO assertIsNoner)rr^rrr r r test_required_subparsers_defaults    z2TestAddSubparsers.test_required_subparsers_defaultcCs8t}|jddd}|d|d}||jdS)NrF)rrrr )rSrrrOrr)rr^rrr r r test_optional_subparserss   z*TestAddSubparsers.test_optional_subparserscCs.||jd||jtddS)Nz)usage: PROG [-h] [--foo] bar {1,2,3} ... aJ usage: PROG [-h] [--foo] bar {1,2,3} ... main description positional arguments: bar bar help {1,2,3} command help optional arguments: -h, --help show this help message and exit --foo foo help )rr^ format_usage format_helptextwrapdedent)rr r r test_helps  zTestAddSubparsers.test_helpcCs6|jdd}||d||tddS)Nz+:-)rz)usage: PROG [-h] [++foo] bar {1,2,3} ... aJ usage: PROG [-h] [++foo] bar {1,2,3} ... main description positional arguments: bar bar help {1,2,3} command help optional arguments: -h, --help show this help message and exit ++foo foo help )rxrrrrr)rr^r r r test_help_extra_prefix_charss    z.TestAddSubparsers.test_help_extra_prefix_charscCs6tddd}|jdddd||tddS) Nr{zmain description)r|r}z--non-breakingruRhelp message containing non-breaking spaces shall not wrap at non-breaking spaces)rr~uB usage: PROG [-h] [--non-breaking] main description optional arguments: -h, --help show this help message and exit --non-breaking help message containing non-breaking spaces shall not wrap at non-breaking spaces )rSr]rrrr)rr^r r r test_help_non_breaking_spacess  z/TestAddSubparsers.test_help_non_breaking_spacescCs6|jdd}||d||tddS)Nz+:/)rz)usage: PROG [+h] [++foo] bar {1,2,3} ... aJ usage: PROG [+h] [++foo] bar {1,2,3} ... main description positional arguments: bar bar help {1,2,3} command help optional arguments: +h, ++help show this help message and exit ++foo foo help )rxrrrrr)rr^r r r test_help_alternate_prefix_charss    z2TestAddSubparsers.test_help_alternate_prefix_charscCs.||jd||jtddS)Nz)usage: PROG [-h] [--foo] bar {1,2,3} ... a usage: PROG [-h] [--foo] bar {1,2,3} ... main description positional arguments: bar bar help {1,2,3} command help 1 1 help 2 2 help 3 3 help optional arguments: -h, --help show this help message and exit --foo foo help )rrrrrr)rr r r test_parser_command_helps   z*TestAddSubparsers.test_parser_command_helpcCsxtddd}|jdddd|jdd d |jd d d d}|d}|d}||d||tddS)Nr{zmain description)r|r}z--foorzfoo help)rr~rzbar help)r~ subcommandsz command helpzadditional text)rr}r~rrz'usage: PROG [-h] [--foo] bar {1,2} ... a usage: PROG [-h] [--foo] bar {1,2} ... main description positional arguments: bar bar help optional arguments: -h, --help show this help message and exit --foo foo help subcommands: command help {1,2} additional text ) rSr]rrrrrrr)rr^rrrr r r test_subparser_title_helps   z+TestAddSubparsers.test_subparser_title_helpc Cs:|t}|j|WdQRX|||jjdS)N)r|r?r^rOr{r exceptionrB)rr}Z expected_helpcmr r r _test_subparser_help s z&TestAddSubparsers._test_subparser_helpcCs|dtddS)Nz5.0 1 -ha usage: PROG bar 1 [-h] [-w W] {a,b,c} 1 description positional arguments: {a,b,c} x help optional arguments: -h, --help show this help message and exit -w W w help )rrr)rr r r test_subparser1_help%s z&TestAddSubparsers.test_subparser1_helpcCs|dtddS)Nz5.0 2 -ha# usage: PROG bar 2 [-h] [-y {1,2,3}] [z [z ...]] 2 description positional arguments: z z help optional arguments: -h, --help show this help message and exit -y {1,2,3} y help )rrr)rr r r test_subparser2_help3s z&TestAddSubparsers.test_subparser2_helpc Cs\|jdd}||dtdddddgf||dtdddddgfdS) NT)rz 0.5 1alias1 bFg?r)rbrr'rz 0.5 1alias2 b)rxrrr{r2)rr^r r r test_alias_invocationAs   z'TestAddSubparsers.test_alias_invocationcCs"|jdd}||jddS)NT)rz 0.5 1alias3 b)rxrzrOr{)rr^r r r test_error_alias_invocationLs z-TestAddSubparsers.test_error_alias_invocationcCs.|jddd}d|_||tddS)NT)rra usage: PROG [-h] [--foo] bar COMMAND ... main description positional arguments: bar bar help optional arguments: -h, --help show this help message and exit --foo foo help commands: COMMAND 1 (1alias1, 1alias2) 1 help 2 2 help 3 3 help )rxZmaxDiffrrrr)rr^r r r test_alias_helpQsz!TestAddSubparsers.test_alias_help)FNF)rrrrrzrxrrrrrrrrrrrrrrrrrrrrrrrXr r )rUr rys4 6   ryc@s(eZdZdZddZddZddZdS) TestPositionalsGroupsz@Tests that order of group positionals matches construction ordercCsZt}|d|d}|d|dtdddd}|d }|||dS) Nrbrrrrrr)rbrrz1 2 3)rSr]rcr2rOr{r)rr^rdexpectedrPr r r test_nongroup_firstos    z)TestPositionalsGroups.test_nongroup_firstcCsZt}|d}|d|d|dtdddd}|d }|||dS) Nxxxrbrrrrr)rbrrz1 2 3)rSrcr]r2rOr{r)rr^rdrrPr r r test_group_firstys    z&TestPositionalsGroups.test_group_firstcCspt}|d}|d|d|d|d}|dtddd d d }|d }|||dS) Nrrbrrrfrellrrrr )rbrrrz1 2 3 4)rSrcr]r2rOr{r)rr^rdrrPr r r test_interleaved_groupss      z-TestPositionalsGroups.test_interleaved_groupsN)rrrrrrrr r r r rls  rcseZdZdZddZfddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZZS) TestParentParsersz5Tests that parsers can be created with parent parserscOs|jtf||dS)N)r|r?)rr/r0r r r rzsz+TestParentParsers.assertArgumentParserErrorcsttdd|_|jd|jd}|d|jdtdd|_|jd|jd|jd }|d tdd|_|jdtdd|_|jdtdd|_ |j }|jd d d |jdd d t j tjd|_dS)NF)rz--wrz-yrrz-brz--dz-ar)rr)rTrrS wxyz_parentr]rc abcd_parentw_parentz_parentab_mutex_parentadd_mutually_exclusive_grouprrbasenamerHrj main_program)rZx_groupc_grouprd)rUr r rs(                 zTestParentParsers.setUpcCs2t|jgd}||dtdddddS)N)parentsz -y 1 2 --w 3rrr)r'rr)rSrrrOr{r2)rr^r r r test_single_parentsz$TestParentParsers.test_single_parentcCs,||jjt|jgd}||jdS)N)r)_test_mutex_abrrOrS)rr^r r r test_single_parent_mutexsz*TestParentParsers.test_single_parent_mutexcCs0|jg}td|d}t|gd}||jdS)NF)rr)r)rrSrrO)rrr^r r r test_single_granparent_mutexs  z.TestParentParsers.test_single_granparent_mutexcCs||gtddd||dgtddd||dgtddd||ddg||ddg||dg||ddg||ddgdS)NF)rrz-aTz-bz-c)rr2rz)rrOr r r rsz TestParentParsers._test_mutex_abc Cs@|j|jg}t|d}||dtddddddddS)N)rz--d 1 --w 2 3 4rrrr )rrrr'rr)rrrSrrOr{r2)rrr^r r r test_multiple_parentss  z'TestParentParsers.test_multiple_parentsc Csb|j|jg}t|d}||dtdddddd||jd||jd dS) N)rz -a --w 2 3TFrr)rrr'rrz -a --w 2 3 -bz -a -b --w 2 3)rrrSrrOr{r2rz)rrr^r r r test_multiple_parents_mutexs  z-TestParentParsers.test_multiple_parents_mutexcCs |jtjtj|j|jgddS)N)r)r|r ArgumentErrorrjrr)rr r r test_conflicting_parentssz*TestParentParsers.test_conflicting_parentscCs |jtjtj|j|jgddS)N)r)r|rrrjrr)rr r r test_conflicting_parents_mutexsz0TestParentParsers.test_conflicting_parents_mutexcCs:|j|jg}t|d}||dtdddddS)N)rz1 2r)r'rr)rrrSrrOr{r2)rrr^r r r test_same_argument_name_parentss  z1TestParentParsers.test_same_argument_name_parentsc CsPt}|}|jd|jgd}|d||dtddddd dS) Nr)rrqzbar -b 1 --d 2 3 4rrrr )rrrrq) rSrrrr]rrOr{r2)rr^rZ abcde_parserr r r test_subparser_parentss  z(TestParentParsers.test_subparser_parentsc Cst}|}|jg}|jd|d}|d}|d|j|jg}|jd|d}|d||d t dd d d ||d t d dd ddd d| |jd | |jd dS)Nrb)rrrrrqzfoo -a 4TFr )rrrzbar -b --w 2 3 4rr)rrr'rrrqz foo -a -b 4z bar -b -a 4) rSrrrrcr]rrrOr{r2rz)rr^rrZ abc_parserrZwxyzabe_parserr r r test_subparser_parents_mutexs"    z.TestParentParsers.test_subparser_parents_mutexc CsJ|j|jg}t|d}|}|j}||td||r|jddj}t}x&|jdgD]}||||q WdS)NT)rr)rrOr?rzr|r{)rrOrWrr r r test_failures_when_requiredt sz#MEMixin.test_failures_when_requiredcCsH|jddj}|j|j}x(|D] \}}||}|||q WdS)NF)r)rrOr~successes_when_not_requiredr{r)rrOr~rr actual_nsr r r test_successes_when_not_requiredz s   z(MEMixin.test_successes_when_not_requiredcCs>|jddj}x*|jD] \}}||}|||qWdS)NT)r)rrOr~r{r)rrOrrrr r r test_successes_when_required s z$MEMixin.test_successes_when_requiredcCs,|jddj}|j}||t|dS)NF)r)rrusage_when_not_requiredrrr)rrexpected_usager r r test_usage_when_not_required sz$MEMixin.test_usage_when_not_requiredcCs,|jddj}|j}||t|dS)NT)r)rrusage_when_requiredrrr)rrrr r r test_usage_when_required sz MEMixin.test_usage_when_requiredcCs2|jddj}|j|j}||t|dS)NF)r)rrrr~rrr)rrr~r r r test_help_when_not_required s z#MEMixin.test_help_when_not_requiredcCs2|jddj}|j|j}||t|dS)NT)r)rrrr~rrr)rrr~r r r test_help_when_required s zMEMixin.test_help_when_requiredN) rrrrrrrrrrrr r r r rl src@sxeZdZdddZddgZdedddfd ed ddfd edd dfd edd dfgZdedddfgZdZdZ dZ dS)TestMutuallyExclusiveSimpleNcCs:tdd}|j|d}|jddd|jddd d d |S) Nr{)r|)rz--barzbar help)r~z--bazrrzbaz help)rrr~)rSrr])rrr^rdr r r r s   z&TestMutuallyExclusiveSimple.get_parserz--bar X --baz Yz --bar X --bazz--bar Xr)rrz--bar X --bar Zrz--baz Yrz--bazrz; usage: PROG [-h] [--bar BAR | --baz [BAZ]] z; usage: PROG [-h] (--bar BAR | --baz [BAZ]) z optional arguments: -h, --help show this help message and exit --bar BAR bar help --baz [BAZ] baz help )N) rrrrrzr2r~rrrr~r r r r r s rc @seZdZdddZdgZdedddddfded ddddfd edddddfd edd dddfgZd edddddfgZd ZdZ dZ dS)TestMutuallyExclusiveLongNcCsRtdd}|jddd|jddd|j|d}|jd d d|jd d d|S) Nr{)r|z--abcdez abcde help)r~z--fghijz fghij help)rz--klmnoz klmno helpz--pqrstz pqrst help)rSr]r)rrr^rdr r r r s  z$TestMutuallyExclusiveLong.get_parserz--klmno X --pqrst Yz --klmno Xr)ZabcdeZfghijZklmnoZpqrstz--abcde Y --klmno Xrz --pqrst Xz--pqrst X --fghij Yrzi usage: PROG [-h] [--abcde ABCDE] [--fghij FGHIJ] [--klmno KLMNO | --pqrst PQRST] zi usage: PROG [-h] [--abcde ABCDE] [--fghij FGHIJ] (--klmno KLMNO | --pqrst PQRST) z optional arguments: -h, --help show this help message and exit --abcde ABCDE abcde help --fghij FGHIJ fghij help --klmno KLMNO klmno help --pqrst PQRST pqrst help )N) rrrrrzr2r~rrrr~r r r r r s  rc@sfeZdZddZdgZdedddfded ddfd ed d dfgZd ed ddfgZdZdZ dZ d S)$TestMutuallyExclusiveFirstSuppressedcCs:tdd}|j|d}|jdtjd|jdddd |S) Nr{)r|)rz-x)r~z-yrzy help)rr~)rSrr]rr)rrr^rdr r r r s   z/TestMutuallyExclusiveFirstSuppressed.get_parserz-x X -yz-x XrT)rrz -x X -x Yrz-yNFrz& usage: PROG [-h] [-y] z$ usage: PROG [-h] -y zx optional arguments: -h, --help show this help message and exit -y y help ) rrrrrzr2r~rrrr~r r r r r src @seZdZddZdddgZdedddd fd ed d dd fd ed dd d fdedddd fgZded ddd fgZdZZ dZ dS)#TestMutuallyExclusiveManySuppressedcCsNtdd}|j|d}|j}|ddtjd|ddtjd|d tjd |S) Nr{)r|)rz--spamr)rr~z--badgerrz --bladder)r~)rSrr]rr)rrr^rdrTr r r r s  z.TestMutuallyExclusiveManySuppressed.get_parserz--spam --badgerz--badger --bladder Bz--bladder B --spamz--spamTN)rrZbladderz--badgerFz --bladder Brz --spam --spamrz! usage: PROG [-h] z[ optional arguments: -h, --help show this help message and exit ) rrrrrzr2r~rrrr~r r r r r s rc @seZdZddZdddddgZded d d d fd eddd d fd edd d gd fdedd d ddgd fgZdedd d d fgZdZdZ dZ d S)*TestMutuallyExclusiveOptionalAndPositionalcCsJtdd}|j|d}|jdddd|jdd d |jd d d dd|S)Nr{)r|)rz--foorFOO)rr~z--spamSPAM)r~rrrBADGER)rrr~)rSrr])rrr^rdr r r r1 s   z5TestMutuallyExclusiveOptionalAndPositional.get_parserz--foo --spam Sz --spam S XzX --foozX Y Z --spam Sz --foo X Yz--fooTNr)rbrrz--spam SFSzX Y ZrrrzM usage: PROG [-h] [--foo | --spam SPAM | badger [badger ...]] zM usage: PROG [-h] (--foo | --spam SPAM | badger [badger ...]) z positional arguments: badger BADGER optional arguments: -h, --help show this help message and exit --foo FOO --spam SPAM SPAM ) rrrrrzr2r~rrrr~r r r r r/ s  rc @seZdZddZddddgZdedd d d d d fd ed dd d d d fd ed d dd d d fd edd d dd d fded dd d dd fded d dddd fgZded d d d d d fded d d dd d fded d d d dd fgZdZZ dZ dS)#TestMutuallyExclusiveOptionalsMixedcCsjtdd}|jdddd|j|d}|jddd d|jd dd d|jd dd d|jdddd|S)Nr{)r|z-xrzx help)rr~)rz-aza helpz-bzb helpz-yzy helpz-czc help)rSr]r)rrr^rdr r r r^ s  z.TestMutuallyExclusiveOptionalsMixed.get_parserz-a -bz-b -cz-a -cz-a -b -cz-aTF)rrrrrz-bz-cz-a -xz-y -bz-x -y -crz-xz-yz: usage: PROG [-h] [-x] [-a] [-b] [-y] [-c] z optional arguments: -h, --help show this help message and exit -x x help -a a help -b b help -y y help -c c help N) rrrrrzr2r~rrrr~r r r r r\ s  rc@s\eZdZdddZddgZdedddfd edd dfgZd edddfgZd Zd Z dZ dS)TestMutuallyExclusiveInGroupNcCsDtdd}|jddd}|j|d}|jddd |jd d d |S) Nr{)r|z Titled groupzGroup description)rr})rz--barzbar help)r~z--bazzbaz help)rSrcrr])rrr^Z titled_group mutex_groupr r r r s   z'TestMutuallyExclusiveInGroup.get_parserz--bar X --baz Yz--baz X --bar Yz--bar Xr)rrz--baz Yrrz9 usage: PROG [-h] [--bar BAR | --baz BAZ] z9 usage: PROG [-h] (--bar BAR | --baz BAZ) z optional arguments: -h, --help show this help message and exit Titled group: Group description --bar BAR bar help --baz BAZ baz help )N) rrrrrzr2r~rrrr~r r r r r s  rc @seZdZddZdddgZdedddd dd fd ed d dd dd fded dd d dd fdedddd d d fded d dd d d fgZd ed ddd dd fded ddd d d fgZdZZ dZ d S)1TestMutuallyExclusiveOptionalsAndPositionalsMixedcCshtdd}|jddd|jdddd |j|d }|jd d d d|jdddd |jdddd |S)Nr{)r|rzx help)r~z-yrzy help)rr~)rrrza help)rr~z-bzb helpz-czc help)rSr]r)rrr^rdr r r r s  ztdd}|jddt|gd}|tdd|gdS)NF)rrb)r)r)rSrrr2rO)rrr^r r r test_set_defaults_parents8 s   z)TestSetDefaults.test_set_defaults_parentscCsLt}|d}|jdd|jdd|tdd|dgdS)NrrE)rbr)rrjrrrrr2rO)rr^Zxparserr r r )test_set_defaults_on_parent_and_subparser> s   z9TestSetDefaults.test_set_defaults_on_parent_and_subparsercCst}|jddddd|d|jddd |jd d d |jd d dd|tddddd|g|jddddd|tddddd|gdS)NWrrr)r'rrrz-wz-xXX)rrr)rrZZ)rrWWYY)rSrr]rr2rO)rr^r r r &test_set_defaults_same_as_add_argumentE s  z6TestSetDefaults.test_set_defaults_same_as_add_argumentcCst}|jddddd|d}|d|jdd d |jd d d |jdd dd|tdd ddd|g|jddddd|tddddd|gdS)Nrrrr)r'rrrrbz-wz-xr )rrr)rrr )rrr r )rSrrcr]rr2rO)rr^rdr r r ,test_set_defaults_same_as_add_argument_groupV s   z|dkr dSGdddt}xdD]\}}||||q"WdS)N HelpTestCasec@s<eZdZddZddZddZddZd d Zd d Zd S)z6TestHelpFormattingMetaclass.__init__..AddTestsc Ssp||_||_x^|j|j|jgD]J}d|j|f}|fdd}y ||_Wntk rZYnXt|||qWdS)Nz%s_%scSs ||dS)Nr )rrmr r r test_wrapper szUTestHelpFormattingMetaclass.__init__..AddTests.__init__..test_wrapper) func_suffixstd_name test_format test_printtest_print_filerrsrK)rZ test_classrrrmrvrr r r r1 s   z?TestHelpFormattingMetaclass.__init__..AddTests.__init__c Sstj|jj|jj}x$t|dgD]}|j|j|jq"Wt|dg}x>|D]6\}}|j|j|j}x|D]}|j|j|jqjWqLWt|dg}|r|} x|D]} | j | j| jqW|S)Nr_argument_group_signaturessubparsers_signatures) rrjr[r/r0rJr]rcrr) rrwr^Z argument_sigZ group_sigsZ group_sigZ argument_sigsrdZsubparsers_sigsrZ subparser_sigr r r rx s(       zBTestHelpFormattingMetaclass.__init__..AddTests._get_parsercSs&t||j}t|}|||dS)N)rJrrrr)rrw parser_textZ expected_textr r r _test s  z.AddTests._testcSs,||}t|d|j}|||dS)Nz format_%s)rxrJrr$)rrwr^rr r r r s zBTestHelpFormattingMetaclass.__init__..AddTests.test_formatc Ssr||}t|d|j}tt|j}tt|jtz|tt|j}Wdtt|j|X|||dS)Nzprint_%s) rxrJrrHrrKrrNr$)rrwr^print_Z old_streamr#r r r r s  zATestHelpFormattingMetaclass.__init__..AddTests.test_printcSs@||}t|d|j}t}|||}|||dS)Nzprint_%s)rxrJrrrNr$)rrwr^r%Zsfiler#r r r r  s  zFTestHelpFormattingMetaclass.__init__..AddTests.test_print_fileN) rrrr1rxr$rrr r r r r r s  r))usagerB)r~rB)r)rr%rrrrrr r r r1 s Hz$TestHelpFormattingMetaclass.__init__N)rrrr1r r r r r srrc@sneZdZdZeddddZedddd d ed d d dedddedddedddgZgZdZedZ dZ dS)TestHelpBiggerOptionalsz;Make sure that argument help aligns when options are longerr{ DESCRIPTIONEPILOG)r|r}epilogz-vz --versionr"z0.1)rr"z-xrzX HELP)rr~z--yzY HELP)r~rbzFOO HELPrzBAR HELPz; usage: PROG [-h] [-v] [-x] [--y Y] foo bar ab DESCRIPTION positional arguments: foo FOO HELP bar BAR HELP optional arguments: -h, --help show this help message and exit -v, --version show program's version number and exit -x X HELP --y Y Y HELP EPILOG z 0.1 N) rrrrr.r[r_r!r&r~r"r r r r r' s   r'c@s<eZdZdZddZejZejZejZdZ e dZ ej Z dS)TestShortColumnszTest extremely small number of columns. TestCase prevents "COLUMNS" from being too small in the tests themselves, but we don't want any exceptions thrown in such cases. Only ugly representation. cCs$t}|dd||jdS)Nr r )rr rrr)rrr r r r s zTestShortColumns.setUpz usage: PROG [-h] [-v] [-x] [--y Y] foo bar a DESCRIPTION positional arguments: foo FOO HELP bar BAR HELP optional arguments: -h, --help show this help message and exit -v, --version show program's version number and exit -x X HELP --y Y Y HELP EPILOG N) rrrrrr'r[r_r!r&r~r"r r r r r+ s r+c@seZdZdZeddddZedddd d ed d d dedddedddedddgZedddedddeddddgfgZdZed Z d!Z d"S)#TestHelpBiggerOptionalGroupsz;Make sure that argument help aligns when options are longerr{r(r))r|r}r*z-vz --versionr"z0.1)rr"z-xrzX HELP)rr~z--yzY HELP)r~rbzFOO HELPrzBAR HELPz GROUP TITLEzGROUP DESCRIPTION)r}rzBAZ HELPz-zrzZ HELP)rr~zN usage: PROG [-h] [-v] [-x] [--y Y] [-z Z [Z ...]] foo bar baz a DESCRIPTION positional arguments: foo FOO HELP bar BAR HELP optional arguments: -h, --help show this help message and exit -v, --version show program's version number and exit -x X HELP --y Y Y HELP GROUP TITLE: GROUP DESCRIPTION baz BAZ HELP -z Z [Z ...] Z HELP EPILOG z 0.1 N) rrrrr.r[r_r!r&r~r"r r r r r,H s     r,c@s^eZdZdZedddZedddded d d ed d d eddd gZgZdZedZ dZ dS)TestHelpBiggerPositionalsz4Make sure that help aligns when arguments are longerUSAGEr()r&r}z-xrzX HELP)rr~z--yzY HELP)r~ZekiekiekifekangzEKI HELPrzBAR HELPz usage: USAGE a  DESCRIPTION positional arguments: ekiekiekifekang EKI HELP bar BAR HELP optional arguments: -h, --help show this help message and exit -x X HELP --y Y Y HELP rN) rrrrr.r[r_r!r&r~r"r r r r r-w s     r-c@sfeZdZdZedddZedddded d d dgZed d deddddgfgZdZedZ dZ dS)TestHelpReformattingz>Make sure that text after short names starts on the first liner{ze oddly formatted description that is so long that it should go onto multiple lines when wrapped)r|r}z-xr zoddly formatted -x help)rr~rrz normal y helprz' oddly formatted group description)r}z-arz` oddly formatted -a help again, so long that it should be wrapped over multiple lines)rr~z2 usage: PROG [-h] [-x XX] [-a] yyy a oddly formatted description that is so long that it should go onto multiple lines when wrapped positional arguments: yyy normal y help optional arguments: -h, --help show this help message and exit -x XX oddly formatted -x help title: oddly formatted group description -a oddly formatted -a help again, so long that it should be wrapped over multiple lines rN) rrrrr.r[r_r!r&r~r"r r r r r/ s  r/c@sbeZdZdZedddZedddded d d dgZed ed dddgfgZdZedZ dZ dS)TestHelpWrappingShortNamesz>Make sure that text after short names starts on the first liner{zZD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD D)r|r}z-xr zxXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HX)rr~rrzdYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHALPHASz-arzPAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHA)rr~z2 usage: PROG [-h] [-x XX] [-a] yyy a D DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD D positional arguments: yyy YH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YH optional arguments: -h, --help show this help message and exit -x XX XHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HX ALPHAS: -a AHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHA rN) rrrrr.r[r_r!r&r~r"r r r r r0 s  r0c@s|eZdZdZedddZeddddd ed d d d edddd gZededddd edddd gfgZdZedZ dZ dS)TestHelpWrappingLongNameszr?z usage: PROG [-h] [-x X] [-y Y] [-z Z] aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb ccccccccccccccccccccccccc a positional arguments: aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb ccccccccccccccccccccccccc optional arguments: -h, --help show this help message and exit -x X -y Y -z Z rN) rrrrr.r[r_r!r&r~r"r r r r rA s   rAc@sbeZdZdZeddZedddeddded d ded ed ed gZgZdZedZ dZ dS)%TestHelpUsageOptionalsPositionalsWrapzr?a usage: PROG [-h] [-x XXXXXXXXXXXXXXXXXXXXXXXXX] [-y YYYYYYYYYYYYYYYYYYYYYYYYY] [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ] aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb ccccccccccccccccccccccccc ae positional arguments: aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb ccccccccccccccccccccccccc optional arguments: -h, --help show this help message and exit -x XXXXXXXXXXXXXXXXXXXXXXXXX -y YYYYYYYYYYYYYYYYYYYYYYYYY -z ZZZZZZZZZZZZZZZZZZZZZZZZZ rN) rrrrr.r[r_r!r&r~r"r r r r rBs      rBc@sPeZdZdZeddZedddeddded d dgZgZd Zed Z d Z dS)TestHelpUsageOptionalsOnlyWrapz@Test usage messages where there are only optionals and they wrapr{)r|z-xr3)rz-yr:z-zr;z usage: PROG [-h] [-x XXXXXXXXXXXXXXXXXXXXXXXXX] [-y YYYYYYYYYYYYYYYYYYYYYYYYY] [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ] z optional arguments: -h, --help show this help message and exit -x XXXXXXXXXXXXXXXXXXXXXXXXX -y YYYYYYYYYYYYYYYYYYYYYYYYY -z ZZZZZZZZZZZZZZZZZZZZZZZZZ rN) rrrrr.r[r_r!r&r~r"r r r r rC9s   rCc@sFeZdZdZedddZedededgZgZdZed Z d Z d S) TestHelpUsagePositionalsOnlyWrapzBTest usage messages where there are only positionals and they wrapr{F)r|rr=r>r?z~ usage: PROG aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb ccccccccccccccccccccccccc z positional arguments: aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb ccccccccccccccccccccccccc rN) rrrrr.r[r_r!r&r~r"r r r r rDSs  rDc @seZdZdZeddZededdeddd d d d ed dddeddddgdddedddeddddgZeded d!ded"d#d$dgfgZd%Z e d&Z d'Z d(S))TestHelpVariableExpansionz:Test that variables are expanded properly in help messagesr{)r|z-xz"x %(prog)s %(default)s %(type)s %%)r:r~z-yrrXXXz y %(prog)s %(default)s %(const)s)rrrr~z--foorz$foo %(prog)s %(default)s %(choices)s)rr~z--barrrErZBBBz!bar %(prog)s %(default)s %(dest)s)rrrr~rzspam %(prog)s %(default)s)r~rg?zbadger %(prog)s %(default)s)rr~rdz-aza %(prog)s %(default)sz-brZzb %(prog)s %(default)szw usage: PROG [-h] [-x X] [-y] [--foo {a,b,c}] [--bar BBB] [-a A] [-b B] spam badger a positional arguments: spam spam PROG None badger badger PROG 0.5 optional arguments: -h, --help show this help message and exit -x X x PROG None int % -y y PROG 42 XXX --foo {a,b,c} foo PROG None a, b, c --bar BBB bar PROG baz bar group: -a A a PROG None -b B b PROG -1 rN) rrrrr.r[rr_r!r&r~r"r r r r rEks&    rEc@s4eZdZdZedddZgZgZdZedZ dZ dS) &TestHelpVariableExpansionUsageSuppliedz@Test that variables are expanded properly when usage= is presentr{z %(prog)s FOO)r|r&z usage: PROG FOO z[ optional arguments: -h, --help show this help message and exit rN) rrrrr.r[r_r!r&r~r"r r r r rGs rGc@s0eZdZdZedddZgZgZdZeZ dZ dS)$TestHelpVariableExpansionNoArgumentsz;Test that variables are expanded properly with no argumentsr{F)r|rz usage: PROG rN) rrrrr.r[r_r!r&r~r"r r r r rHs rHc@sFeZdZdZedejdZedddedddgZgZ d Z d Z d Z d S) TestHelpSuppressUsagez3Test that items can be suppressed in usage messagesr{)r|r&z--foozfoo help)r~rz spam helpz positional arguments: spam spam help optional arguments: -h, --help show this help message and exit --foo FOO foo help rN) rrrrr.rrr[r_r!r~r&r"r r r r rIs rIc@sJeZdZdZedddZedejdedddgZgZ d Z e d Z d Z d S) TestHelpSuppressOptionalz?Test that optional arguments can be suppressed in help messagesr{F)r|rz--foo)r~rz spam helpz! usage: PROG spam zA positional arguments: spam spam help rN) rrrrr.r[rrr_r!r&r~r"r r r r rJs  rJc@s\eZdZdZeddZedddedddgZed ed ejdgfgZ d Z e d Z d Z dS)TestHelpSuppressOptionalGroupzeZdZdZeddZeddddgZgZdZed Z d Z d S) TestHelpRequiredOptionalz.Test that required options don't look optionalr{)r|z--fooTzfoo help)rr~z+ usage: PROG [-h] --foo FOO zz optional arguments: -h, --help show this help message and exit --foo FOO foo help rN) rrrrr.r[r_r!r&r~r"r r r r rMs rMc@sNeZdZdZeddddZedddd ed d d d gZgZdZedZ dZ dS)TestHelpAlternatePrefixCharsz:Test that options display with different prefix charactersr{z^;F)r|rrz^^foorzfoo help)rr~z;bz;;barzbar help)r~z- usage: PROG [^^foo] [;b BAR] zq optional arguments: ^^foo foo help ;b BAR, ;;bar BAR bar help rN) rrrrr.r[r_r!r&r~r"r r r r rN#s rNc@sHeZdZdZedddZedddedd dgZgZd Zed Z d Z d S)TestHelpNoHelpOptionalz=Test that the --help argument can be suppressed help messagesr{F)r|rz--foozfoo help)r~rz spam helpz- usage: PROG [--foo FOO] spam z positional arguments: spam spam help optional arguments: --foo FOO foo help rN) rrrrr.r[r_r!r&r~r"r r r r rO8s  rOc@s>eZdZdZeddZededgZgZdZedZ dZ d S) TestHelpNonez1Test that no errors occur if no help is specifiedr{)r|z--foorz2 usage: PROG [-h] [--foo FOO] spam z positional arguments: spam optional arguments: -h, --help show this help message and exit --foo FOO rN) rrrrr.r[r_r!r&r~r"r r r r rPOs  rPc @sjeZdZdZeddZeddddded d d d ded ddddedddddgZgZdZedZ dZ dS)TestHelpTupleMetavarz"Test specifying metavar as a tupler{)r|z-wr'r)ZW1ZW2)r~rrz-xrr)ZX1ZX2z-yrr)ZY1ZY2ZY3z-zrr)ZZ1z] usage: PROG [-h] [-w W1 [W2 ...]] [-x [X1 [X2 ...]]] [-y Y1 Y2 Y3] [-z [Z1]] z optional arguments: -h, --help show this help message and exit -w W1 [W2 ...] w -x [X1 [X2 ...]] x -y Y1 Y2 Y3 y -z [Z1] z rN) rrrrr.r[r_r!r&r~r"r r r r rQgs rQc@sdeZdZdZedejddZedddedd dgZed d d ed ddgfgZ dZ e dZ dZ dS)TestHelpRawTextzTest the RawTextHelpFormatterr{z7Keep the formatting exactly as it is written here )r|formatter_classr}z--fooz- foo help should also appear as given here)r~rz spam helprz? This text should be indented exactly like it is here )r}z--barzbar helpz> usage: PROG [-h] [--foo FOO] [--bar BAR] spam a Keep the formatting exactly as it is written here positional arguments: spam spam help optional arguments: -h, --help show this help message and exit --foo FOO foo help should also appear as given here title: This text should be indented exactly like it is here --bar BAR bar help rN) rrrrr.rRawTextHelpFormatterr[r_r!r&r~r"r r r r rRs  rRc@sdeZdZdZedejddZedddedd dgZed d d ed ddgfgZ dZ e dZ dZ dS)TestHelpRawDescriptionzTest the RawTextHelpFormatterr{z7Keep the formatting exactly as it is written here )r|rSr}z--fooz4 foo help should not retain this odd formatting)r~rz spam helprz? This text should be indented exactly like it is here )r}z--barzbar helpz> usage: PROG [-h] [--foo FOO] [--bar BAR] spam a Keep the formatting exactly as it is written here positional arguments: spam spam help optional arguments: -h, --help show this help message and exit --foo FOO foo help should not retain this odd formatting title: This text should be indented exactly like it is here --bar BAR bar help rN) rrrrr.rRawDescriptionHelpFormatterr[r_r!r&r~r"r r r r rUs  rUc @seZdZdZedejddZedddedd d d ed d dedddddgZedddede dddgfgZ dZ e dZ dZ dS)TestHelpArgumentDefaultsz&Test the ArgumentDefaultsHelpFormatterr{r})r|rSr}z--fooz)foo help - oh and by the way, %(default)s)r~z--barrzbar help)rr~rz spam helprrZwoodenz badger help)rrr~r)r}z--bazrzbaz help)r:rr~zO usage: PROG [-h] [--foo FOO] [--bar] [--baz BAZ] spam [badger] a description positional arguments: spam spam help badger badger help (default: wooden) optional arguments: -h, --help show this help message and exit --foo FOO foo help - oh and by the way, None --bar bar help (default: False) title: description --baz BAZ baz help (default: 42) rN)rrrrr.rArgumentDefaultsHelpFormatterr[r_rr!r&r~r"r r r r rWs    rWc@sBeZdZdZedddZeddddd gZgZd Zed Z d Z d S)TestHelpVersionActionz,Test the default help for the version actionr{r})r|r}z-Vz --versionr"z3.6)rr"z& usage: PROG [-h] [-V] z description optional arguments: -h, --help show this help message and exit -V, --version show program's version number and exit rN) rrrrr.r[r_r!r&r~r"r r r r rY s rYc@sTeZdZdZeddZeddddejded d d ed d d gZgZ dZ e dZ dS)TestHelpVersionActionSuppresszCTest that the --version argument can be suppressed in help messagesr{)r|z-vz --versionr"z1.0)rr"r~z--foozfoo help)r~rz spam helpz2 usage: PROG [-h] [--foo FOO] spam z positional arguments: spam spam help optional arguments: -h, --help show this help message and exit --foo FOO foo help N) rrrrr.r[rrr_r!r&r~r r r r rZs   rZc@sLeZdZdZedddZeddddd gZd d d DZd ZedZ dZ dS)TestHelpSubparsersOrderingz5Test ordering of subcommands in help matches the coder{zdisplay some subcommands)r|r}z-vz --versionr"z0.1)rr"cCsg|]}t|dqS))r%)r.)r5r%r r r r7>sz%TestHelpSubparsersOrdering.)rrrrrqz6 usage: PROG [-h] [-v] {a,b,c,d,e} ... z display some subcommands positional arguments: {a,b,c,d,e} optional arguments: -h, --help show this help message and exit -v, --version show program's version number and exit z 0.1 N) rrrrr.r[r_r"r&r~r"r r r r r[8s r[c@sPeZdZdZedddZeddddd gZd Zd d eDZd Z e dZ dZ dS)"TestHelpSubparsersWithHelpOrderingz5Test ordering of subcommands in help matches the coder{zdisplay some subcommands)r|r}z-vz --versionr"z0.1)rr"))rza subcommand help)rzb subcommand help)rzc subcommand help)rzd subcommand help)rqze subcommand helpcCsg|]\}}t||dqS))r%r~)r.)r5r%r~r r r r7bsz-TestHelpSubparsersWithHelpOrdering.z6 usage: PROG [-h] [-v] {a,b,c,d,e} ... a display some subcommands positional arguments: {a,b,c,d,e} a a subcommand help b b subcommand help c c subcommand help d d subcommand help e e subcommand help optional arguments: -h, --help show this help message and exit -v, --version show program's version number and exit z 0.1 N) rrrrr.r[r_Zsubcommand_datar"r&r~r"r r r r r\Usr\c@s`eZdZdZddZeddejdZede ded eded e d d gZ gZ d Z e dZdZdS)TestHelpMetavarTypeFormatterrcCs|S)Nr )rhr r r custom_typesz(TestHelpMetavarTypeFormatter.custom_typer{r})r|r}rSr)r:z-bz-cz SOME FLOAT)r:rzF usage: PROG [-h] [-b custom_type] [-c SOME FLOAT] int z description positional arguments: int optional arguments: -h, --help show this help message and exit -b custom_type -c SOME FLOAT N)rrrrr^r.rMetavarTypeHelpFormatterr[rrr_r!r&r~r"r r r r r]s    r]c@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd S)!TestInvalidArgumentConstructorsz-Test a bunch of invalid Argument constructorscOs"t}|jt|jf||dS)N)rrjr|rsr])rr/r0r^r r r assertTypeErrors z/TestInvalidArgumentConstructors.assertTypeErrorcOs"t}|jt|jf||dS)N)rrjr|rr])rr/r0r^r r r assertValueErrors z0TestInvalidArgumentConstructors.assertValueErrorcCs<|jddd|jddd|jddd|jdiddS) Nz-x)rz-yrb)callbackr )Z callback_args)Zcallback_kwargs)ra)rr r r test_invalid_keyword_argumentssz>TestInvalidArgumentConstructors.test_invalid_keyword_argumentscCs&|xdD]}|j|dqWdS)N)rr)r)ra)rrr r r test_missing_destinations z8TestInvalidArgumentConstructors.test_missing_destinationcCs|d|ddS)Nz--z---)rb)rr r r test_invalid_option_stringss z;TestInvalidArgumentConstructors.test_invalid_option_stringscCs$|jddd|jdttfddS)Nz--foor)r:)rbrr)rr r r test_invalid_typesz1TestInvalidArgumentConstructors.test_invalid_typec Csl|jddd|jddd|jdddt}|t}|jdddWdQRX|dt|jdS) Nz-xrb)rrz--foo)rrz store-truezunknown action) rbrrjr|rr]rrr)rr^rr r r test_invalid_actions z3TestInvalidArgumentConstructors.test_invalid_actionc CsNt}|jdd|t}|jdddWdQRX|dt|jdS)Nrb)rrrz+dest supplied twice for positional argument)rrjr]r|rrrr)rr^rr r r test_multiple_dests   z2TestInvalidArgumentConstructors.test_multiple_destcCsLxFdD]>}x8ttdtddtddgD]}|jd d|i|q(WqWdS) N)rrrrr)r:r)rZab)r-xr)rj)rrra)rrZattrsr r r test_no_argument_actionss z8TestInvalidArgumentConstructors.test_no_argument_actionscCs2x,dD]$}|jdd|d|jdd|dqWdS)N)rrrz-xrb)rrr)rr)ra)rrr r r !test_no_argument_no_const_actionss zATestInvalidArgumentConstructors.test_no_argument_no_const_actionscCsdx^dD]V}|jdd|d|jdd|dx0dD](}|jdd||d|jdd||dq0WqWdS) N)rrz-xr)rrr)rErrrb)rrr)rb)rrrr r r #test_more_than_one_argument_actionss   zCTestInvalidArgumentConstructors.test_more_than_one_argument_actionscCs"xdD]}|jdd|dqWdS)N)rrz-xr)rr)ra)rrr r r test_required_const_actionss z;TestInvalidArgumentConstructors.test_required_const_actionscCs4|jddd|jdddd|jddtjddS)Nrparsers)rr{)rr|)rr\)rarrj)rr r r "test_parsers_action_missing_paramsszBTestInvalidArgumentConstructors.test_parsers_action_missing_paramscCs|jddddS)NrbT)r)ra)rr r r test_required_positionalsz8TestInvalidArgumentConstructors.test_required_positionalcs`GdddtGfdddt}t}|j|jd|d|j|jd|ddS)Nc@s eZdZdS)zITestInvalidArgumentConstructors.test_user_defined_action..SuccessN)rrrr r r r Successsrrcs"eZdZdfdd ZddZdS)zHTestInvalidArgumentConstructors.test_user_defined_action..ActionFcs"|dkr|kr|krdS)Nrr )roption_stringsrrrr)rrr r r1szQTestInvalidArgumentConstructors.test_user_defined_action..Action.__init__c_sdS)Nr )rr/r0r r r rrszQTestInvalidArgumentConstructors.test_user_defined_action..Action.__call__N)F)rrrr1rrr )rrr r rusruz--spam)rrrr)r@rrrjr|r])rrur^r )rrr test_user_defined_actions   z8TestInvalidArgumentConstructors.test_user_defined_actionN)rrrrrarbrdrerfrgrhrirkrlrmrnrprqrtr r r r r`s   r`c@seZdZddZddZdS)TestActionsReturnedcCsXt}|d}||jd|dd}||jd|dd}||jddS) Nz--foorbz-bz--barrz-xz-yr)rrjr]rr)rr^rr r r r"s   zTestActionsReturned.test_destc Cst}|jddddtddgddd d }||jd||jd||jd||jt||j ddg||j d||j d||j d dS) Nz--foorrrrErrZBARr)rrrr:rr~rr) rrjr]rrrrrr:rr~rr)rr^rr r r test_misc+s   zTestActionsReturned.test_miscN)rrrrrvr r r r ru s ruc@s$eZdZddZddZddZdS)TestConflictHandlingcCs|jttjdddS)Nrb)conflict_handler)r|rrrj)rr r r test_bad_type@s z"TestConflictHandling.test_bad_typecCsDt}|d|tj|jd|d|tj|jddS)Nz-xz--spam)rrjr]r|r)rr^r r r test_conflict_errorDs   z(TestConflictHandling.test_conflict_errorcCsztj}|ddd}|jddd|jddd||td|jd d d |jd d d ||td dS)Nr{Zresolve)r|rxz-xzOLD X)r~zNEW Xz usage: PROG [-h] [-x X] optional arguments: -h, --help show this help message and exit -x X NEW X z--spamZOLD_SPAM)rZNEW_SPAMz usage: PROG [-h] [-x X] [--spam NEW_SPAM] optional arguments: -h, --help show this help message and exit -x X NEW X --spam NEW_SPAM )rrjr]rrrr)rrr^r r r test_resolve_errorMs z'TestConflictHandling.test_resolve_errorN)rrrryrzr{r r r r rw>s rwc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)TestOptionalsHelpVersionActionsz!Test the help and version actionsc Cs<|t}||WdQRX|||jjdS)N)r|r?rOr{rrrrB)rr^r}rr r r assertPrintHelpExitns z3TestOptionalsHelpVersionActions.assertPrintHelpExitcGs|t|j|dS)N)r|r?rO)rr^r/r r r rzssz9TestOptionalsHelpVersionActions.assertArgumentParserErrorcCsDt}|jddddd||d||d|tt|ddS) Nz-vz --versionr"z1.0)rr"z-hz--helpformat_version)rSr]r}r|AttributeErrorrJ)rr^r r r test_versionvs   z,TestOptionalsHelpVersionActions.test_versionc CsRtdd}|jddddd|t}|dgWdQRX|d|jjdS) NZPPP)r|z-vz --versionr"z %(prog)s 3.5)rr"zPPP 3.5 )rSr]r|r?rOrrrB)rr^rr r r test_version_format}s   z3TestOptionalsHelpVersionActions.test_version_formatcCsHtdd}|jddddd||d||d |tt|d dS) NF)rz-vz --versionr"z1.0)rr"z-hz--helpr~)rSr]rzr|rrJ)rr^r r r test_version_no_helps    z4TestOptionalsHelpVersionActions.test_version_no_helpc CsPtdd}|jdddd|t}|dgWdQRX|d|jjdS)NrF)r|z-Vr"z %(prog)s 3.7)rr"zXXX 3.7 )rSr]r|r?rOrrrB)rr^rr r r test_version_actions   z3TestOptionalsHelpVersionActions.test_version_actioncCs>tdd}||d||d||d||ddS)NF)rz-hz--helpz-vz --version)rSrz)rr^r r r test_no_helps     z,TestOptionalsHelpVersionActions.test_no_helpcCsZt}|jddd|jddd||d||d||d|tt|ddS) Nz-xr~)rz-yr"z-vz --versionr~)rSr]r}rzr|rrJ)rr^r r r test_alternate_help_versions   z;TestOptionalsHelpVersionActions.test_alternate_help_versioncCst}|jdddd|jddd|dd dd d d g}|d dg}xV|D]N}x|D]}d||f}qXW|||d|||d|tt|dqNWdS)Nz --versionr"z1.0)rr"z-xr)rrrrbz-x barzbaz -xz --bad-optionz foo bar bazz %s %%s %sz-hz--helpr~)rSr]r}r|rrJ)rr^Zvalid_prefixesZvalid_suffixesprefixsuffixrr r r !test_help_version_extra_argumentss    zATestOptionalsHelpVersionActions.test_help_version_extra_argumentsN) rrrrr}rzrrrrrrrr r r r r|ks r|c@sPeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ dS) TestStringsz3Test str() and repr() on Optionals and PositionalscCs&x ttgD]}||||q WdS)N)rr4r)robjZ result_stringfuncr r r assertStringEqualszTestStrings.assertStringEqualc Cs:tjdddgdddddd d gd d d }d}|||dS)Nz--fooz-az-brrrrrErrZHELPZMETAVAR)rsrr:rrrr~rzAction(option_strings=['--foo', '-a', '-b'], dest='b', nargs='+', const=None, default=42, type='int', choices=[1, 2, 3], help='HELP', metavar='METAVAR'))rrur)roptionrhr r r test_optionalszTestStrings.test_optionalc Cs8tjgdtdddddgddd}d t}|||dS) Nrrg@g?g?zH HH HzMV MV MV)rsrr:rrrr~rzAction(option_strings=[], dest='x', nargs='?', const=None, default=2.5, type=%r, choices=[0.5, 1.5, 2.5], help='H HH H', metavar='MV MV MV'))rrurr)rargumentrhr r r test_argumentszTestStrings.test_argumentcCs"tjddd}d}|||dS)Nrr)rbrzNamespace(bar='spam', foo=42))rrr)rrrhr r r test_namespaceszTestStrings.test_namespacecCs$tjfddi}d}|||dS)N"quotezNamespace(**{'"': 'quote'}))rrr)rrrhr r r 'test_namespace_starkwargs_notidentifiersz3TestStrings.test_namespace_starkwargs_notidentifiercCs,tjfddiddi}d}|||dS)NrrErrz Namespace(a=1, **{'"': 'quote'}))rrr)rrrhr r r 2test_namespace_kwargs_and_starkwargs_notidentifiersz>TestStrings.test_namespace_kwargs_and_starkwargs_notidentifiercCs$tjfddi}d}|||dS)NZvalidTzNamespace(valid=True))rrr)rrrhr r r $test_namespace_starkwargs_identifiersz0TestStrings.test_namespace_starkwargs_identifiercCs&tjdd}dtj}|||dS)Nr{)r|zvArgumentParser(prog='PROG', usage=None, description=None, formatter_class=%r, conflict_handler='error', add_help=True))rrj HelpFormatterr)rr^rhr r r test_parsers zTestStrings.test_parserN) rrrrrrrrrrrrr r r r rsrc@s$eZdZddZddZddZdS) TestNamespacecCsFt}|tt|dtjddd}||jd||jddS)Nrrr)rr)rrr|rrJrrr)rrr r r test_constructors zTestNamespace.test_constructorcCstjddd}tjddd}tjdd}tjdd}||||||||||||||||||k|||k|||k|||kdS)NrEr)rr)rr)r)r)rrrZassertNotEqual assertTrue)rZns1Zns2Zns3Zns4r r r test_equalitys       zTestNamespace.test_equalitycCs6tjddd}||dt||dtdS)NrEr)rr)rrZassertIsr>NotImplemented__ne__)rrr r r $test_equality_returns_notimplementedsz2TestNamespace.test_equality_returns_notimplementedN)rrrrrrr r r r rsrc@s$eZdZddZddZddZdS) TestEncodingc Cs>tj|\}}|d7}t|ddd}|WdQRXdS)Nz.pyr2zutf-8)r9)rrsplitextr(rG)rrrlrr r r _test_module_encoding%sz"TestEncoding._test_module_encodingcCs|tjdS)N)rr__file__)rr r r test_argparse_module_encoding+sz*TestEncoding.test_argparse_module_encodingcCs|tdS)N)rr)rr r r "test_test_argparse_module_encoding.sz/TestEncoding.test_test_argparse_module_encodingN)rrrrrrr r r r r#src@seZdZddZdS)TestArgumentErrorcCs$d}td|}|t||dS)Nz my error here)rrrr)rryrWr r r test_argument_error7s z%TestArgumentError.test_argument_errorN)rrrrr r r r r5src@seZdZddZdS)TestArgumentTypeErrorc CsXdd}tddd}|jd|d|t}|dgWdQRX|d |jjdS) NcSstddS)Nzspam!)rArgumentTypeError)rhr r r rDsz.spamr{F)r|rr)r:rFz-usage: PROG x PROG: error: argument x: spam! )rSr]r|r?rOrrrC)rrr^rr r r test_argument_type_errorBs  z.TestArgumentTypeError.test_argument_type_errorN)rrrrr r r r r@src@s$eZdZddZddZddZdS)TestMessageContentErrorc Cs.tddd}|jdtd|jdtdd|jd td d |t}|gWdQRXt|j}||d||d ||d |t}|d gWdQRXt|j}| |d||d ||d |t}|d dgWdQRXt|j}| |d| |d ||d dS)Nr{r)r|r&req_pos)r:z-req_optT)r:rZneed_oner)r:rreq_optZ myXargumentz -req_opt=1) rSr]rrr|r?rOr assertRegexr)rr^rryr r r %test_missing_argument_name_in_messageTs,               z=TestMessageContentError.test_missing_argument_name_in_messagec Cstddd}|jdtd|jdtdd|jd td dd |t}|gWdQRXt|j}| |d| |d | |d |t}|dgWdQRXt|j}| |d| |d | |d dS)Nr{r)r|r&r)r:z --req_optT)r:rz --opt_optr)r:rrrZopt_optz --req_opt=1) rSr]rrboolr|r?rOrrr)rr^rryr r r %test_optional_optional_not_in_messagems"           z=TestMessageContentError.test_optional_optional_not_in_messagec Csltddd}|d|jdddd|t}|gWdQRXt|j}||d||ddS) Nr{r)r|r&rZoptional_positionalrr`)rr) rSr]r|r?rOrrrr)rr^rryr r r 'test_optional_positional_not_in_messages     z?TestMessageContentError.test_optional_positional_not_in_messageN)rrrrrrr r r r rRsrc@seZdZddZdS)TestTypeFunctionCallOnlyOncecsHfdd}t}|jd|dd|d}tdd|dS) Ncs|ddS)Nzspam! foo_converted)r)Zstring_to_convert)rr r rs zLTestTypeFunctionCallOnlyOnce.test_type_function_call_only_once..spamz--foor)r:rz --foo spam!r)rb)rrjr]rOr{rr2)rrr^r/r )rr !test_type_function_call_only_onces  z>TestTypeFunctionCallOnlyOnce.test_type_function_call_only_onceN)rrrrr r r r rsrc@s,eZdZddZddZddZddZd S) TestTypeFunctionCalledOnDefaultcsDfdd}t}|jd|dd|g}tdd|dS)Ncs|ddS)Nrr)r)int_to_convert)rr r rs z]TestTypeFunctionCalledOnDefault.test_type_function_call_with_non_string_default..spamz--foor)r:r)rb)rrjr]rOrr2)rrr^r/r )rr /test_type_function_call_with_non_string_defaults   zOTestTypeFunctionCalledOnDefault.test_type_function_call_with_non_string_defaultcCs@dd}t}|jd|dd|g}|tdd|dS)NcSsdS)Nrr )rr r r rszYTestTypeFunctionCalledOnDefault.test_type_function_call_with_string_default..spamz--foo0)r:rr)rb)rrjr]rOrr2)rrr^r/r r r +test_type_function_call_with_string_defaults  zKTestTypeFunctionCalledOnDefault.test_type_function_call_with_string_defaultcCs@dd}t}|jd|dd|g}|tdd|dS)NcSs|dS)Nrr )Zstr_to_convertr r r extendszYTestTypeFunctionCalledOnDefault.test_no_double_type_conversion_of_default..extendz--testr)r:rz**)rl)rrjr]rOrr2)rrr^r/r r r )test_no_double_type_conversion_of_defaults  zITestTypeFunctionCalledOnDefault.test_no_double_type_conversion_of_defaultcCs8t}|jddtgdd|g}||jgdS)Nz--testrlr)rr:rr)rrjr]rrOrrl)rr^r/r r r test_issue_15906s    z0TestTypeFunctionCalledOnDefault.test_issue_15906N)rrrrrrrr r r r rs   rc@s<eZdZddZddZddZddZd d Zd d Zd S)TestParseKnownArgscCst}|ddS)Nr )rrjrO)rr^r r r test_arguments_tuplesz'TestParseKnownArgs.test_arguments_tuplecCst}|gdS)N)rrjrO)rr^r r r test_arguments_listsz&TestParseKnownArgs.test_arguments_listcCs t}|d|ddS)Nr)r)rrjr]rO)rr^r r r test_arguments_tuple_positionals z2TestParseKnownArgs.test_arguments_tuple_positionalcCs"t}|d|dgdS)Nr)rrjr]rO)rr^r r r test_arguments_list_positionals z1TestParseKnownArgs.test_arguments_list_positionalcCsJt}|d|d\}}|tdd||ddg|dS)Nz--fooz--foo F --bar --bazF)rbz--barz--baz)rrjr]rr{rr2)rr^r/extrasr r r test_optionalss  z!TestParseKnownArgs.test_optionalscCs|t}|jdddtd|jddd|dd d d dd d g}||\}}|tddd d||d d d g|dS)Nz-vrrE)rrr:z--spamr)rrrrfz--foorr rT)vrr)rrjr]rrrr2)rr^rjr/rr r r test_mixeds zTestParseKnownArgs.test_mixedN) rrrrrrrrrr r r r rs rc@s,eZdZddZddZddZddZd S) TestIntermixedArgsc Cs tjdd}|jddd|jdddd }|d |jd d td d}||}|tdd ddddgd|||\}}|tdd dgd||dddg|d}| |\}}|tdd ddgd||dddg|| |j ||j ddS)Nr{)r|z--foorb)rz--barrT)rrcmdrestr)rr:zcmd --foo x 1 --bar y 2 3rrrErr)rrrbrrrrz!cmd --foo x 1 --error 2 --bar y 3z--error) rrjr]rr{parse_intermixed_argsrr2rparse_known_intermixed_argsrr&r)rr^rrjr/rr r r test_basics"    zTestIntermixedArgs.test_basicc Csjtdd}|d|d|jdddd}|t}||WdQRX|t|jd dS) Nr{)r|z-zrrz...)rz X A B -z Zz\.\.\.) rSr]r{r|rsrrrr)rr^rjrr r r test_remainders    z!TestIntermixedArgs.test_remaindercCstdd}|jdd}|jdddd|jd d d |jd d ddd|d}|tddgddd||t|jd||j ddS)Nr{)r|T)rz--foorr)rr~z--spamr)r~rrrr)rrr~z 1 --foo 2rr)rrbrz1 2) rSrr]rr{rr2r|r?r)rr^rdr/r r r test_exclusive#s  z!TestIntermixedArgs.test_exclusivecCshtdd}|jdd}|jdddd|jd d d |jd d ddd|t|jg||jddS)Nr{)r|T)rz--foorr)rr~z--spamr)r~rrrr)rrr~)rSrr]r|rsrrr)rr^rdr r r test_exclusive_incompatible/s  z.TestIntermixedArgs.test_exclusive_incompatibleN)rrrrrrrr r r r rs  rc@seZdZddZdS)!TestIntermixedMessageContentErrorc Cstddd}|jdtd|jdtdd|t}|gWdQRXt|j}||d||d |t}| gWdQRXt|j}| |d||d dS) Nr{r)r|r&r)r:z-req_optT)r:rr) rSr]rrr|r?rOrrrZassertNotRegex)rr^rryr r r r<s        zGTestIntermixedMessageContentError.test_missing_argument_name_in_messageN)rrrrr r r r r9src@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Z dd?Z"d@dAZ#dBdCZ$dDdEZ%dFdGZ&dHdIZ'dJdKZ(dLdMZ)dNdOZ*dPdQZ+dRdSZ,dTdUZ-dVdWZ.dXdYZ/dZd[Z0d\d]Z1d^d_Z2d`S)aTestAddArgumentMetavarz,length of metavar tuple does not match nargscCst}|jd||ddS)Nz--foo)rr)rrjr])rrrr^r r r do_test_no_exceptionUsz+TestAddArgumentMetavar.do_test_no_exceptionc CsHt}|t}|jd||dWdQRX||jjd|jdS)Nz--foo)rrr) rrjr|rr]rrr/EXPECTED_MESSAGE)rrrr^rr r r do_test_exceptionYs z(TestAddArgumentMetavar.do_test_exceptioncCs|jddddS)Nr)rr)r)rr r r test_nargs_None_metavar_stringasz5TestAddArgumentMetavar.test_nargs_None_metavar_stringcCs|jdtddS)N)rr)rtuple)rr r r test_nargs_None_metavar_length0dsz6TestAddArgumentMetavar.test_nargs_None_metavar_length0cCs|jddddS)N)r)rr)r)rr r r test_nargs_None_metavar_length1gsz6TestAddArgumentMetavar.test_nargs_None_metavar_length1cCs|jddddS)N)rr)rr)r)rr r r test_nargs_None_metavar_length2jsz6TestAddArgumentMetavar.test_nargs_None_metavar_length2cCs|jddddS)N)rrr)rr)r)rr r r test_nargs_None_metavar_length3msz6TestAddArgumentMetavar.test_nargs_None_metavar_length3cCs|jddddS)Nrr)rr)r)rr r r "test_nargs_optional_metavar_stringrsz9TestAddArgumentMetavar.test_nargs_optional_metavar_stringcCs|jdtddS)Nr)rr)rr)rr r r #test_nargs_optional_metavar_length0usz:TestAddArgumentMetavar.test_nargs_optional_metavar_length0cCs|jddddS)Nr)r)rr)r)rr r r #test_nargs_optional_metavar_length1xsz:TestAddArgumentMetavar.test_nargs_optional_metavar_length1cCs|jddddS)Nr)rr)rr)r)rr r r #test_nargs_optional_metavar_length2{sz:TestAddArgumentMetavar.test_nargs_optional_metavar_length2cCs|jddddS)Nr)rrr)rr)r)rr r r #test_nargs_optional_metavar_length3~sz:TestAddArgumentMetavar.test_nargs_optional_metavar_length3cCs|jddddS)Nrr)rr)r)rr r r $test_nargs_zeroormore_metavar_stringsz;TestAddArgumentMetavar.test_nargs_zeroormore_metavar_stringcCs|jdtddS)Nr)rr)rr)rr r r %test_nargs_zeroormore_metavar_length0sz.)r<rr9rr8r)rr9r r r 'test_all_exports_everything_but_modulessz6TestImportStar.test_all_exports_everything_but_modulesN)rrrrlr r r r r rsrc@seZdZddZddZdS)TestWrappingMetavarcCs"td|_d}|jjd|ddS)NZ9this_is_spammy_prog_with_a_long_name_sorry_about_the_namezz--proxy)r)rSr^r])rrr r r rszTestWrappingMetavar.setUpcCs |j}||tddS)Na( usage: this_is_spammy_prog_with_a_long_name_sorry_about_the_name [-h] [--proxy ] optional arguments: -h, --help show this help message and exit --proxy )r^rrrr)r help_textr r r test_help_with_metavars  z*TestWrappingMetavar.test_help_with_metavarN)rrrrr r r r r r  s r cCsttit_tt_dS)N)rZ run_unittestrrDrFrrQr r r r test_main#s r __main__)codecsrrr!rrHrrZunittestriorrlrrrr rrr.r2r@r?rRrjrSr:rYrrZrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrrrrrrrrr!r#r+r1rArLrWZ eq_stderrrDrIrMrOrQZskipIfrrUrVrXrYr_rbrergrmrvryrrrrrrrrrrrrrrrrrrrrrrrrrrr'r+r,r-r/r0r2r4r5r6r8r<r@rArBrCrDrErGrHrIrJrKrLrMrNrOrPrQrRrUrWrYrZr[r\r]r`rurwr|rrrrrrrrrrrrrr r rr r r r s      q                                           ## Ci'4&0#+!$-*(-   ^R$7/4(6,$&"$. 10(+"~-LG% =1*?,