test_argparse.py 182 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993499449954996499749984999500050015002500350045005500650075008500950105011501250135014501550165017501850195020502150225023502450255026502750285029503050315032503350345035503650375038503950405041504250435044504550465047504850495050505150525053505450555056505750585059506050615062506350645065506650675068506950705071507250735074507550765077507850795080508150825083508450855086508750885089509050915092509350945095509650975098509951005101510251035104510551065107510851095110511151125113511451155116511751185119512051215122512351245125512651275128512951305131513251335134513551365137513851395140514151425143514451455146514751485149515051515152515351545155515651575158515951605161516251635164516551665167516851695170517151725173517451755176517751785179518051815182518351845185518651875188518951905191519251935194519551965197519851995200520152025203520452055206520752085209521052115212521352145215521652175218521952205221522252235224522552265227522852295230523152325233523452355236523752385239524052415242524352445245524652475248524952505251525252535254525552565257525852595260526152625263526452655266526752685269527052715272527352745275527652775278527952805281528252835284528552865287528852895290529152925293529452955296529752985299530053015302530353045305530653075308530953105311531253135314531553165317531853195320532153225323532453255326532753285329533053315332533353345335533653375338533953405341534253435344534553465347534853495350535153525353535453555356535753585359536053615362536353645365536653675368536953705371537253735374537553765377537853795380538153825383538453855386538753885389539053915392539353945395539653975398539954005401540254035404540554065407540854095410541154125413541454155416541754185419542054215422542354245425542654275428542954305431543254335434543554365437543854395440544154425443544454455446544754485449545054515452545354545455545654575458545954605461546254635464546554665467546854695470547154725473547454755476547754785479548054815482548354845485548654875488548954905491549254935494549554965497549854995500550155025503550455055506550755085509551055115512551355145515551655175518551955205521552255235524552555265527552855295530553155325533553455355536553755385539554055415542554355445545554655475548554955505551555255535554555555565557555855595560556155625563556455655566556755685569557055715572557355745575
  1. # Author: Steven J. Bethard <steven.bethard@gmail.com>.
  2. import inspect
  3. import io
  4. import operator
  5. import os
  6. import shutil
  7. import stat
  8. import sys
  9. import textwrap
  10. import tempfile
  11. import unittest
  12. import argparse
  13. import warnings
  14. from test.support import os_helper
  15. from unittest import mock
  16. class StdIOBuffer(io.TextIOWrapper):
  17. '''Replacement for writable io.StringIO that behaves more like real file
  18. Unlike StringIO, provides a buffer attribute that holds the underlying
  19. binary data, allowing it to replace sys.stdout/sys.stderr in more
  20. contexts.
  21. '''
  22. def __init__(self, initial_value='', newline='\n'):
  23. initial_value = initial_value.encode('utf-8')
  24. super().__init__(io.BufferedWriter(io.BytesIO(initial_value)),
  25. 'utf-8', newline=newline)
  26. def getvalue(self):
  27. self.flush()
  28. return self.buffer.raw.getvalue().decode('utf-8')
  29. class TestCase(unittest.TestCase):
  30. def setUp(self):
  31. # The tests assume that line wrapping occurs at 80 columns, but this
  32. # behaviour can be overridden by setting the COLUMNS environment
  33. # variable. To ensure that this width is used, set COLUMNS to 80.
  34. env = self.enterContext(os_helper.EnvironmentVarGuard())
  35. env['COLUMNS'] = '80'
  36. @os_helper.skip_unless_working_chmod
  37. class TempDirMixin(object):
  38. def setUp(self):
  39. self.temp_dir = tempfile.mkdtemp()
  40. self.old_dir = os.getcwd()
  41. os.chdir(self.temp_dir)
  42. def tearDown(self):
  43. os.chdir(self.old_dir)
  44. for root, dirs, files in os.walk(self.temp_dir, topdown=False):
  45. for name in files:
  46. os.chmod(os.path.join(self.temp_dir, name), stat.S_IWRITE)
  47. shutil.rmtree(self.temp_dir, True)
  48. def create_writable_file(self, filename):
  49. file_path = os.path.join(self.temp_dir, filename)
  50. with open(file_path, 'w', encoding="utf-8") as file:
  51. file.write(filename)
  52. return file_path
  53. def create_readonly_file(self, filename):
  54. os.chmod(self.create_writable_file(filename), stat.S_IREAD)
  55. class Sig(object):
  56. def __init__(self, *args, **kwargs):
  57. self.args = args
  58. self.kwargs = kwargs
  59. class NS(object):
  60. def __init__(self, **kwargs):
  61. self.__dict__.update(kwargs)
  62. def __repr__(self):
  63. sorted_items = sorted(self.__dict__.items())
  64. kwarg_str = ', '.join(['%s=%r' % tup for tup in sorted_items])
  65. return '%s(%s)' % (type(self).__name__, kwarg_str)
  66. def __eq__(self, other):
  67. return vars(self) == vars(other)
  68. class ArgumentParserError(Exception):
  69. def __init__(self, message, stdout=None, stderr=None, error_code=None):
  70. Exception.__init__(self, message, stdout, stderr)
  71. self.message = message
  72. self.stdout = stdout
  73. self.stderr = stderr
  74. self.error_code = error_code
  75. def stderr_to_parser_error(parse_args, *args, **kwargs):
  76. # if this is being called recursively and stderr or stdout is already being
  77. # redirected, simply call the function and let the enclosing function
  78. # catch the exception
  79. if isinstance(sys.stderr, StdIOBuffer) or isinstance(sys.stdout, StdIOBuffer):
  80. return parse_args(*args, **kwargs)
  81. # if this is not being called recursively, redirect stderr and
  82. # use it as the ArgumentParserError message
  83. old_stdout = sys.stdout
  84. old_stderr = sys.stderr
  85. sys.stdout = StdIOBuffer()
  86. sys.stderr = StdIOBuffer()
  87. try:
  88. try:
  89. result = parse_args(*args, **kwargs)
  90. for key in list(vars(result)):
  91. attr = getattr(result, key)
  92. if attr is sys.stdout:
  93. setattr(result, key, old_stdout)
  94. elif attr is sys.stdout.buffer:
  95. setattr(result, key, getattr(old_stdout, 'buffer', BIN_STDOUT_SENTINEL))
  96. elif attr is sys.stderr:
  97. setattr(result, key, old_stderr)
  98. elif attr is sys.stderr.buffer:
  99. setattr(result, key, getattr(old_stderr, 'buffer', BIN_STDERR_SENTINEL))
  100. return result
  101. except SystemExit as e:
  102. code = e.code
  103. stdout = sys.stdout.getvalue()
  104. stderr = sys.stderr.getvalue()
  105. raise ArgumentParserError(
  106. "SystemExit", stdout, stderr, code) from None
  107. finally:
  108. sys.stdout = old_stdout
  109. sys.stderr = old_stderr
  110. class ErrorRaisingArgumentParser(argparse.ArgumentParser):
  111. def parse_args(self, *args, **kwargs):
  112. parse_args = super(ErrorRaisingArgumentParser, self).parse_args
  113. return stderr_to_parser_error(parse_args, *args, **kwargs)
  114. def exit(self, *args, **kwargs):
  115. exit = super(ErrorRaisingArgumentParser, self).exit
  116. return stderr_to_parser_error(exit, *args, **kwargs)
  117. def error(self, *args, **kwargs):
  118. error = super(ErrorRaisingArgumentParser, self).error
  119. return stderr_to_parser_error(error, *args, **kwargs)
  120. class ParserTesterMetaclass(type):
  121. """Adds parser tests using the class attributes.
  122. Classes of this type should specify the following attributes:
  123. argument_signatures -- a list of Sig objects which specify
  124. the signatures of Argument objects to be created
  125. failures -- a list of args lists that should cause the parser
  126. to fail
  127. successes -- a list of (initial_args, options, remaining_args) tuples
  128. where initial_args specifies the string args to be parsed,
  129. options is a dict that should match the vars() of the options
  130. parsed out of initial_args, and remaining_args should be any
  131. remaining unparsed arguments
  132. """
  133. def __init__(cls, name, bases, bodydict):
  134. if name == 'ParserTestCase':
  135. return
  136. # default parser signature is empty
  137. if not hasattr(cls, 'parser_signature'):
  138. cls.parser_signature = Sig()
  139. if not hasattr(cls, 'parser_class'):
  140. cls.parser_class = ErrorRaisingArgumentParser
  141. # ---------------------------------------
  142. # functions for adding optional arguments
  143. # ---------------------------------------
  144. def no_groups(parser, argument_signatures):
  145. """Add all arguments directly to the parser"""
  146. for sig in argument_signatures:
  147. parser.add_argument(*sig.args, **sig.kwargs)
  148. def one_group(parser, argument_signatures):
  149. """Add all arguments under a single group in the parser"""
  150. group = parser.add_argument_group('foo')
  151. for sig in argument_signatures:
  152. group.add_argument(*sig.args, **sig.kwargs)
  153. def many_groups(parser, argument_signatures):
  154. """Add each argument in its own group to the parser"""
  155. for i, sig in enumerate(argument_signatures):
  156. group = parser.add_argument_group('foo:%i' % i)
  157. group.add_argument(*sig.args, **sig.kwargs)
  158. # --------------------------
  159. # functions for parsing args
  160. # --------------------------
  161. def listargs(parser, args):
  162. """Parse the args by passing in a list"""
  163. return parser.parse_args(args)
  164. def sysargs(parser, args):
  165. """Parse the args by defaulting to sys.argv"""
  166. old_sys_argv = sys.argv
  167. sys.argv = [old_sys_argv[0]] + args
  168. try:
  169. return parser.parse_args()
  170. finally:
  171. sys.argv = old_sys_argv
  172. # class that holds the combination of one optional argument
  173. # addition method and one arg parsing method
  174. class AddTests(object):
  175. def __init__(self, tester_cls, add_arguments, parse_args):
  176. self._add_arguments = add_arguments
  177. self._parse_args = parse_args
  178. add_arguments_name = self._add_arguments.__name__
  179. parse_args_name = self._parse_args.__name__
  180. for test_func in [self.test_failures, self.test_successes]:
  181. func_name = test_func.__name__
  182. names = func_name, add_arguments_name, parse_args_name
  183. test_name = '_'.join(names)
  184. def wrapper(self, test_func=test_func):
  185. test_func(self)
  186. try:
  187. wrapper.__name__ = test_name
  188. except TypeError:
  189. pass
  190. setattr(tester_cls, test_name, wrapper)
  191. def _get_parser(self, tester):
  192. args = tester.parser_signature.args
  193. kwargs = tester.parser_signature.kwargs
  194. parser = tester.parser_class(*args, **kwargs)
  195. self._add_arguments(parser, tester.argument_signatures)
  196. return parser
  197. def test_failures(self, tester):
  198. parser = self._get_parser(tester)
  199. for args_str in tester.failures:
  200. args = args_str.split()
  201. with tester.assertRaises(ArgumentParserError, msg=args):
  202. parser.parse_args(args)
  203. def test_successes(self, tester):
  204. parser = self._get_parser(tester)
  205. for args, expected_ns in tester.successes:
  206. if isinstance(args, str):
  207. args = args.split()
  208. result_ns = self._parse_args(parser, args)
  209. tester.assertEqual(expected_ns, result_ns)
  210. # add tests for each combination of an optionals adding method
  211. # and an arg parsing method
  212. for add_arguments in [no_groups, one_group, many_groups]:
  213. for parse_args in [listargs, sysargs]:
  214. AddTests(cls, add_arguments, parse_args)
  215. bases = TestCase,
  216. ParserTestCase = ParserTesterMetaclass('ParserTestCase', bases, {})
  217. # ===============
  218. # Optionals tests
  219. # ===============
  220. class TestOptionalsSingleDash(ParserTestCase):
  221. """Test an Optional with a single-dash option string"""
  222. argument_signatures = [Sig('-x')]
  223. failures = ['-x', 'a', '--foo', '-x --foo', '-x -y']
  224. successes = [
  225. ('', NS(x=None)),
  226. ('-x a', NS(x='a')),
  227. ('-xa', NS(x='a')),
  228. ('-x -1', NS(x='-1')),
  229. ('-x-1', NS(x='-1')),
  230. ]
  231. class TestOptionalsSingleDashCombined(ParserTestCase):
  232. """Test an Optional with a single-dash option string"""
  233. argument_signatures = [
  234. Sig('-x', action='store_true'),
  235. Sig('-yyy', action='store_const', const=42),
  236. Sig('-z'),
  237. ]
  238. failures = ['a', '--foo', '-xa', '-x --foo', '-x -z', '-z -x',
  239. '-yx', '-yz a', '-yyyx', '-yyyza', '-xyza', '-x=']
  240. successes = [
  241. ('', NS(x=False, yyy=None, z=None)),
  242. ('-x', NS(x=True, yyy=None, z=None)),
  243. ('-za', NS(x=False, yyy=None, z='a')),
  244. ('-z a', NS(x=False, yyy=None, z='a')),
  245. ('-xza', NS(x=True, yyy=None, z='a')),
  246. ('-xz a', NS(x=True, yyy=None, z='a')),
  247. ('-x -za', NS(x=True, yyy=None, z='a')),
  248. ('-x -z a', NS(x=True, yyy=None, z='a')),
  249. ('-y', NS(x=False, yyy=42, z=None)),
  250. ('-yyy', NS(x=False, yyy=42, z=None)),
  251. ('-x -yyy -za', NS(x=True, yyy=42, z='a')),
  252. ('-x -yyy -z a', NS(x=True, yyy=42, z='a')),
  253. ]
  254. class TestOptionalsSingleDashLong(ParserTestCase):
  255. """Test an Optional with a multi-character single-dash option string"""
  256. argument_signatures = [Sig('-foo')]
  257. failures = ['-foo', 'a', '--foo', '-foo --foo', '-foo -y', '-fooa']
  258. successes = [
  259. ('', NS(foo=None)),
  260. ('-foo a', NS(foo='a')),
  261. ('-foo -1', NS(foo='-1')),
  262. ('-fo a', NS(foo='a')),
  263. ('-f a', NS(foo='a')),
  264. ]
  265. class TestOptionalsSingleDashSubsetAmbiguous(ParserTestCase):
  266. """Test Optionals where option strings are subsets of each other"""
  267. argument_signatures = [Sig('-f'), Sig('-foobar'), Sig('-foorab')]
  268. failures = ['-f', '-foo', '-fo', '-foo b', '-foob', '-fooba', '-foora']
  269. successes = [
  270. ('', NS(f=None, foobar=None, foorab=None)),
  271. ('-f a', NS(f='a', foobar=None, foorab=None)),
  272. ('-fa', NS(f='a', foobar=None, foorab=None)),
  273. ('-foa', NS(f='oa', foobar=None, foorab=None)),
  274. ('-fooa', NS(f='ooa', foobar=None, foorab=None)),
  275. ('-foobar a', NS(f=None, foobar='a', foorab=None)),
  276. ('-foorab a', NS(f=None, foobar=None, foorab='a')),
  277. ]
  278. class TestOptionalsSingleDashAmbiguous(ParserTestCase):
  279. """Test Optionals that partially match but are not subsets"""
  280. argument_signatures = [Sig('-foobar'), Sig('-foorab')]
  281. failures = ['-f', '-f a', '-fa', '-foa', '-foo', '-fo', '-foo b']
  282. successes = [
  283. ('', NS(foobar=None, foorab=None)),
  284. ('-foob a', NS(foobar='a', foorab=None)),
  285. ('-foor a', NS(foobar=None, foorab='a')),
  286. ('-fooba a', NS(foobar='a', foorab=None)),
  287. ('-foora a', NS(foobar=None, foorab='a')),
  288. ('-foobar a', NS(foobar='a', foorab=None)),
  289. ('-foorab a', NS(foobar=None, foorab='a')),
  290. ]
  291. class TestOptionalsNumeric(ParserTestCase):
  292. """Test an Optional with a short opt string"""
  293. argument_signatures = [Sig('-1', dest='one')]
  294. failures = ['-1', 'a', '-1 --foo', '-1 -y', '-1 -1', '-1 -2']
  295. successes = [
  296. ('', NS(one=None)),
  297. ('-1 a', NS(one='a')),
  298. ('-1a', NS(one='a')),
  299. ('-1-2', NS(one='-2')),
  300. ]
  301. class TestOptionalsDoubleDash(ParserTestCase):
  302. """Test an Optional with a double-dash option string"""
  303. argument_signatures = [Sig('--foo')]
  304. failures = ['--foo', '-f', '-f a', 'a', '--foo -x', '--foo --bar']
  305. successes = [
  306. ('', NS(foo=None)),
  307. ('--foo a', NS(foo='a')),
  308. ('--foo=a', NS(foo='a')),
  309. ('--foo -2.5', NS(foo='-2.5')),
  310. ('--foo=-2.5', NS(foo='-2.5')),
  311. ]
  312. class TestOptionalsDoubleDashPartialMatch(ParserTestCase):
  313. """Tests partial matching with a double-dash option string"""
  314. argument_signatures = [
  315. Sig('--badger', action='store_true'),
  316. Sig('--bat'),
  317. ]
  318. failures = ['--bar', '--b', '--ba', '--b=2', '--ba=4', '--badge 5']
  319. successes = [
  320. ('', NS(badger=False, bat=None)),
  321. ('--bat X', NS(badger=False, bat='X')),
  322. ('--bad', NS(badger=True, bat=None)),
  323. ('--badg', NS(badger=True, bat=None)),
  324. ('--badge', NS(badger=True, bat=None)),
  325. ('--badger', NS(badger=True, bat=None)),
  326. ]
  327. class TestOptionalsDoubleDashPrefixMatch(ParserTestCase):
  328. """Tests when one double-dash option string is a prefix of another"""
  329. argument_signatures = [
  330. Sig('--badger', action='store_true'),
  331. Sig('--ba'),
  332. ]
  333. failures = ['--bar', '--b', '--ba', '--b=2', '--badge 5']
  334. successes = [
  335. ('', NS(badger=False, ba=None)),
  336. ('--ba X', NS(badger=False, ba='X')),
  337. ('--ba=X', NS(badger=False, ba='X')),
  338. ('--bad', NS(badger=True, ba=None)),
  339. ('--badg', NS(badger=True, ba=None)),
  340. ('--badge', NS(badger=True, ba=None)),
  341. ('--badger', NS(badger=True, ba=None)),
  342. ]
  343. class TestOptionalsSingleDoubleDash(ParserTestCase):
  344. """Test an Optional with single- and double-dash option strings"""
  345. argument_signatures = [
  346. Sig('-f', action='store_true'),
  347. Sig('--bar'),
  348. Sig('-baz', action='store_const', const=42),
  349. ]
  350. failures = ['--bar', '-fbar', '-fbaz', '-bazf', '-b B', 'B']
  351. successes = [
  352. ('', NS(f=False, bar=None, baz=None)),
  353. ('-f', NS(f=True, bar=None, baz=None)),
  354. ('--ba B', NS(f=False, bar='B', baz=None)),
  355. ('-f --bar B', NS(f=True, bar='B', baz=None)),
  356. ('-f -b', NS(f=True, bar=None, baz=42)),
  357. ('-ba -f', NS(f=True, bar=None, baz=42)),
  358. ]
  359. class TestOptionalsAlternatePrefixChars(ParserTestCase):
  360. """Test an Optional with option strings with custom prefixes"""
  361. parser_signature = Sig(prefix_chars='+:/', add_help=False)
  362. argument_signatures = [
  363. Sig('+f', action='store_true'),
  364. Sig('::bar'),
  365. Sig('/baz', action='store_const', const=42),
  366. ]
  367. failures = ['--bar', '-fbar', '-b B', 'B', '-f', '--bar B', '-baz', '-h', '--help', '+h', '::help', '/help']
  368. successes = [
  369. ('', NS(f=False, bar=None, baz=None)),
  370. ('+f', NS(f=True, bar=None, baz=None)),
  371. ('::ba B', NS(f=False, bar='B', baz=None)),
  372. ('+f ::bar B', NS(f=True, bar='B', baz=None)),
  373. ('+f /b', NS(f=True, bar=None, baz=42)),
  374. ('/ba +f', NS(f=True, bar=None, baz=42)),
  375. ]
  376. class TestOptionalsAlternatePrefixCharsAddedHelp(ParserTestCase):
  377. """When ``-`` not in prefix_chars, default operators created for help
  378. should use the prefix_chars in use rather than - or --
  379. http://bugs.python.org/issue9444"""
  380. parser_signature = Sig(prefix_chars='+:/', add_help=True)
  381. argument_signatures = [
  382. Sig('+f', action='store_true'),
  383. Sig('::bar'),
  384. Sig('/baz', action='store_const', const=42),
  385. ]
  386. failures = ['--bar', '-fbar', '-b B', 'B', '-f', '--bar B', '-baz']
  387. successes = [
  388. ('', NS(f=False, bar=None, baz=None)),
  389. ('+f', NS(f=True, bar=None, baz=None)),
  390. ('::ba B', NS(f=False, bar='B', baz=None)),
  391. ('+f ::bar B', NS(f=True, bar='B', baz=None)),
  392. ('+f /b', NS(f=True, bar=None, baz=42)),
  393. ('/ba +f', NS(f=True, bar=None, baz=42))
  394. ]
  395. class TestOptionalsAlternatePrefixCharsMultipleShortArgs(ParserTestCase):
  396. """Verify that Optionals must be called with their defined prefixes"""
  397. parser_signature = Sig(prefix_chars='+-', add_help=False)
  398. argument_signatures = [
  399. Sig('-x', action='store_true'),
  400. Sig('+y', action='store_true'),
  401. Sig('+z', action='store_true'),
  402. ]
  403. failures = ['-w',
  404. '-xyz',
  405. '+x',
  406. '-y',
  407. '+xyz',
  408. ]
  409. successes = [
  410. ('', NS(x=False, y=False, z=False)),
  411. ('-x', NS(x=True, y=False, z=False)),
  412. ('+y -x', NS(x=True, y=True, z=False)),
  413. ('+yz -x', NS(x=True, y=True, z=True)),
  414. ]
  415. class TestOptionalsShortLong(ParserTestCase):
  416. """Test a combination of single- and double-dash option strings"""
  417. argument_signatures = [
  418. Sig('-v', '--verbose', '-n', '--noisy', action='store_true'),
  419. ]
  420. failures = ['--x --verbose', '-N', 'a', '-v x']
  421. successes = [
  422. ('', NS(verbose=False)),
  423. ('-v', NS(verbose=True)),
  424. ('--verbose', NS(verbose=True)),
  425. ('-n', NS(verbose=True)),
  426. ('--noisy', NS(verbose=True)),
  427. ]
  428. class TestOptionalsDest(ParserTestCase):
  429. """Tests various means of setting destination"""
  430. argument_signatures = [Sig('--foo-bar'), Sig('--baz', dest='zabbaz')]
  431. failures = ['a']
  432. successes = [
  433. ('--foo-bar f', NS(foo_bar='f', zabbaz=None)),
  434. ('--baz g', NS(foo_bar=None, zabbaz='g')),
  435. ('--foo-bar h --baz i', NS(foo_bar='h', zabbaz='i')),
  436. ('--baz j --foo-bar k', NS(foo_bar='k', zabbaz='j')),
  437. ]
  438. class TestOptionalsDefault(ParserTestCase):
  439. """Tests specifying a default for an Optional"""
  440. argument_signatures = [Sig('-x'), Sig('-y', default=42)]
  441. failures = ['a']
  442. successes = [
  443. ('', NS(x=None, y=42)),
  444. ('-xx', NS(x='x', y=42)),
  445. ('-yy', NS(x=None, y='y')),
  446. ]
  447. class TestOptionalsNargsDefault(ParserTestCase):
  448. """Tests not specifying the number of args for an Optional"""
  449. argument_signatures = [Sig('-x')]
  450. failures = ['a', '-x']
  451. successes = [
  452. ('', NS(x=None)),
  453. ('-x a', NS(x='a')),
  454. ]
  455. class TestOptionalsNargs1(ParserTestCase):
  456. """Tests specifying 1 arg for an Optional"""
  457. argument_signatures = [Sig('-x', nargs=1)]
  458. failures = ['a', '-x']
  459. successes = [
  460. ('', NS(x=None)),
  461. ('-x a', NS(x=['a'])),
  462. ]
  463. class TestOptionalsNargs3(ParserTestCase):
  464. """Tests specifying 3 args for an Optional"""
  465. argument_signatures = [Sig('-x', nargs=3)]
  466. failures = ['a', '-x', '-x a', '-x a b', 'a -x', 'a -x b']
  467. successes = [
  468. ('', NS(x=None)),
  469. ('-x a b c', NS(x=['a', 'b', 'c'])),
  470. ]
  471. class TestOptionalsNargsOptional(ParserTestCase):
  472. """Tests specifying an Optional arg for an Optional"""
  473. argument_signatures = [
  474. Sig('-w', nargs='?'),
  475. Sig('-x', nargs='?', const=42),
  476. Sig('-y', nargs='?', default='spam'),
  477. Sig('-z', nargs='?', type=int, const='42', default='84'),
  478. ]
  479. failures = ['2']
  480. successes = [
  481. ('', NS(w=None, x=None, y='spam', z=84)),
  482. ('-w', NS(w=None, x=None, y='spam', z=84)),
  483. ('-w 2', NS(w='2', x=None, y='spam', z=84)),
  484. ('-x', NS(w=None, x=42, y='spam', z=84)),
  485. ('-x 2', NS(w=None, x='2', y='spam', z=84)),
  486. ('-y', NS(w=None, x=None, y=None, z=84)),
  487. ('-y 2', NS(w=None, x=None, y='2', z=84)),
  488. ('-z', NS(w=None, x=None, y='spam', z=42)),
  489. ('-z 2', NS(w=None, x=None, y='spam', z=2)),
  490. ]
  491. class TestOptionalsNargsZeroOrMore(ParserTestCase):
  492. """Tests specifying args for an Optional that accepts zero or more"""
  493. argument_signatures = [
  494. Sig('-x', nargs='*'),
  495. Sig('-y', nargs='*', default='spam'),
  496. ]
  497. failures = ['a']
  498. successes = [
  499. ('', NS(x=None, y='spam')),
  500. ('-x', NS(x=[], y='spam')),
  501. ('-x a', NS(x=['a'], y='spam')),
  502. ('-x a b', NS(x=['a', 'b'], y='spam')),
  503. ('-y', NS(x=None, y=[])),
  504. ('-y a', NS(x=None, y=['a'])),
  505. ('-y a b', NS(x=None, y=['a', 'b'])),
  506. ]
  507. class TestOptionalsNargsOneOrMore(ParserTestCase):
  508. """Tests specifying args for an Optional that accepts one or more"""
  509. argument_signatures = [
  510. Sig('-x', nargs='+'),
  511. Sig('-y', nargs='+', default='spam'),
  512. ]
  513. failures = ['a', '-x', '-y', 'a -x', 'a -y b']
  514. successes = [
  515. ('', NS(x=None, y='spam')),
  516. ('-x a', NS(x=['a'], y='spam')),
  517. ('-x a b', NS(x=['a', 'b'], y='spam')),
  518. ('-y a', NS(x=None, y=['a'])),
  519. ('-y a b', NS(x=None, y=['a', 'b'])),
  520. ]
  521. class TestOptionalsChoices(ParserTestCase):
  522. """Tests specifying the choices for an Optional"""
  523. argument_signatures = [
  524. Sig('-f', choices='abc'),
  525. Sig('-g', type=int, choices=range(5))]
  526. failures = ['a', '-f d', '-fad', '-ga', '-g 6']
  527. successes = [
  528. ('', NS(f=None, g=None)),
  529. ('-f a', NS(f='a', g=None)),
  530. ('-f c', NS(f='c', g=None)),
  531. ('-g 0', NS(f=None, g=0)),
  532. ('-g 03', NS(f=None, g=3)),
  533. ('-fb -g4', NS(f='b', g=4)),
  534. ]
  535. class TestOptionalsRequired(ParserTestCase):
  536. """Tests an optional action that is required"""
  537. argument_signatures = [
  538. Sig('-x', type=int, required=True),
  539. ]
  540. failures = ['a', '']
  541. successes = [
  542. ('-x 1', NS(x=1)),
  543. ('-x42', NS(x=42)),
  544. ]
  545. class TestOptionalsActionStore(ParserTestCase):
  546. """Tests the store action for an Optional"""
  547. argument_signatures = [Sig('-x', action='store')]
  548. failures = ['a', 'a -x']
  549. successes = [
  550. ('', NS(x=None)),
  551. ('-xfoo', NS(x='foo')),
  552. ]
  553. class TestOptionalsActionStoreConst(ParserTestCase):
  554. """Tests the store_const action for an Optional"""
  555. argument_signatures = [Sig('-y', action='store_const', const=object)]
  556. failures = ['a']
  557. successes = [
  558. ('', NS(y=None)),
  559. ('-y', NS(y=object)),
  560. ]
  561. class TestOptionalsActionStoreFalse(ParserTestCase):
  562. """Tests the store_false action for an Optional"""
  563. argument_signatures = [Sig('-z', action='store_false')]
  564. failures = ['a', '-za', '-z a']
  565. successes = [
  566. ('', NS(z=True)),
  567. ('-z', NS(z=False)),
  568. ]
  569. class TestOptionalsActionStoreTrue(ParserTestCase):
  570. """Tests the store_true action for an Optional"""
  571. argument_signatures = [Sig('--apple', action='store_true')]
  572. failures = ['a', '--apple=b', '--apple b']
  573. successes = [
  574. ('', NS(apple=False)),
  575. ('--apple', NS(apple=True)),
  576. ]
  577. class TestBooleanOptionalAction(ParserTestCase):
  578. """Tests BooleanOptionalAction"""
  579. argument_signatures = [Sig('--foo', action=argparse.BooleanOptionalAction)]
  580. failures = ['--foo bar', '--foo=bar']
  581. successes = [
  582. ('', NS(foo=None)),
  583. ('--foo', NS(foo=True)),
  584. ('--no-foo', NS(foo=False)),
  585. ('--foo --no-foo', NS(foo=False)), # useful for aliases
  586. ('--no-foo --foo', NS(foo=True)),
  587. ]
  588. def test_const(self):
  589. # See bpo-40862
  590. parser = argparse.ArgumentParser()
  591. with self.assertRaises(TypeError) as cm:
  592. parser.add_argument('--foo', const=True, action=argparse.BooleanOptionalAction)
  593. self.assertIn("got an unexpected keyword argument 'const'", str(cm.exception))
  594. class TestBooleanOptionalActionRequired(ParserTestCase):
  595. """Tests BooleanOptionalAction required"""
  596. argument_signatures = [
  597. Sig('--foo', required=True, action=argparse.BooleanOptionalAction)
  598. ]
  599. failures = ['']
  600. successes = [
  601. ('--foo', NS(foo=True)),
  602. ('--no-foo', NS(foo=False)),
  603. ]
  604. class TestOptionalsActionAppend(ParserTestCase):
  605. """Tests the append action for an Optional"""
  606. argument_signatures = [Sig('--baz', action='append')]
  607. failures = ['a', '--baz', 'a --baz', '--baz a b']
  608. successes = [
  609. ('', NS(baz=None)),
  610. ('--baz a', NS(baz=['a'])),
  611. ('--baz a --baz b', NS(baz=['a', 'b'])),
  612. ]
  613. class TestOptionalsActionAppendWithDefault(ParserTestCase):
  614. """Tests the append action for an Optional"""
  615. argument_signatures = [Sig('--baz', action='append', default=['X'])]
  616. failures = ['a', '--baz', 'a --baz', '--baz a b']
  617. successes = [
  618. ('', NS(baz=['X'])),
  619. ('--baz a', NS(baz=['X', 'a'])),
  620. ('--baz a --baz b', NS(baz=['X', 'a', 'b'])),
  621. ]
  622. class TestConstActionsMissingConstKwarg(ParserTestCase):
  623. """Tests that const gets default value of None when not provided"""
  624. argument_signatures = [
  625. Sig('-f', action='append_const'),
  626. Sig('--foo', action='append_const'),
  627. Sig('-b', action='store_const'),
  628. Sig('--bar', action='store_const')
  629. ]
  630. failures = ['-f v', '--foo=bar', '--foo bar']
  631. successes = [
  632. ('', NS(f=None, foo=None, b=None, bar=None)),
  633. ('-f', NS(f=[None], foo=None, b=None, bar=None)),
  634. ('--foo', NS(f=None, foo=[None], b=None, bar=None)),
  635. ('-b', NS(f=None, foo=None, b=None, bar=None)),
  636. ('--bar', NS(f=None, foo=None, b=None, bar=None)),
  637. ]
  638. class TestOptionalsActionAppendConst(ParserTestCase):
  639. """Tests the append_const action for an Optional"""
  640. argument_signatures = [
  641. Sig('-b', action='append_const', const=Exception),
  642. Sig('-c', action='append', dest='b'),
  643. ]
  644. failures = ['a', '-c', 'a -c', '-bx', '-b x']
  645. successes = [
  646. ('', NS(b=None)),
  647. ('-b', NS(b=[Exception])),
  648. ('-b -cx -b -cyz', NS(b=[Exception, 'x', Exception, 'yz'])),
  649. ]
  650. class TestOptionalsActionAppendConstWithDefault(ParserTestCase):
  651. """Tests the append_const action for an Optional"""
  652. argument_signatures = [
  653. Sig('-b', action='append_const', const=Exception, default=['X']),
  654. Sig('-c', action='append', dest='b'),
  655. ]
  656. failures = ['a', '-c', 'a -c', '-bx', '-b x']
  657. successes = [
  658. ('', NS(b=['X'])),
  659. ('-b', NS(b=['X', Exception])),
  660. ('-b -cx -b -cyz', NS(b=['X', Exception, 'x', Exception, 'yz'])),
  661. ]
  662. class TestOptionalsActionCount(ParserTestCase):
  663. """Tests the count action for an Optional"""
  664. argument_signatures = [Sig('-x', action='count')]
  665. failures = ['a', '-x a', '-x b', '-x a -x b']
  666. successes = [
  667. ('', NS(x=None)),
  668. ('-x', NS(x=1)),
  669. ]
  670. class TestOptionalsAllowLongAbbreviation(ParserTestCase):
  671. """Allow long options to be abbreviated unambiguously"""
  672. argument_signatures = [
  673. Sig('--foo'),
  674. Sig('--foobaz'),
  675. Sig('--fooble', action='store_true'),
  676. ]
  677. failures = ['--foob 5', '--foob']
  678. successes = [
  679. ('', NS(foo=None, foobaz=None, fooble=False)),
  680. ('--foo 7', NS(foo='7', foobaz=None, fooble=False)),
  681. ('--fooba a', NS(foo=None, foobaz='a', fooble=False)),
  682. ('--foobl --foo g', NS(foo='g', foobaz=None, fooble=True)),
  683. ]
  684. class TestOptionalsDisallowLongAbbreviation(ParserTestCase):
  685. """Do not allow abbreviations of long options at all"""
  686. parser_signature = Sig(allow_abbrev=False)
  687. argument_signatures = [
  688. Sig('--foo'),
  689. Sig('--foodle', action='store_true'),
  690. Sig('--foonly'),
  691. ]
  692. failures = ['-foon 3', '--foon 3', '--food', '--food --foo 2']
  693. successes = [
  694. ('', NS(foo=None, foodle=False, foonly=None)),
  695. ('--foo 3', NS(foo='3', foodle=False, foonly=None)),
  696. ('--foonly 7 --foodle --foo 2', NS(foo='2', foodle=True, foonly='7')),
  697. ]
  698. class TestOptionalsDisallowLongAbbreviationPrefixChars(ParserTestCase):
  699. """Disallowing abbreviations works with alternative prefix characters"""
  700. parser_signature = Sig(prefix_chars='+', allow_abbrev=False)
  701. argument_signatures = [
  702. Sig('++foo'),
  703. Sig('++foodle', action='store_true'),
  704. Sig('++foonly'),
  705. ]
  706. failures = ['+foon 3', '++foon 3', '++food', '++food ++foo 2']
  707. successes = [
  708. ('', NS(foo=None, foodle=False, foonly=None)),
  709. ('++foo 3', NS(foo='3', foodle=False, foonly=None)),
  710. ('++foonly 7 ++foodle ++foo 2', NS(foo='2', foodle=True, foonly='7')),
  711. ]
  712. class TestDisallowLongAbbreviationAllowsShortGrouping(ParserTestCase):
  713. """Do not allow abbreviations of long options at all"""
  714. parser_signature = Sig(allow_abbrev=False)
  715. argument_signatures = [
  716. Sig('-r'),
  717. Sig('-c', action='count'),
  718. ]
  719. failures = ['-r', '-c -r']
  720. successes = [
  721. ('', NS(r=None, c=None)),
  722. ('-ra', NS(r='a', c=None)),
  723. ('-rcc', NS(r='cc', c=None)),
  724. ('-cc', NS(r=None, c=2)),
  725. ('-cc -ra', NS(r='a', c=2)),
  726. ('-ccrcc', NS(r='cc', c=2)),
  727. ]
  728. class TestDisallowLongAbbreviationAllowsShortGroupingPrefix(ParserTestCase):
  729. """Short option grouping works with custom prefix and allow_abbrev=False"""
  730. parser_signature = Sig(prefix_chars='+', allow_abbrev=False)
  731. argument_signatures = [
  732. Sig('+r'),
  733. Sig('+c', action='count'),
  734. ]
  735. failures = ['+r', '+c +r']
  736. successes = [
  737. ('', NS(r=None, c=None)),
  738. ('+ra', NS(r='a', c=None)),
  739. ('+rcc', NS(r='cc', c=None)),
  740. ('+cc', NS(r=None, c=2)),
  741. ('+cc +ra', NS(r='a', c=2)),
  742. ('+ccrcc', NS(r='cc', c=2)),
  743. ]
  744. # ================
  745. # Positional tests
  746. # ================
  747. class TestPositionalsNargsNone(ParserTestCase):
  748. """Test a Positional that doesn't specify nargs"""
  749. argument_signatures = [Sig('foo')]
  750. failures = ['', '-x', 'a b']
  751. successes = [
  752. ('a', NS(foo='a')),
  753. ]
  754. class TestPositionalsNargs1(ParserTestCase):
  755. """Test a Positional that specifies an nargs of 1"""
  756. argument_signatures = [Sig('foo', nargs=1)]
  757. failures = ['', '-x', 'a b']
  758. successes = [
  759. ('a', NS(foo=['a'])),
  760. ]
  761. class TestPositionalsNargs2(ParserTestCase):
  762. """Test a Positional that specifies an nargs of 2"""
  763. argument_signatures = [Sig('foo', nargs=2)]
  764. failures = ['', 'a', '-x', 'a b c']
  765. successes = [
  766. ('a b', NS(foo=['a', 'b'])),
  767. ]
  768. class TestPositionalsNargsZeroOrMore(ParserTestCase):
  769. """Test a Positional that specifies unlimited nargs"""
  770. argument_signatures = [Sig('foo', nargs='*')]
  771. failures = ['-x']
  772. successes = [
  773. ('', NS(foo=[])),
  774. ('a', NS(foo=['a'])),
  775. ('a b', NS(foo=['a', 'b'])),
  776. ]
  777. class TestPositionalsNargsZeroOrMoreDefault(ParserTestCase):
  778. """Test a Positional that specifies unlimited nargs and a default"""
  779. argument_signatures = [Sig('foo', nargs='*', default='bar')]
  780. failures = ['-x']
  781. successes = [
  782. ('', NS(foo='bar')),
  783. ('a', NS(foo=['a'])),
  784. ('a b', NS(foo=['a', 'b'])),
  785. ]
  786. class TestPositionalsNargsOneOrMore(ParserTestCase):
  787. """Test a Positional that specifies one or more nargs"""
  788. argument_signatures = [Sig('foo', nargs='+')]
  789. failures = ['', '-x']
  790. successes = [
  791. ('a', NS(foo=['a'])),
  792. ('a b', NS(foo=['a', 'b'])),
  793. ]
  794. class TestPositionalsNargsOptional(ParserTestCase):
  795. """Tests an Optional Positional"""
  796. argument_signatures = [Sig('foo', nargs='?')]
  797. failures = ['-x', 'a b']
  798. successes = [
  799. ('', NS(foo=None)),
  800. ('a', NS(foo='a')),
  801. ]
  802. class TestPositionalsNargsOptionalDefault(ParserTestCase):
  803. """Tests an Optional Positional with a default value"""
  804. argument_signatures = [Sig('foo', nargs='?', default=42)]
  805. failures = ['-x', 'a b']
  806. successes = [
  807. ('', NS(foo=42)),
  808. ('a', NS(foo='a')),
  809. ]
  810. class TestPositionalsNargsOptionalConvertedDefault(ParserTestCase):
  811. """Tests an Optional Positional with a default value
  812. that needs to be converted to the appropriate type.
  813. """
  814. argument_signatures = [
  815. Sig('foo', nargs='?', type=int, default='42'),
  816. ]
  817. failures = ['-x', 'a b', '1 2']
  818. successes = [
  819. ('', NS(foo=42)),
  820. ('1', NS(foo=1)),
  821. ]
  822. class TestPositionalsNargsNoneNone(ParserTestCase):
  823. """Test two Positionals that don't specify nargs"""
  824. argument_signatures = [Sig('foo'), Sig('bar')]
  825. failures = ['', '-x', 'a', 'a b c']
  826. successes = [
  827. ('a b', NS(foo='a', bar='b')),
  828. ]
  829. class TestPositionalsNargsNone1(ParserTestCase):
  830. """Test a Positional with no nargs followed by one with 1"""
  831. argument_signatures = [Sig('foo'), Sig('bar', nargs=1)]
  832. failures = ['', '--foo', 'a', 'a b c']
  833. successes = [
  834. ('a b', NS(foo='a', bar=['b'])),
  835. ]
  836. class TestPositionalsNargs2None(ParserTestCase):
  837. """Test a Positional with 2 nargs followed by one with none"""
  838. argument_signatures = [Sig('foo', nargs=2), Sig('bar')]
  839. failures = ['', '--foo', 'a', 'a b', 'a b c d']
  840. successes = [
  841. ('a b c', NS(foo=['a', 'b'], bar='c')),
  842. ]
  843. class TestPositionalsNargsNoneZeroOrMore(ParserTestCase):
  844. """Test a Positional with no nargs followed by one with unlimited"""
  845. argument_signatures = [Sig('foo'), Sig('bar', nargs='*')]
  846. failures = ['', '--foo']
  847. successes = [
  848. ('a', NS(foo='a', bar=[])),
  849. ('a b', NS(foo='a', bar=['b'])),
  850. ('a b c', NS(foo='a', bar=['b', 'c'])),
  851. ]
  852. class TestPositionalsNargsNoneOneOrMore(ParserTestCase):
  853. """Test a Positional with no nargs followed by one with one or more"""
  854. argument_signatures = [Sig('foo'), Sig('bar', nargs='+')]
  855. failures = ['', '--foo', 'a']
  856. successes = [
  857. ('a b', NS(foo='a', bar=['b'])),
  858. ('a b c', NS(foo='a', bar=['b', 'c'])),
  859. ]
  860. class TestPositionalsNargsNoneOptional(ParserTestCase):
  861. """Test a Positional with no nargs followed by one with an Optional"""
  862. argument_signatures = [Sig('foo'), Sig('bar', nargs='?')]
  863. failures = ['', '--foo', 'a b c']
  864. successes = [
  865. ('a', NS(foo='a', bar=None)),
  866. ('a b', NS(foo='a', bar='b')),
  867. ]
  868. class TestPositionalsNargsZeroOrMoreNone(ParserTestCase):
  869. """Test a Positional with unlimited nargs followed by one with none"""
  870. argument_signatures = [Sig('foo', nargs='*'), Sig('bar')]
  871. failures = ['', '--foo']
  872. successes = [
  873. ('a', NS(foo=[], bar='a')),
  874. ('a b', NS(foo=['a'], bar='b')),
  875. ('a b c', NS(foo=['a', 'b'], bar='c')),
  876. ]
  877. class TestPositionalsNargsOneOrMoreNone(ParserTestCase):
  878. """Test a Positional with one or more nargs followed by one with none"""
  879. argument_signatures = [Sig('foo', nargs='+'), Sig('bar')]
  880. failures = ['', '--foo', 'a']
  881. successes = [
  882. ('a b', NS(foo=['a'], bar='b')),
  883. ('a b c', NS(foo=['a', 'b'], bar='c')),
  884. ]
  885. class TestPositionalsNargsOptionalNone(ParserTestCase):
  886. """Test a Positional with an Optional nargs followed by one with none"""
  887. argument_signatures = [Sig('foo', nargs='?', default=42), Sig('bar')]
  888. failures = ['', '--foo', 'a b c']
  889. successes = [
  890. ('a', NS(foo=42, bar='a')),
  891. ('a b', NS(foo='a', bar='b')),
  892. ]
  893. class TestPositionalsNargs2ZeroOrMore(ParserTestCase):
  894. """Test a Positional with 2 nargs followed by one with unlimited"""
  895. argument_signatures = [Sig('foo', nargs=2), Sig('bar', nargs='*')]
  896. failures = ['', '--foo', 'a']
  897. successes = [
  898. ('a b', NS(foo=['a', 'b'], bar=[])),
  899. ('a b c', NS(foo=['a', 'b'], bar=['c'])),
  900. ]
  901. class TestPositionalsNargs2OneOrMore(ParserTestCase):
  902. """Test a Positional with 2 nargs followed by one with one or more"""
  903. argument_signatures = [Sig('foo', nargs=2), Sig('bar', nargs='+')]
  904. failures = ['', '--foo', 'a', 'a b']
  905. successes = [
  906. ('a b c', NS(foo=['a', 'b'], bar=['c'])),
  907. ]
  908. class TestPositionalsNargs2Optional(ParserTestCase):
  909. """Test a Positional with 2 nargs followed by one optional"""
  910. argument_signatures = [Sig('foo', nargs=2), Sig('bar', nargs='?')]
  911. failures = ['', '--foo', 'a', 'a b c d']
  912. successes = [
  913. ('a b', NS(foo=['a', 'b'], bar=None)),
  914. ('a b c', NS(foo=['a', 'b'], bar='c')),
  915. ]
  916. class TestPositionalsNargsZeroOrMore1(ParserTestCase):
  917. """Test a Positional with unlimited nargs followed by one with 1"""
  918. argument_signatures = [Sig('foo', nargs='*'), Sig('bar', nargs=1)]
  919. failures = ['', '--foo', ]
  920. successes = [
  921. ('a', NS(foo=[], bar=['a'])),
  922. ('a b', NS(foo=['a'], bar=['b'])),
  923. ('a b c', NS(foo=['a', 'b'], bar=['c'])),
  924. ]
  925. class TestPositionalsNargsOneOrMore1(ParserTestCase):
  926. """Test a Positional with one or more nargs followed by one with 1"""
  927. argument_signatures = [Sig('foo', nargs='+'), Sig('bar', nargs=1)]
  928. failures = ['', '--foo', 'a']
  929. successes = [
  930. ('a b', NS(foo=['a'], bar=['b'])),
  931. ('a b c', NS(foo=['a', 'b'], bar=['c'])),
  932. ]
  933. class TestPositionalsNargsOptional1(ParserTestCase):
  934. """Test a Positional with an Optional nargs followed by one with 1"""
  935. argument_signatures = [Sig('foo', nargs='?'), Sig('bar', nargs=1)]
  936. failures = ['', '--foo', 'a b c']
  937. successes = [
  938. ('a', NS(foo=None, bar=['a'])),
  939. ('a b', NS(foo='a', bar=['b'])),
  940. ]
  941. class TestPositionalsNargsNoneZeroOrMore1(ParserTestCase):
  942. """Test three Positionals: no nargs, unlimited nargs and 1 nargs"""
  943. argument_signatures = [
  944. Sig('foo'),
  945. Sig('bar', nargs='*'),
  946. Sig('baz', nargs=1),
  947. ]
  948. failures = ['', '--foo', 'a']
  949. successes = [
  950. ('a b', NS(foo='a', bar=[], baz=['b'])),
  951. ('a b c', NS(foo='a', bar=['b'], baz=['c'])),
  952. ]
  953. class TestPositionalsNargsNoneOneOrMore1(ParserTestCase):
  954. """Test three Positionals: no nargs, one or more nargs and 1 nargs"""
  955. argument_signatures = [
  956. Sig('foo'),
  957. Sig('bar', nargs='+'),
  958. Sig('baz', nargs=1),
  959. ]
  960. failures = ['', '--foo', 'a', 'b']
  961. successes = [
  962. ('a b c', NS(foo='a', bar=['b'], baz=['c'])),
  963. ('a b c d', NS(foo='a', bar=['b', 'c'], baz=['d'])),
  964. ]
  965. class TestPositionalsNargsNoneOptional1(ParserTestCase):
  966. """Test three Positionals: no nargs, optional narg and 1 nargs"""
  967. argument_signatures = [
  968. Sig('foo'),
  969. Sig('bar', nargs='?', default=0.625),
  970. Sig('baz', nargs=1),
  971. ]
  972. failures = ['', '--foo', 'a']
  973. successes = [
  974. ('a b', NS(foo='a', bar=0.625, baz=['b'])),
  975. ('a b c', NS(foo='a', bar='b', baz=['c'])),
  976. ]
  977. class TestPositionalsNargsOptionalOptional(ParserTestCase):
  978. """Test two optional nargs"""
  979. argument_signatures = [
  980. Sig('foo', nargs='?'),
  981. Sig('bar', nargs='?', default=42),
  982. ]
  983. failures = ['--foo', 'a b c']
  984. successes = [
  985. ('', NS(foo=None, bar=42)),
  986. ('a', NS(foo='a', bar=42)),
  987. ('a b', NS(foo='a', bar='b')),
  988. ]
  989. class TestPositionalsNargsOptionalZeroOrMore(ParserTestCase):
  990. """Test an Optional narg followed by unlimited nargs"""
  991. argument_signatures = [Sig('foo', nargs='?'), Sig('bar', nargs='*')]
  992. failures = ['--foo']
  993. successes = [
  994. ('', NS(foo=None, bar=[])),
  995. ('a', NS(foo='a', bar=[])),
  996. ('a b', NS(foo='a', bar=['b'])),
  997. ('a b c', NS(foo='a', bar=['b', 'c'])),
  998. ]
  999. class TestPositionalsNargsOptionalOneOrMore(ParserTestCase):
  1000. """Test an Optional narg followed by one or more nargs"""
  1001. argument_signatures = [Sig('foo', nargs='?'), Sig('bar', nargs='+')]
  1002. failures = ['', '--foo']
  1003. successes = [
  1004. ('a', NS(foo=None, bar=['a'])),
  1005. ('a b', NS(foo='a', bar=['b'])),
  1006. ('a b c', NS(foo='a', bar=['b', 'c'])),
  1007. ]
  1008. class TestPositionalsChoicesString(ParserTestCase):
  1009. """Test a set of single-character choices"""
  1010. argument_signatures = [Sig('spam', choices=set('abcdefg'))]
  1011. failures = ['', '--foo', 'h', '42', 'ef']
  1012. successes = [
  1013. ('a', NS(spam='a')),
  1014. ('g', NS(spam='g')),
  1015. ]
  1016. class TestPositionalsChoicesInt(ParserTestCase):
  1017. """Test a set of integer choices"""
  1018. argument_signatures = [Sig('spam', type=int, choices=range(20))]
  1019. failures = ['', '--foo', 'h', '42', 'ef']
  1020. successes = [
  1021. ('4', NS(spam=4)),
  1022. ('15', NS(spam=15)),
  1023. ]
  1024. class TestPositionalsActionAppend(ParserTestCase):
  1025. """Test the 'append' action"""
  1026. argument_signatures = [
  1027. Sig('spam', action='append'),
  1028. Sig('spam', action='append', nargs=2),
  1029. ]
  1030. failures = ['', '--foo', 'a', 'a b', 'a b c d']
  1031. successes = [
  1032. ('a b c', NS(spam=['a', ['b', 'c']])),
  1033. ]
  1034. # ========================================
  1035. # Combined optionals and positionals tests
  1036. # ========================================
  1037. class TestOptionalsNumericAndPositionals(ParserTestCase):
  1038. """Tests negative number args when numeric options are present"""
  1039. argument_signatures = [
  1040. Sig('x', nargs='?'),
  1041. Sig('-4', dest='y', action='store_true'),
  1042. ]
  1043. failures = ['-2', '-315']
  1044. successes = [
  1045. ('', NS(x=None, y=False)),
  1046. ('a', NS(x='a', y=False)),
  1047. ('-4', NS(x=None, y=True)),
  1048. ('-4 a', NS(x='a', y=True)),
  1049. ]
  1050. class TestOptionalsAlmostNumericAndPositionals(ParserTestCase):
  1051. """Tests negative number args when almost numeric options are present"""
  1052. argument_signatures = [
  1053. Sig('x', nargs='?'),
  1054. Sig('-k4', dest='y', action='store_true'),
  1055. ]
  1056. failures = ['-k3']
  1057. successes = [
  1058. ('', NS(x=None, y=False)),
  1059. ('-2', NS(x='-2', y=False)),
  1060. ('a', NS(x='a', y=False)),
  1061. ('-k4', NS(x=None, y=True)),
  1062. ('-k4 a', NS(x='a', y=True)),
  1063. ]
  1064. class TestEmptyAndSpaceContainingArguments(ParserTestCase):
  1065. argument_signatures = [
  1066. Sig('x', nargs='?'),
  1067. Sig('-y', '--yyy', dest='y'),
  1068. ]
  1069. failures = ['-y']
  1070. successes = [
  1071. ([''], NS(x='', y=None)),
  1072. (['a badger'], NS(x='a badger', y=None)),
  1073. (['-a badger'], NS(x='-a badger', y=None)),
  1074. (['-y', ''], NS(x=None, y='')),
  1075. (['-y', 'a badger'], NS(x=None, y='a badger')),
  1076. (['-y', '-a badger'], NS(x=None, y='-a badger')),
  1077. (['--yyy=a badger'], NS(x=None, y='a badger')),
  1078. (['--yyy=-a badger'], NS(x=None, y='-a badger')),
  1079. ]
  1080. class TestPrefixCharacterOnlyArguments(ParserTestCase):
  1081. parser_signature = Sig(prefix_chars='-+')
  1082. argument_signatures = [
  1083. Sig('-', dest='x', nargs='?', const='badger'),
  1084. Sig('+', dest='y', type=int, default=42),
  1085. Sig('-+-', dest='z', action='store_true'),
  1086. ]
  1087. failures = ['-y', '+ -']
  1088. successes = [
  1089. ('', NS(x=None, y=42, z=False)),
  1090. ('-', NS(x='badger', y=42, z=False)),
  1091. ('- X', NS(x='X', y=42, z=False)),
  1092. ('+ -3', NS(x=None, y=-3, z=False)),
  1093. ('-+-', NS(x=None, y=42, z=True)),
  1094. ('- ===', NS(x='===', y=42, z=False)),
  1095. ]
  1096. class TestNargsZeroOrMore(ParserTestCase):
  1097. """Tests specifying args for an Optional that accepts zero or more"""
  1098. argument_signatures = [Sig('-x', nargs='*'), Sig('y', nargs='*')]
  1099. failures = []
  1100. successes = [
  1101. ('', NS(x=None, y=[])),
  1102. ('-x', NS(x=[], y=[])),
  1103. ('-x a', NS(x=['a'], y=[])),
  1104. ('-x a -- b', NS(x=['a'], y=['b'])),
  1105. ('a', NS(x=None, y=['a'])),
  1106. ('a -x', NS(x=[], y=['a'])),
  1107. ('a -x b', NS(x=['b'], y=['a'])),
  1108. ]
  1109. class TestNargsRemainder(ParserTestCase):
  1110. """Tests specifying a positional with nargs=REMAINDER"""
  1111. argument_signatures = [Sig('x'), Sig('y', nargs='...'), Sig('-z')]
  1112. failures = ['', '-z', '-z Z']
  1113. successes = [
  1114. ('X', NS(x='X', y=[], z=None)),
  1115. ('-z Z X', NS(x='X', y=[], z='Z')),
  1116. ('X A B -z Z', NS(x='X', y=['A', 'B', '-z', 'Z'], z=None)),
  1117. ('X Y --foo', NS(x='X', y=['Y', '--foo'], z=None)),
  1118. ]
  1119. class TestOptionLike(ParserTestCase):
  1120. """Tests options that may or may not be arguments"""
  1121. argument_signatures = [
  1122. Sig('-x', type=float),
  1123. Sig('-3', type=float, dest='y'),
  1124. Sig('z', nargs='*'),
  1125. ]
  1126. failures = ['-x', '-y2.5', '-xa', '-x -a',
  1127. '-x -3', '-x -3.5', '-3 -3.5',
  1128. '-x -2.5', '-x -2.5 a', '-3 -.5',
  1129. 'a x -1', '-x -1 a', '-3 -1 a']
  1130. successes = [
  1131. ('', NS(x=None, y=None, z=[])),
  1132. ('-x 2.5', NS(x=2.5, y=None, z=[])),
  1133. ('-x 2.5 a', NS(x=2.5, y=None, z=['a'])),
  1134. ('-3.5', NS(x=None, y=0.5, z=[])),
  1135. ('-3-.5', NS(x=None, y=-0.5, z=[])),
  1136. ('-3 .5', NS(x=None, y=0.5, z=[])),
  1137. ('a -3.5', NS(x=None, y=0.5, z=['a'])),
  1138. ('a', NS(x=None, y=None, z=['a'])),
  1139. ('a -x 1', NS(x=1.0, y=None, z=['a'])),
  1140. ('-x 1 a', NS(x=1.0, y=None, z=['a'])),
  1141. ('-3 1 a', NS(x=None, y=1.0, z=['a'])),
  1142. ]
  1143. class TestDefaultSuppress(ParserTestCase):
  1144. """Test actions with suppressed defaults"""
  1145. argument_signatures = [
  1146. Sig('foo', nargs='?', default=argparse.SUPPRESS),
  1147. Sig('bar', nargs='*', default=argparse.SUPPRESS),
  1148. Sig('--baz', action='store_true', default=argparse.SUPPRESS),
  1149. ]
  1150. failures = ['-x']
  1151. successes = [
  1152. ('', NS()),
  1153. ('a', NS(foo='a')),
  1154. ('a b', NS(foo='a', bar=['b'])),
  1155. ('--baz', NS(baz=True)),
  1156. ('a --baz', NS(foo='a', baz=True)),
  1157. ('--baz a b', NS(foo='a', bar=['b'], baz=True)),
  1158. ]
  1159. class TestParserDefaultSuppress(ParserTestCase):
  1160. """Test actions with a parser-level default of SUPPRESS"""
  1161. parser_signature = Sig(argument_default=argparse.SUPPRESS)
  1162. argument_signatures = [
  1163. Sig('foo', nargs='?'),
  1164. Sig('bar', nargs='*'),
  1165. Sig('--baz', action='store_true'),
  1166. ]
  1167. failures = ['-x']
  1168. successes = [
  1169. ('', NS()),
  1170. ('a', NS(foo='a')),
  1171. ('a b', NS(foo='a', bar=['b'])),
  1172. ('--baz', NS(baz=True)),
  1173. ('a --baz', NS(foo='a', baz=True)),
  1174. ('--baz a b', NS(foo='a', bar=['b'], baz=True)),
  1175. ]
  1176. class TestParserDefault42(ParserTestCase):
  1177. """Test actions with a parser-level default of 42"""
  1178. parser_signature = Sig(argument_default=42)
  1179. argument_signatures = [
  1180. Sig('--version', action='version', version='1.0'),
  1181. Sig('foo', nargs='?'),
  1182. Sig('bar', nargs='*'),
  1183. Sig('--baz', action='store_true'),
  1184. ]
  1185. failures = ['-x']
  1186. successes = [
  1187. ('', NS(foo=42, bar=42, baz=42, version=42)),
  1188. ('a', NS(foo='a', bar=42, baz=42, version=42)),
  1189. ('a b', NS(foo='a', bar=['b'], baz=42, version=42)),
  1190. ('--baz', NS(foo=42, bar=42, baz=True, version=42)),
  1191. ('a --baz', NS(foo='a', bar=42, baz=True, version=42)),
  1192. ('--baz a b', NS(foo='a', bar=['b'], baz=True, version=42)),
  1193. ]
  1194. class TestArgumentsFromFile(TempDirMixin, ParserTestCase):
  1195. """Test reading arguments from a file"""
  1196. def setUp(self):
  1197. super(TestArgumentsFromFile, self).setUp()
  1198. file_texts = [
  1199. ('hello', 'hello world!\n'),
  1200. ('recursive', '-a\n'
  1201. 'A\n'
  1202. '@hello'),
  1203. ('invalid', '@no-such-path\n'),
  1204. ]
  1205. for path, text in file_texts:
  1206. with open(path, 'w', encoding="utf-8") as file:
  1207. file.write(text)
  1208. parser_signature = Sig(fromfile_prefix_chars='@')
  1209. argument_signatures = [
  1210. Sig('-a'),
  1211. Sig('x'),
  1212. Sig('y', nargs='+'),
  1213. ]
  1214. failures = ['', '-b', 'X', '@invalid', '@missing']
  1215. successes = [
  1216. ('X Y', NS(a=None, x='X', y=['Y'])),
  1217. ('X -a A Y Z', NS(a='A', x='X', y=['Y', 'Z'])),
  1218. ('@hello X', NS(a=None, x='hello world!', y=['X'])),
  1219. ('X @hello', NS(a=None, x='X', y=['hello world!'])),
  1220. ('-a B @recursive Y Z', NS(a='A', x='hello world!', y=['Y', 'Z'])),
  1221. ('X @recursive Z -a B', NS(a='B', x='X', y=['hello world!', 'Z'])),
  1222. (["-a", "", "X", "Y"], NS(a='', x='X', y=['Y'])),
  1223. ]
  1224. class TestArgumentsFromFileConverter(TempDirMixin, ParserTestCase):
  1225. """Test reading arguments from a file"""
  1226. def setUp(self):
  1227. super(TestArgumentsFromFileConverter, self).setUp()
  1228. file_texts = [
  1229. ('hello', 'hello world!\n'),
  1230. ]
  1231. for path, text in file_texts:
  1232. with open(path, 'w', encoding="utf-8") as file:
  1233. file.write(text)
  1234. class FromFileConverterArgumentParser(ErrorRaisingArgumentParser):
  1235. def convert_arg_line_to_args(self, arg_line):
  1236. for arg in arg_line.split():
  1237. if not arg.strip():
  1238. continue
  1239. yield arg
  1240. parser_class = FromFileConverterArgumentParser
  1241. parser_signature = Sig(fromfile_prefix_chars='@')
  1242. argument_signatures = [
  1243. Sig('y', nargs='+'),
  1244. ]
  1245. failures = []
  1246. successes = [
  1247. ('@hello X', NS(y=['hello', 'world!', 'X'])),
  1248. ]
  1249. # =====================
  1250. # Type conversion tests
  1251. # =====================
  1252. class TestFileTypeRepr(TestCase):
  1253. def test_r(self):
  1254. type = argparse.FileType('r')
  1255. self.assertEqual("FileType('r')", repr(type))
  1256. def test_wb_1(self):
  1257. type = argparse.FileType('wb', 1)
  1258. self.assertEqual("FileType('wb', 1)", repr(type))
  1259. def test_r_latin(self):
  1260. type = argparse.FileType('r', encoding='latin_1')
  1261. self.assertEqual("FileType('r', encoding='latin_1')", repr(type))
  1262. def test_w_big5_ignore(self):
  1263. type = argparse.FileType('w', encoding='big5', errors='ignore')
  1264. self.assertEqual("FileType('w', encoding='big5', errors='ignore')",
  1265. repr(type))
  1266. def test_r_1_replace(self):
  1267. type = argparse.FileType('r', 1, errors='replace')
  1268. self.assertEqual("FileType('r', 1, errors='replace')", repr(type))
  1269. BIN_STDOUT_SENTINEL = object()
  1270. BIN_STDERR_SENTINEL = object()
  1271. class StdStreamComparer:
  1272. def __init__(self, attr):
  1273. # We try to use the actual stdXXX.buffer attribute as our
  1274. # marker, but but under some test environments,
  1275. # sys.stdout/err are replaced by io.StringIO which won't have .buffer,
  1276. # so we use a sentinel simply to show that the tests do the right thing
  1277. # for any buffer supporting object
  1278. self.getattr = operator.attrgetter(attr)
  1279. if attr == 'stdout.buffer':
  1280. self.backupattr = BIN_STDOUT_SENTINEL
  1281. elif attr == 'stderr.buffer':
  1282. self.backupattr = BIN_STDERR_SENTINEL
  1283. else:
  1284. self.backupattr = object() # Not equal to anything
  1285. def __eq__(self, other):
  1286. try:
  1287. return other == self.getattr(sys)
  1288. except AttributeError:
  1289. return other == self.backupattr
  1290. eq_stdin = StdStreamComparer('stdin')
  1291. eq_stdout = StdStreamComparer('stdout')
  1292. eq_stderr = StdStreamComparer('stderr')
  1293. eq_bstdin = StdStreamComparer('stdin.buffer')
  1294. eq_bstdout = StdStreamComparer('stdout.buffer')
  1295. eq_bstderr = StdStreamComparer('stderr.buffer')
  1296. class RFile(object):
  1297. seen = {}
  1298. def __init__(self, name):
  1299. self.name = name
  1300. def __eq__(self, other):
  1301. if other in self.seen:
  1302. text = self.seen[other]
  1303. else:
  1304. text = self.seen[other] = other.read()
  1305. other.close()
  1306. if not isinstance(text, str):
  1307. text = text.decode('ascii')
  1308. return self.name == other.name == text
  1309. class TestFileTypeR(TempDirMixin, ParserTestCase):
  1310. """Test the FileType option/argument type for reading files"""
  1311. def setUp(self):
  1312. super(TestFileTypeR, self).setUp()
  1313. for file_name in ['foo', 'bar']:
  1314. with open(os.path.join(self.temp_dir, file_name),
  1315. 'w', encoding="utf-8") as file:
  1316. file.write(file_name)
  1317. self.create_readonly_file('readonly')
  1318. argument_signatures = [
  1319. Sig('-x', type=argparse.FileType()),
  1320. Sig('spam', type=argparse.FileType('r')),
  1321. ]
  1322. failures = ['-x', '', 'non-existent-file.txt']
  1323. successes = [
  1324. ('foo', NS(x=None, spam=RFile('foo'))),
  1325. ('-x foo bar', NS(x=RFile('foo'), spam=RFile('bar'))),
  1326. ('bar -x foo', NS(x=RFile('foo'), spam=RFile('bar'))),
  1327. ('-x - -', NS(x=eq_stdin, spam=eq_stdin)),
  1328. ('readonly', NS(x=None, spam=RFile('readonly'))),
  1329. ]
  1330. class TestFileTypeDefaults(TempDirMixin, ParserTestCase):
  1331. """Test that a file is not created unless the default is needed"""
  1332. def setUp(self):
  1333. super(TestFileTypeDefaults, self).setUp()
  1334. file = open(os.path.join(self.temp_dir, 'good'), 'w', encoding="utf-8")
  1335. file.write('good')
  1336. file.close()
  1337. argument_signatures = [
  1338. Sig('-c', type=argparse.FileType('r'), default='no-file.txt'),
  1339. ]
  1340. # should provoke no such file error
  1341. failures = ['']
  1342. # should not provoke error because default file is created
  1343. successes = [('-c good', NS(c=RFile('good')))]
  1344. class TestFileTypeRB(TempDirMixin, ParserTestCase):
  1345. """Test the FileType option/argument type for reading files"""
  1346. def setUp(self):
  1347. super(TestFileTypeRB, self).setUp()
  1348. for file_name in ['foo', 'bar']:
  1349. with open(os.path.join(self.temp_dir, file_name),
  1350. 'w', encoding="utf-8") as file:
  1351. file.write(file_name)
  1352. argument_signatures = [
  1353. Sig('-x', type=argparse.FileType('rb')),
  1354. Sig('spam', type=argparse.FileType('rb')),
  1355. ]
  1356. failures = ['-x', '']
  1357. successes = [
  1358. ('foo', NS(x=None, spam=RFile('foo'))),
  1359. ('-x foo bar', NS(x=RFile('foo'), spam=RFile('bar'))),
  1360. ('bar -x foo', NS(x=RFile('foo'), spam=RFile('bar'))),
  1361. ('-x - -', NS(x=eq_bstdin, spam=eq_bstdin)),
  1362. ]
  1363. class WFile(object):
  1364. seen = set()
  1365. def __init__(self, name):
  1366. self.name = name
  1367. def __eq__(self, other):
  1368. if other not in self.seen:
  1369. text = 'Check that file is writable.'
  1370. if 'b' in other.mode:
  1371. text = text.encode('ascii')
  1372. other.write(text)
  1373. other.close()
  1374. self.seen.add(other)
  1375. return self.name == other.name
  1376. @os_helper.skip_if_dac_override
  1377. class TestFileTypeW(TempDirMixin, ParserTestCase):
  1378. """Test the FileType option/argument type for writing files"""
  1379. def setUp(self):
  1380. super().setUp()
  1381. self.create_readonly_file('readonly')
  1382. self.create_writable_file('writable')
  1383. argument_signatures = [
  1384. Sig('-x', type=argparse.FileType('w')),
  1385. Sig('spam', type=argparse.FileType('w')),
  1386. ]
  1387. failures = ['-x', '', 'readonly']
  1388. successes = [
  1389. ('foo', NS(x=None, spam=WFile('foo'))),
  1390. ('writable', NS(x=None, spam=WFile('writable'))),
  1391. ('-x foo bar', NS(x=WFile('foo'), spam=WFile('bar'))),
  1392. ('bar -x foo', NS(x=WFile('foo'), spam=WFile('bar'))),
  1393. ('-x - -', NS(x=eq_stdout, spam=eq_stdout)),
  1394. ]
  1395. @os_helper.skip_if_dac_override
  1396. class TestFileTypeX(TempDirMixin, ParserTestCase):
  1397. """Test the FileType option/argument type for writing new files only"""
  1398. def setUp(self):
  1399. super().setUp()
  1400. self.create_readonly_file('readonly')
  1401. self.create_writable_file('writable')
  1402. argument_signatures = [
  1403. Sig('-x', type=argparse.FileType('x')),
  1404. Sig('spam', type=argparse.FileType('x')),
  1405. ]
  1406. failures = ['-x', '', 'readonly', 'writable']
  1407. successes = [
  1408. ('-x foo bar', NS(x=WFile('foo'), spam=WFile('bar'))),
  1409. ('-x - -', NS(x=eq_stdout, spam=eq_stdout)),
  1410. ]
  1411. @os_helper.skip_if_dac_override
  1412. class TestFileTypeWB(TempDirMixin, ParserTestCase):
  1413. """Test the FileType option/argument type for writing binary files"""
  1414. argument_signatures = [
  1415. Sig('-x', type=argparse.FileType('wb')),
  1416. Sig('spam', type=argparse.FileType('wb')),
  1417. ]
  1418. failures = ['-x', '']
  1419. successes = [
  1420. ('foo', NS(x=None, spam=WFile('foo'))),
  1421. ('-x foo bar', NS(x=WFile('foo'), spam=WFile('bar'))),
  1422. ('bar -x foo', NS(x=WFile('foo'), spam=WFile('bar'))),
  1423. ('-x - -', NS(x=eq_bstdout, spam=eq_bstdout)),
  1424. ]
  1425. @os_helper.skip_if_dac_override
  1426. class TestFileTypeXB(TestFileTypeX):
  1427. "Test the FileType option/argument type for writing new binary files only"
  1428. argument_signatures = [
  1429. Sig('-x', type=argparse.FileType('xb')),
  1430. Sig('spam', type=argparse.FileType('xb')),
  1431. ]
  1432. successes = [
  1433. ('-x foo bar', NS(x=WFile('foo'), spam=WFile('bar'))),
  1434. ('-x - -', NS(x=eq_bstdout, spam=eq_bstdout)),
  1435. ]
  1436. class TestFileTypeOpenArgs(TestCase):
  1437. """Test that open (the builtin) is correctly called"""
  1438. def test_open_args(self):
  1439. FT = argparse.FileType
  1440. cases = [
  1441. (FT('rb'), ('rb', -1, None, None)),
  1442. (FT('w', 1), ('w', 1, None, None)),
  1443. (FT('w', errors='replace'), ('w', -1, None, 'replace')),
  1444. (FT('wb', encoding='big5'), ('wb', -1, 'big5', None)),
  1445. (FT('w', 0, 'l1', 'strict'), ('w', 0, 'l1', 'strict')),
  1446. ]
  1447. with mock.patch('builtins.open') as m:
  1448. for type, args in cases:
  1449. type('foo')
  1450. m.assert_called_with('foo', *args)
  1451. class TestFileTypeMissingInitialization(TestCase):
  1452. """
  1453. Test that add_argument throws an error if FileType class
  1454. object was passed instead of instance of FileType
  1455. """
  1456. def test(self):
  1457. parser = argparse.ArgumentParser()
  1458. with self.assertRaises(ValueError) as cm:
  1459. parser.add_argument('-x', type=argparse.FileType)
  1460. self.assertEqual(
  1461. '%r is a FileType class object, instance of it must be passed'
  1462. % (argparse.FileType,),
  1463. str(cm.exception)
  1464. )
  1465. class TestTypeCallable(ParserTestCase):
  1466. """Test some callables as option/argument types"""
  1467. argument_signatures = [
  1468. Sig('--eggs', type=complex),
  1469. Sig('spam', type=float),
  1470. ]
  1471. failures = ['a', '42j', '--eggs a', '--eggs 2i']
  1472. successes = [
  1473. ('--eggs=42 42', NS(eggs=42, spam=42.0)),
  1474. ('--eggs 2j -- -1.5', NS(eggs=2j, spam=-1.5)),
  1475. ('1024.675', NS(eggs=None, spam=1024.675)),
  1476. ]
  1477. class TestTypeUserDefined(ParserTestCase):
  1478. """Test a user-defined option/argument type"""
  1479. class MyType(TestCase):
  1480. def __init__(self, value):
  1481. self.value = value
  1482. def __eq__(self, other):
  1483. return (type(self), self.value) == (type(other), other.value)
  1484. argument_signatures = [
  1485. Sig('-x', type=MyType),
  1486. Sig('spam', type=MyType),
  1487. ]
  1488. failures = []
  1489. successes = [
  1490. ('a -x b', NS(x=MyType('b'), spam=MyType('a'))),
  1491. ('-xf g', NS(x=MyType('f'), spam=MyType('g'))),
  1492. ]
  1493. class TestTypeClassicClass(ParserTestCase):
  1494. """Test a classic class type"""
  1495. class C:
  1496. def __init__(self, value):
  1497. self.value = value
  1498. def __eq__(self, other):
  1499. return (type(self), self.value) == (type(other), other.value)
  1500. argument_signatures = [
  1501. Sig('-x', type=C),
  1502. Sig('spam', type=C),
  1503. ]
  1504. failures = []
  1505. successes = [
  1506. ('a -x b', NS(x=C('b'), spam=C('a'))),
  1507. ('-xf g', NS(x=C('f'), spam=C('g'))),
  1508. ]
  1509. class TestTypeRegistration(TestCase):
  1510. """Test a user-defined type by registering it"""
  1511. def test(self):
  1512. def get_my_type(string):
  1513. return 'my_type{%s}' % string
  1514. parser = argparse.ArgumentParser()
  1515. parser.register('type', 'my_type', get_my_type)
  1516. parser.add_argument('-x', type='my_type')
  1517. parser.add_argument('y', type='my_type')
  1518. self.assertEqual(parser.parse_args('1'.split()),
  1519. NS(x=None, y='my_type{1}'))
  1520. self.assertEqual(parser.parse_args('-x 1 42'.split()),
  1521. NS(x='my_type{1}', y='my_type{42}'))
  1522. # ============
  1523. # Action tests
  1524. # ============
  1525. class TestActionUserDefined(ParserTestCase):
  1526. """Test a user-defined option/argument action"""
  1527. class OptionalAction(argparse.Action):
  1528. def __call__(self, parser, namespace, value, option_string=None):
  1529. try:
  1530. # check destination and option string
  1531. assert self.dest == 'spam', 'dest: %s' % self.dest
  1532. assert option_string == '-s', 'flag: %s' % option_string
  1533. # when option is before argument, badger=2, and when
  1534. # option is after argument, badger=<whatever was set>
  1535. expected_ns = NS(spam=0.25)
  1536. if value in [0.125, 0.625]:
  1537. expected_ns.badger = 2
  1538. elif value in [2.0]:
  1539. expected_ns.badger = 84
  1540. else:
  1541. raise AssertionError('value: %s' % value)
  1542. assert expected_ns == namespace, ('expected %s, got %s' %
  1543. (expected_ns, namespace))
  1544. except AssertionError as e:
  1545. raise ArgumentParserError('opt_action failed: %s' % e)
  1546. setattr(namespace, 'spam', value)
  1547. class PositionalAction(argparse.Action):
  1548. def __call__(self, parser, namespace, value, option_string=None):
  1549. try:
  1550. assert option_string is None, ('option_string: %s' %
  1551. option_string)
  1552. # check destination
  1553. assert self.dest == 'badger', 'dest: %s' % self.dest
  1554. # when argument is before option, spam=0.25, and when
  1555. # option is after argument, spam=<whatever was set>
  1556. expected_ns = NS(badger=2)
  1557. if value in [42, 84]:
  1558. expected_ns.spam = 0.25
  1559. elif value in [1]:
  1560. expected_ns.spam = 0.625
  1561. elif value in [2]:
  1562. expected_ns.spam = 0.125
  1563. else:
  1564. raise AssertionError('value: %s' % value)
  1565. assert expected_ns == namespace, ('expected %s, got %s' %
  1566. (expected_ns, namespace))
  1567. except AssertionError as e:
  1568. raise ArgumentParserError('arg_action failed: %s' % e)
  1569. setattr(namespace, 'badger', value)
  1570. argument_signatures = [
  1571. Sig('-s', dest='spam', action=OptionalAction,
  1572. type=float, default=0.25),
  1573. Sig('badger', action=PositionalAction,
  1574. type=int, nargs='?', default=2),
  1575. ]
  1576. failures = []
  1577. successes = [
  1578. ('-s0.125', NS(spam=0.125, badger=2)),
  1579. ('42', NS(spam=0.25, badger=42)),
  1580. ('-s 0.625 1', NS(spam=0.625, badger=1)),
  1581. ('84 -s2', NS(spam=2.0, badger=84)),
  1582. ]
  1583. class TestActionRegistration(TestCase):
  1584. """Test a user-defined action supplied by registering it"""
  1585. class MyAction(argparse.Action):
  1586. def __call__(self, parser, namespace, values, option_string=None):
  1587. setattr(namespace, self.dest, 'foo[%s]' % values)
  1588. def test(self):
  1589. parser = argparse.ArgumentParser()
  1590. parser.register('action', 'my_action', self.MyAction)
  1591. parser.add_argument('badger', action='my_action')
  1592. self.assertEqual(parser.parse_args(['1']), NS(badger='foo[1]'))
  1593. self.assertEqual(parser.parse_args(['42']), NS(badger='foo[42]'))
  1594. class TestActionExtend(ParserTestCase):
  1595. argument_signatures = [
  1596. Sig('--foo', action="extend", nargs="+", type=str),
  1597. ]
  1598. failures = ()
  1599. successes = [
  1600. ('--foo f1 --foo f2 f3 f4', NS(foo=['f1', 'f2', 'f3', 'f4'])),
  1601. ]
  1602. # ================
  1603. # Subparsers tests
  1604. # ================
  1605. class TestAddSubparsers(TestCase):
  1606. """Test the add_subparsers method"""
  1607. def assertArgumentParserError(self, *args, **kwargs):
  1608. self.assertRaises(ArgumentParserError, *args, **kwargs)
  1609. def _get_parser(self, subparser_help=False, prefix_chars=None,
  1610. aliases=False):
  1611. # create a parser with a subparsers argument
  1612. if prefix_chars:
  1613. parser = ErrorRaisingArgumentParser(
  1614. prog='PROG', description='main description', prefix_chars=prefix_chars)
  1615. parser.add_argument(
  1616. prefix_chars[0] * 2 + 'foo', action='store_true', help='foo help')
  1617. else:
  1618. parser = ErrorRaisingArgumentParser(
  1619. prog='PROG', description='main description')
  1620. parser.add_argument(
  1621. '--foo', action='store_true', help='foo help')
  1622. parser.add_argument(
  1623. 'bar', type=float, help='bar help')
  1624. # check that only one subparsers argument can be added
  1625. subparsers_kwargs = {'required': False}
  1626. if aliases:
  1627. subparsers_kwargs['metavar'] = 'COMMAND'
  1628. subparsers_kwargs['title'] = 'commands'
  1629. else:
  1630. subparsers_kwargs['help'] = 'command help'
  1631. subparsers = parser.add_subparsers(**subparsers_kwargs)
  1632. self.assertArgumentParserError(parser.add_subparsers)
  1633. # add first sub-parser
  1634. parser1_kwargs = dict(description='1 description')
  1635. if subparser_help:
  1636. parser1_kwargs['help'] = '1 help'
  1637. if aliases:
  1638. parser1_kwargs['aliases'] = ['1alias1', '1alias2']
  1639. parser1 = subparsers.add_parser('1', **parser1_kwargs)
  1640. parser1.add_argument('-w', type=int, help='w help')
  1641. parser1.add_argument('x', choices='abc', help='x help')
  1642. # add second sub-parser
  1643. parser2_kwargs = dict(description='2 description')
  1644. if subparser_help:
  1645. parser2_kwargs['help'] = '2 help'
  1646. parser2 = subparsers.add_parser('2', **parser2_kwargs)
  1647. parser2.add_argument('-y', choices='123', help='y help')
  1648. parser2.add_argument('z', type=complex, nargs='*', help='z help')
  1649. # add third sub-parser
  1650. parser3_kwargs = dict(description='3 description')
  1651. if subparser_help:
  1652. parser3_kwargs['help'] = '3 help'
  1653. parser3 = subparsers.add_parser('3', **parser3_kwargs)
  1654. parser3.add_argument('t', type=int, help='t help')
  1655. parser3.add_argument('u', nargs='...', help='u help')
  1656. # return the main parser
  1657. return parser
  1658. def setUp(self):
  1659. super().setUp()
  1660. self.parser = self._get_parser()
  1661. self.command_help_parser = self._get_parser(subparser_help=True)
  1662. def test_parse_args_failures(self):
  1663. # check some failure cases:
  1664. for args_str in ['', 'a', 'a a', '0.5 a', '0.5 1',
  1665. '0.5 1 -y', '0.5 2 -w']:
  1666. args = args_str.split()
  1667. self.assertArgumentParserError(self.parser.parse_args, args)
  1668. def test_parse_args(self):
  1669. # check some non-failure cases:
  1670. self.assertEqual(
  1671. self.parser.parse_args('0.5 1 b -w 7'.split()),
  1672. NS(foo=False, bar=0.5, w=7, x='b'),
  1673. )
  1674. self.assertEqual(
  1675. self.parser.parse_args('0.25 --foo 2 -y 2 3j -- -1j'.split()),
  1676. NS(foo=True, bar=0.25, y='2', z=[3j, -1j]),
  1677. )
  1678. self.assertEqual(
  1679. self.parser.parse_args('--foo 0.125 1 c'.split()),
  1680. NS(foo=True, bar=0.125, w=None, x='c'),
  1681. )
  1682. self.assertEqual(
  1683. self.parser.parse_args('-1.5 3 11 -- a --foo 7 -- b'.split()),
  1684. NS(foo=False, bar=-1.5, t=11, u=['a', '--foo', '7', '--', 'b']),
  1685. )
  1686. def test_parse_known_args(self):
  1687. self.assertEqual(
  1688. self.parser.parse_known_args('0.5 1 b -w 7'.split()),
  1689. (NS(foo=False, bar=0.5, w=7, x='b'), []),
  1690. )
  1691. self.assertEqual(
  1692. self.parser.parse_known_args('0.5 -p 1 b -w 7'.split()),
  1693. (NS(foo=False, bar=0.5, w=7, x='b'), ['-p']),
  1694. )
  1695. self.assertEqual(
  1696. self.parser.parse_known_args('0.5 1 b -w 7 -p'.split()),
  1697. (NS(foo=False, bar=0.5, w=7, x='b'), ['-p']),
  1698. )
  1699. self.assertEqual(
  1700. self.parser.parse_known_args('0.5 1 b -q -rs -w 7'.split()),
  1701. (NS(foo=False, bar=0.5, w=7, x='b'), ['-q', '-rs']),
  1702. )
  1703. self.assertEqual(
  1704. self.parser.parse_known_args('0.5 -W 1 b -X Y -w 7 Z'.split()),
  1705. (NS(foo=False, bar=0.5, w=7, x='b'), ['-W', '-X', 'Y', 'Z']),
  1706. )
  1707. def test_dest(self):
  1708. parser = ErrorRaisingArgumentParser()
  1709. parser.add_argument('--foo', action='store_true')
  1710. subparsers = parser.add_subparsers(dest='bar')
  1711. parser1 = subparsers.add_parser('1')
  1712. parser1.add_argument('baz')
  1713. self.assertEqual(NS(foo=False, bar='1', baz='2'),
  1714. parser.parse_args('1 2'.split()))
  1715. def _test_required_subparsers(self, parser):
  1716. # Should parse the sub command
  1717. ret = parser.parse_args(['run'])
  1718. self.assertEqual(ret.command, 'run')
  1719. # Error when the command is missing
  1720. self.assertArgumentParserError(parser.parse_args, ())
  1721. def test_required_subparsers_via_attribute(self):
  1722. parser = ErrorRaisingArgumentParser()
  1723. subparsers = parser.add_subparsers(dest='command')
  1724. subparsers.required = True
  1725. subparsers.add_parser('run')
  1726. self._test_required_subparsers(parser)
  1727. def test_required_subparsers_via_kwarg(self):
  1728. parser = ErrorRaisingArgumentParser()
  1729. subparsers = parser.add_subparsers(dest='command', required=True)
  1730. subparsers.add_parser('run')
  1731. self._test_required_subparsers(parser)
  1732. def test_required_subparsers_default(self):
  1733. parser = ErrorRaisingArgumentParser()
  1734. subparsers = parser.add_subparsers(dest='command')
  1735. subparsers.add_parser('run')
  1736. # No error here
  1737. ret = parser.parse_args(())
  1738. self.assertIsNone(ret.command)
  1739. def test_required_subparsers_no_destination_error(self):
  1740. parser = ErrorRaisingArgumentParser()
  1741. subparsers = parser.add_subparsers(required=True)
  1742. subparsers.add_parser('foo')
  1743. subparsers.add_parser('bar')
  1744. with self.assertRaises(ArgumentParserError) as excinfo:
  1745. parser.parse_args(())
  1746. self.assertRegex(
  1747. excinfo.exception.stderr,
  1748. 'error: the following arguments are required: {foo,bar}\n$'
  1749. )
  1750. def test_wrong_argument_subparsers_no_destination_error(self):
  1751. parser = ErrorRaisingArgumentParser()
  1752. subparsers = parser.add_subparsers(required=True)
  1753. subparsers.add_parser('foo')
  1754. subparsers.add_parser('bar')
  1755. with self.assertRaises(ArgumentParserError) as excinfo:
  1756. parser.parse_args(('baz',))
  1757. self.assertRegex(
  1758. excinfo.exception.stderr,
  1759. r"error: argument {foo,bar}: invalid choice: 'baz' \(choose from 'foo', 'bar'\)\n$"
  1760. )
  1761. def test_optional_subparsers(self):
  1762. parser = ErrorRaisingArgumentParser()
  1763. subparsers = parser.add_subparsers(dest='command', required=False)
  1764. subparsers.add_parser('run')
  1765. # No error here
  1766. ret = parser.parse_args(())
  1767. self.assertIsNone(ret.command)
  1768. def test_help(self):
  1769. self.assertEqual(self.parser.format_usage(),
  1770. 'usage: PROG [-h] [--foo] bar {1,2,3} ...\n')
  1771. self.assertEqual(self.parser.format_help(), textwrap.dedent('''\
  1772. usage: PROG [-h] [--foo] bar {1,2,3} ...
  1773. main description
  1774. positional arguments:
  1775. bar bar help
  1776. {1,2,3} command help
  1777. options:
  1778. -h, --help show this help message and exit
  1779. --foo foo help
  1780. '''))
  1781. def test_help_extra_prefix_chars(self):
  1782. # Make sure - is still used for help if it is a non-first prefix char
  1783. parser = self._get_parser(prefix_chars='+:-')
  1784. self.assertEqual(parser.format_usage(),
  1785. 'usage: PROG [-h] [++foo] bar {1,2,3} ...\n')
  1786. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  1787. usage: PROG [-h] [++foo] bar {1,2,3} ...
  1788. main description
  1789. positional arguments:
  1790. bar bar help
  1791. {1,2,3} command help
  1792. options:
  1793. -h, --help show this help message and exit
  1794. ++foo foo help
  1795. '''))
  1796. def test_help_non_breaking_spaces(self):
  1797. parser = ErrorRaisingArgumentParser(
  1798. prog='PROG', description='main description')
  1799. parser.add_argument(
  1800. "--non-breaking", action='store_false',
  1801. help='help message containing non-breaking spaces shall not '
  1802. 'wrap\N{NO-BREAK SPACE}at non-breaking spaces')
  1803. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  1804. usage: PROG [-h] [--non-breaking]
  1805. main description
  1806. options:
  1807. -h, --help show this help message and exit
  1808. --non-breaking help message containing non-breaking spaces shall not
  1809. wrap\N{NO-BREAK SPACE}at non-breaking spaces
  1810. '''))
  1811. def test_help_blank(self):
  1812. # Issue 24444
  1813. parser = ErrorRaisingArgumentParser(
  1814. prog='PROG', description='main description')
  1815. parser.add_argument(
  1816. 'foo',
  1817. help=' ')
  1818. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  1819. usage: PROG [-h] foo
  1820. main description
  1821. positional arguments:
  1822. foo \n
  1823. options:
  1824. -h, --help show this help message and exit
  1825. '''))
  1826. parser = ErrorRaisingArgumentParser(
  1827. prog='PROG', description='main description')
  1828. parser.add_argument(
  1829. 'foo', choices=[],
  1830. help='%(choices)s')
  1831. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  1832. usage: PROG [-h] {}
  1833. main description
  1834. positional arguments:
  1835. {} \n
  1836. options:
  1837. -h, --help show this help message and exit
  1838. '''))
  1839. def test_help_alternate_prefix_chars(self):
  1840. parser = self._get_parser(prefix_chars='+:/')
  1841. self.assertEqual(parser.format_usage(),
  1842. 'usage: PROG [+h] [++foo] bar {1,2,3} ...\n')
  1843. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  1844. usage: PROG [+h] [++foo] bar {1,2,3} ...
  1845. main description
  1846. positional arguments:
  1847. bar bar help
  1848. {1,2,3} command help
  1849. options:
  1850. +h, ++help show this help message and exit
  1851. ++foo foo help
  1852. '''))
  1853. def test_parser_command_help(self):
  1854. self.assertEqual(self.command_help_parser.format_usage(),
  1855. 'usage: PROG [-h] [--foo] bar {1,2,3} ...\n')
  1856. self.assertEqual(self.command_help_parser.format_help(),
  1857. textwrap.dedent('''\
  1858. usage: PROG [-h] [--foo] bar {1,2,3} ...
  1859. main description
  1860. positional arguments:
  1861. bar bar help
  1862. {1,2,3} command help
  1863. 1 1 help
  1864. 2 2 help
  1865. 3 3 help
  1866. options:
  1867. -h, --help show this help message and exit
  1868. --foo foo help
  1869. '''))
  1870. def test_subparser_title_help(self):
  1871. parser = ErrorRaisingArgumentParser(prog='PROG',
  1872. description='main description')
  1873. parser.add_argument('--foo', action='store_true', help='foo help')
  1874. parser.add_argument('bar', help='bar help')
  1875. subparsers = parser.add_subparsers(title='subcommands',
  1876. description='command help',
  1877. help='additional text')
  1878. parser1 = subparsers.add_parser('1')
  1879. parser2 = subparsers.add_parser('2')
  1880. self.assertEqual(parser.format_usage(),
  1881. 'usage: PROG [-h] [--foo] bar {1,2} ...\n')
  1882. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  1883. usage: PROG [-h] [--foo] bar {1,2} ...
  1884. main description
  1885. positional arguments:
  1886. bar bar help
  1887. options:
  1888. -h, --help show this help message and exit
  1889. --foo foo help
  1890. subcommands:
  1891. command help
  1892. {1,2} additional text
  1893. '''))
  1894. def _test_subparser_help(self, args_str, expected_help):
  1895. with self.assertRaises(ArgumentParserError) as cm:
  1896. self.parser.parse_args(args_str.split())
  1897. self.assertEqual(expected_help, cm.exception.stdout)
  1898. def test_subparser1_help(self):
  1899. self._test_subparser_help('5.0 1 -h', textwrap.dedent('''\
  1900. usage: PROG bar 1 [-h] [-w W] {a,b,c}
  1901. 1 description
  1902. positional arguments:
  1903. {a,b,c} x help
  1904. options:
  1905. -h, --help show this help message and exit
  1906. -w W w help
  1907. '''))
  1908. def test_subparser2_help(self):
  1909. self._test_subparser_help('5.0 2 -h', textwrap.dedent('''\
  1910. usage: PROG bar 2 [-h] [-y {1,2,3}] [z ...]
  1911. 2 description
  1912. positional arguments:
  1913. z z help
  1914. options:
  1915. -h, --help show this help message and exit
  1916. -y {1,2,3} y help
  1917. '''))
  1918. def test_alias_invocation(self):
  1919. parser = self._get_parser(aliases=True)
  1920. self.assertEqual(
  1921. parser.parse_known_args('0.5 1alias1 b'.split()),
  1922. (NS(foo=False, bar=0.5, w=None, x='b'), []),
  1923. )
  1924. self.assertEqual(
  1925. parser.parse_known_args('0.5 1alias2 b'.split()),
  1926. (NS(foo=False, bar=0.5, w=None, x='b'), []),
  1927. )
  1928. def test_error_alias_invocation(self):
  1929. parser = self._get_parser(aliases=True)
  1930. self.assertArgumentParserError(parser.parse_args,
  1931. '0.5 1alias3 b'.split())
  1932. def test_alias_help(self):
  1933. parser = self._get_parser(aliases=True, subparser_help=True)
  1934. self.maxDiff = None
  1935. self.assertEqual(parser.format_help(), textwrap.dedent("""\
  1936. usage: PROG [-h] [--foo] bar COMMAND ...
  1937. main description
  1938. positional arguments:
  1939. bar bar help
  1940. options:
  1941. -h, --help show this help message and exit
  1942. --foo foo help
  1943. commands:
  1944. COMMAND
  1945. 1 (1alias1, 1alias2)
  1946. 1 help
  1947. 2 2 help
  1948. 3 3 help
  1949. """))
  1950. # ============
  1951. # Groups tests
  1952. # ============
  1953. class TestPositionalsGroups(TestCase):
  1954. """Tests that order of group positionals matches construction order"""
  1955. def test_nongroup_first(self):
  1956. parser = ErrorRaisingArgumentParser()
  1957. parser.add_argument('foo')
  1958. group = parser.add_argument_group('g')
  1959. group.add_argument('bar')
  1960. parser.add_argument('baz')
  1961. expected = NS(foo='1', bar='2', baz='3')
  1962. result = parser.parse_args('1 2 3'.split())
  1963. self.assertEqual(expected, result)
  1964. def test_group_first(self):
  1965. parser = ErrorRaisingArgumentParser()
  1966. group = parser.add_argument_group('xxx')
  1967. group.add_argument('foo')
  1968. parser.add_argument('bar')
  1969. parser.add_argument('baz')
  1970. expected = NS(foo='1', bar='2', baz='3')
  1971. result = parser.parse_args('1 2 3'.split())
  1972. self.assertEqual(expected, result)
  1973. def test_interleaved_groups(self):
  1974. parser = ErrorRaisingArgumentParser()
  1975. group = parser.add_argument_group('xxx')
  1976. parser.add_argument('foo')
  1977. group.add_argument('bar')
  1978. parser.add_argument('baz')
  1979. group = parser.add_argument_group('yyy')
  1980. group.add_argument('frell')
  1981. expected = NS(foo='1', bar='2', baz='3', frell='4')
  1982. result = parser.parse_args('1 2 3 4'.split())
  1983. self.assertEqual(expected, result)
  1984. # ===================
  1985. # Parent parser tests
  1986. # ===================
  1987. class TestParentParsers(TestCase):
  1988. """Tests that parsers can be created with parent parsers"""
  1989. def assertArgumentParserError(self, *args, **kwargs):
  1990. self.assertRaises(ArgumentParserError, *args, **kwargs)
  1991. def setUp(self):
  1992. super().setUp()
  1993. self.wxyz_parent = ErrorRaisingArgumentParser(add_help=False)
  1994. self.wxyz_parent.add_argument('--w')
  1995. x_group = self.wxyz_parent.add_argument_group('x')
  1996. x_group.add_argument('-y')
  1997. self.wxyz_parent.add_argument('z')
  1998. self.abcd_parent = ErrorRaisingArgumentParser(add_help=False)
  1999. self.abcd_parent.add_argument('a')
  2000. self.abcd_parent.add_argument('-b')
  2001. c_group = self.abcd_parent.add_argument_group('c')
  2002. c_group.add_argument('--d')
  2003. self.w_parent = ErrorRaisingArgumentParser(add_help=False)
  2004. self.w_parent.add_argument('--w')
  2005. self.z_parent = ErrorRaisingArgumentParser(add_help=False)
  2006. self.z_parent.add_argument('z')
  2007. # parents with mutually exclusive groups
  2008. self.ab_mutex_parent = ErrorRaisingArgumentParser(add_help=False)
  2009. group = self.ab_mutex_parent.add_mutually_exclusive_group()
  2010. group.add_argument('-a', action='store_true')
  2011. group.add_argument('-b', action='store_true')
  2012. self.main_program = os.path.basename(sys.argv[0])
  2013. def test_single_parent(self):
  2014. parser = ErrorRaisingArgumentParser(parents=[self.wxyz_parent])
  2015. self.assertEqual(parser.parse_args('-y 1 2 --w 3'.split()),
  2016. NS(w='3', y='1', z='2'))
  2017. def test_single_parent_mutex(self):
  2018. self._test_mutex_ab(self.ab_mutex_parent.parse_args)
  2019. parser = ErrorRaisingArgumentParser(parents=[self.ab_mutex_parent])
  2020. self._test_mutex_ab(parser.parse_args)
  2021. def test_single_granparent_mutex(self):
  2022. parents = [self.ab_mutex_parent]
  2023. parser = ErrorRaisingArgumentParser(add_help=False, parents=parents)
  2024. parser = ErrorRaisingArgumentParser(parents=[parser])
  2025. self._test_mutex_ab(parser.parse_args)
  2026. def _test_mutex_ab(self, parse_args):
  2027. self.assertEqual(parse_args([]), NS(a=False, b=False))
  2028. self.assertEqual(parse_args(['-a']), NS(a=True, b=False))
  2029. self.assertEqual(parse_args(['-b']), NS(a=False, b=True))
  2030. self.assertArgumentParserError(parse_args, ['-a', '-b'])
  2031. self.assertArgumentParserError(parse_args, ['-b', '-a'])
  2032. self.assertArgumentParserError(parse_args, ['-c'])
  2033. self.assertArgumentParserError(parse_args, ['-a', '-c'])
  2034. self.assertArgumentParserError(parse_args, ['-b', '-c'])
  2035. def test_multiple_parents(self):
  2036. parents = [self.abcd_parent, self.wxyz_parent]
  2037. parser = ErrorRaisingArgumentParser(parents=parents)
  2038. self.assertEqual(parser.parse_args('--d 1 --w 2 3 4'.split()),
  2039. NS(a='3', b=None, d='1', w='2', y=None, z='4'))
  2040. def test_multiple_parents_mutex(self):
  2041. parents = [self.ab_mutex_parent, self.wxyz_parent]
  2042. parser = ErrorRaisingArgumentParser(parents=parents)
  2043. self.assertEqual(parser.parse_args('-a --w 2 3'.split()),
  2044. NS(a=True, b=False, w='2', y=None, z='3'))
  2045. self.assertArgumentParserError(
  2046. parser.parse_args, '-a --w 2 3 -b'.split())
  2047. self.assertArgumentParserError(
  2048. parser.parse_args, '-a -b --w 2 3'.split())
  2049. def test_conflicting_parents(self):
  2050. self.assertRaises(
  2051. argparse.ArgumentError,
  2052. argparse.ArgumentParser,
  2053. parents=[self.w_parent, self.wxyz_parent])
  2054. def test_conflicting_parents_mutex(self):
  2055. self.assertRaises(
  2056. argparse.ArgumentError,
  2057. argparse.ArgumentParser,
  2058. parents=[self.abcd_parent, self.ab_mutex_parent])
  2059. def test_same_argument_name_parents(self):
  2060. parents = [self.wxyz_parent, self.z_parent]
  2061. parser = ErrorRaisingArgumentParser(parents=parents)
  2062. self.assertEqual(parser.parse_args('1 2'.split()),
  2063. NS(w=None, y=None, z='2'))
  2064. def test_subparser_parents(self):
  2065. parser = ErrorRaisingArgumentParser()
  2066. subparsers = parser.add_subparsers()
  2067. abcde_parser = subparsers.add_parser('bar', parents=[self.abcd_parent])
  2068. abcde_parser.add_argument('e')
  2069. self.assertEqual(parser.parse_args('bar -b 1 --d 2 3 4'.split()),
  2070. NS(a='3', b='1', d='2', e='4'))
  2071. def test_subparser_parents_mutex(self):
  2072. parser = ErrorRaisingArgumentParser()
  2073. subparsers = parser.add_subparsers()
  2074. parents = [self.ab_mutex_parent]
  2075. abc_parser = subparsers.add_parser('foo', parents=parents)
  2076. c_group = abc_parser.add_argument_group('c_group')
  2077. c_group.add_argument('c')
  2078. parents = [self.wxyz_parent, self.ab_mutex_parent]
  2079. wxyzabe_parser = subparsers.add_parser('bar', parents=parents)
  2080. wxyzabe_parser.add_argument('e')
  2081. self.assertEqual(parser.parse_args('foo -a 4'.split()),
  2082. NS(a=True, b=False, c='4'))
  2083. self.assertEqual(parser.parse_args('bar -b --w 2 3 4'.split()),
  2084. NS(a=False, b=True, w='2', y=None, z='3', e='4'))
  2085. self.assertArgumentParserError(
  2086. parser.parse_args, 'foo -a -b 4'.split())
  2087. self.assertArgumentParserError(
  2088. parser.parse_args, 'bar -b -a 4'.split())
  2089. def test_parent_help(self):
  2090. parents = [self.abcd_parent, self.wxyz_parent]
  2091. parser = ErrorRaisingArgumentParser(parents=parents)
  2092. parser_help = parser.format_help()
  2093. progname = self.main_program
  2094. self.assertEqual(parser_help, textwrap.dedent('''\
  2095. usage: {}{}[-h] [-b B] [--d D] [--w W] [-y Y] a z
  2096. positional arguments:
  2097. a
  2098. z
  2099. options:
  2100. -h, --help show this help message and exit
  2101. -b B
  2102. --w W
  2103. c:
  2104. --d D
  2105. x:
  2106. -y Y
  2107. '''.format(progname, ' ' if progname else '' )))
  2108. def test_groups_parents(self):
  2109. parent = ErrorRaisingArgumentParser(add_help=False)
  2110. g = parent.add_argument_group(title='g', description='gd')
  2111. g.add_argument('-w')
  2112. g.add_argument('-x')
  2113. m = parent.add_mutually_exclusive_group()
  2114. m.add_argument('-y')
  2115. m.add_argument('-z')
  2116. parser = ErrorRaisingArgumentParser(parents=[parent])
  2117. self.assertRaises(ArgumentParserError, parser.parse_args,
  2118. ['-y', 'Y', '-z', 'Z'])
  2119. parser_help = parser.format_help()
  2120. progname = self.main_program
  2121. self.assertEqual(parser_help, textwrap.dedent('''\
  2122. usage: {}{}[-h] [-w W] [-x X] [-y Y | -z Z]
  2123. options:
  2124. -h, --help show this help message and exit
  2125. -y Y
  2126. -z Z
  2127. g:
  2128. gd
  2129. -w W
  2130. -x X
  2131. '''.format(progname, ' ' if progname else '' )))
  2132. # ==============================
  2133. # Mutually exclusive group tests
  2134. # ==============================
  2135. class TestMutuallyExclusiveGroupErrors(TestCase):
  2136. def test_invalid_add_argument_group(self):
  2137. parser = ErrorRaisingArgumentParser()
  2138. raises = self.assertRaises
  2139. raises(TypeError, parser.add_mutually_exclusive_group, title='foo')
  2140. def test_invalid_add_argument(self):
  2141. parser = ErrorRaisingArgumentParser()
  2142. group = parser.add_mutually_exclusive_group()
  2143. add_argument = group.add_argument
  2144. raises = self.assertRaises
  2145. raises(ValueError, add_argument, '--foo', required=True)
  2146. raises(ValueError, add_argument, 'bar')
  2147. raises(ValueError, add_argument, 'bar', nargs='+')
  2148. raises(ValueError, add_argument, 'bar', nargs=1)
  2149. raises(ValueError, add_argument, 'bar', nargs=argparse.PARSER)
  2150. def test_help(self):
  2151. parser = ErrorRaisingArgumentParser(prog='PROG')
  2152. group1 = parser.add_mutually_exclusive_group()
  2153. group1.add_argument('--foo', action='store_true')
  2154. group1.add_argument('--bar', action='store_false')
  2155. group2 = parser.add_mutually_exclusive_group()
  2156. group2.add_argument('--soup', action='store_true')
  2157. group2.add_argument('--nuts', action='store_false')
  2158. expected = '''\
  2159. usage: PROG [-h] [--foo | --bar] [--soup | --nuts]
  2160. options:
  2161. -h, --help show this help message and exit
  2162. --foo
  2163. --bar
  2164. --soup
  2165. --nuts
  2166. '''
  2167. self.assertEqual(parser.format_help(), textwrap.dedent(expected))
  2168. def test_empty_group(self):
  2169. # See issue 26952
  2170. parser = argparse.ArgumentParser()
  2171. group = parser.add_mutually_exclusive_group()
  2172. with self.assertRaises(ValueError):
  2173. parser.parse_args(['-h'])
  2174. class MEMixin(object):
  2175. def test_failures_when_not_required(self):
  2176. parse_args = self.get_parser(required=False).parse_args
  2177. error = ArgumentParserError
  2178. for args_string in self.failures:
  2179. self.assertRaises(error, parse_args, args_string.split())
  2180. def test_failures_when_required(self):
  2181. parse_args = self.get_parser(required=True).parse_args
  2182. error = ArgumentParserError
  2183. for args_string in self.failures + ['']:
  2184. self.assertRaises(error, parse_args, args_string.split())
  2185. def test_successes_when_not_required(self):
  2186. parse_args = self.get_parser(required=False).parse_args
  2187. successes = self.successes + self.successes_when_not_required
  2188. for args_string, expected_ns in successes:
  2189. actual_ns = parse_args(args_string.split())
  2190. self.assertEqual(actual_ns, expected_ns)
  2191. def test_successes_when_required(self):
  2192. parse_args = self.get_parser(required=True).parse_args
  2193. for args_string, expected_ns in self.successes:
  2194. actual_ns = parse_args(args_string.split())
  2195. self.assertEqual(actual_ns, expected_ns)
  2196. def test_usage_when_not_required(self):
  2197. format_usage = self.get_parser(required=False).format_usage
  2198. expected_usage = self.usage_when_not_required
  2199. self.assertEqual(format_usage(), textwrap.dedent(expected_usage))
  2200. def test_usage_when_required(self):
  2201. format_usage = self.get_parser(required=True).format_usage
  2202. expected_usage = self.usage_when_required
  2203. self.assertEqual(format_usage(), textwrap.dedent(expected_usage))
  2204. def test_help_when_not_required(self):
  2205. format_help = self.get_parser(required=False).format_help
  2206. help = self.usage_when_not_required + self.help
  2207. self.assertEqual(format_help(), textwrap.dedent(help))
  2208. def test_help_when_required(self):
  2209. format_help = self.get_parser(required=True).format_help
  2210. help = self.usage_when_required + self.help
  2211. self.assertEqual(format_help(), textwrap.dedent(help))
  2212. class TestMutuallyExclusiveSimple(MEMixin, TestCase):
  2213. def get_parser(self, required=None):
  2214. parser = ErrorRaisingArgumentParser(prog='PROG')
  2215. group = parser.add_mutually_exclusive_group(required=required)
  2216. group.add_argument('--bar', help='bar help')
  2217. group.add_argument('--baz', nargs='?', const='Z', help='baz help')
  2218. return parser
  2219. failures = ['--bar X --baz Y', '--bar X --baz']
  2220. successes = [
  2221. ('--bar X', NS(bar='X', baz=None)),
  2222. ('--bar X --bar Z', NS(bar='Z', baz=None)),
  2223. ('--baz Y', NS(bar=None, baz='Y')),
  2224. ('--baz', NS(bar=None, baz='Z')),
  2225. ]
  2226. successes_when_not_required = [
  2227. ('', NS(bar=None, baz=None)),
  2228. ]
  2229. usage_when_not_required = '''\
  2230. usage: PROG [-h] [--bar BAR | --baz [BAZ]]
  2231. '''
  2232. usage_when_required = '''\
  2233. usage: PROG [-h] (--bar BAR | --baz [BAZ])
  2234. '''
  2235. help = '''\
  2236. options:
  2237. -h, --help show this help message and exit
  2238. --bar BAR bar help
  2239. --baz [BAZ] baz help
  2240. '''
  2241. class TestMutuallyExclusiveLong(MEMixin, TestCase):
  2242. def get_parser(self, required=None):
  2243. parser = ErrorRaisingArgumentParser(prog='PROG')
  2244. parser.add_argument('--abcde', help='abcde help')
  2245. parser.add_argument('--fghij', help='fghij help')
  2246. group = parser.add_mutually_exclusive_group(required=required)
  2247. group.add_argument('--klmno', help='klmno help')
  2248. group.add_argument('--pqrst', help='pqrst help')
  2249. return parser
  2250. failures = ['--klmno X --pqrst Y']
  2251. successes = [
  2252. ('--klmno X', NS(abcde=None, fghij=None, klmno='X', pqrst=None)),
  2253. ('--abcde Y --klmno X',
  2254. NS(abcde='Y', fghij=None, klmno='X', pqrst=None)),
  2255. ('--pqrst X', NS(abcde=None, fghij=None, klmno=None, pqrst='X')),
  2256. ('--pqrst X --fghij Y',
  2257. NS(abcde=None, fghij='Y', klmno=None, pqrst='X')),
  2258. ]
  2259. successes_when_not_required = [
  2260. ('', NS(abcde=None, fghij=None, klmno=None, pqrst=None)),
  2261. ]
  2262. usage_when_not_required = '''\
  2263. usage: PROG [-h] [--abcde ABCDE] [--fghij FGHIJ]
  2264. [--klmno KLMNO | --pqrst PQRST]
  2265. '''
  2266. usage_when_required = '''\
  2267. usage: PROG [-h] [--abcde ABCDE] [--fghij FGHIJ]
  2268. (--klmno KLMNO | --pqrst PQRST)
  2269. '''
  2270. help = '''\
  2271. options:
  2272. -h, --help show this help message and exit
  2273. --abcde ABCDE abcde help
  2274. --fghij FGHIJ fghij help
  2275. --klmno KLMNO klmno help
  2276. --pqrst PQRST pqrst help
  2277. '''
  2278. class TestMutuallyExclusiveFirstSuppressed(MEMixin, TestCase):
  2279. def get_parser(self, required):
  2280. parser = ErrorRaisingArgumentParser(prog='PROG')
  2281. group = parser.add_mutually_exclusive_group(required=required)
  2282. group.add_argument('-x', help=argparse.SUPPRESS)
  2283. group.add_argument('-y', action='store_false', help='y help')
  2284. return parser
  2285. failures = ['-x X -y']
  2286. successes = [
  2287. ('-x X', NS(x='X', y=True)),
  2288. ('-x X -x Y', NS(x='Y', y=True)),
  2289. ('-y', NS(x=None, y=False)),
  2290. ]
  2291. successes_when_not_required = [
  2292. ('', NS(x=None, y=True)),
  2293. ]
  2294. usage_when_not_required = '''\
  2295. usage: PROG [-h] [-y]
  2296. '''
  2297. usage_when_required = '''\
  2298. usage: PROG [-h] -y
  2299. '''
  2300. help = '''\
  2301. options:
  2302. -h, --help show this help message and exit
  2303. -y y help
  2304. '''
  2305. class TestMutuallyExclusiveManySuppressed(MEMixin, TestCase):
  2306. def get_parser(self, required):
  2307. parser = ErrorRaisingArgumentParser(prog='PROG')
  2308. group = parser.add_mutually_exclusive_group(required=required)
  2309. add = group.add_argument
  2310. add('--spam', action='store_true', help=argparse.SUPPRESS)
  2311. add('--badger', action='store_false', help=argparse.SUPPRESS)
  2312. add('--bladder', help=argparse.SUPPRESS)
  2313. return parser
  2314. failures = [
  2315. '--spam --badger',
  2316. '--badger --bladder B',
  2317. '--bladder B --spam',
  2318. ]
  2319. successes = [
  2320. ('--spam', NS(spam=True, badger=True, bladder=None)),
  2321. ('--badger', NS(spam=False, badger=False, bladder=None)),
  2322. ('--bladder B', NS(spam=False, badger=True, bladder='B')),
  2323. ('--spam --spam', NS(spam=True, badger=True, bladder=None)),
  2324. ]
  2325. successes_when_not_required = [
  2326. ('', NS(spam=False, badger=True, bladder=None)),
  2327. ]
  2328. usage_when_required = usage_when_not_required = '''\
  2329. usage: PROG [-h]
  2330. '''
  2331. help = '''\
  2332. options:
  2333. -h, --help show this help message and exit
  2334. '''
  2335. class TestMutuallyExclusiveOptionalAndPositional(MEMixin, TestCase):
  2336. def get_parser(self, required):
  2337. parser = ErrorRaisingArgumentParser(prog='PROG')
  2338. group = parser.add_mutually_exclusive_group(required=required)
  2339. group.add_argument('--foo', action='store_true', help='FOO')
  2340. group.add_argument('--spam', help='SPAM')
  2341. group.add_argument('badger', nargs='*', default='X', help='BADGER')
  2342. return parser
  2343. failures = [
  2344. '--foo --spam S',
  2345. '--spam S X',
  2346. 'X --foo',
  2347. 'X Y Z --spam S',
  2348. '--foo X Y',
  2349. ]
  2350. successes = [
  2351. ('--foo', NS(foo=True, spam=None, badger='X')),
  2352. ('--spam S', NS(foo=False, spam='S', badger='X')),
  2353. ('X', NS(foo=False, spam=None, badger=['X'])),
  2354. ('X Y Z', NS(foo=False, spam=None, badger=['X', 'Y', 'Z'])),
  2355. ]
  2356. successes_when_not_required = [
  2357. ('', NS(foo=False, spam=None, badger='X')),
  2358. ]
  2359. usage_when_not_required = '''\
  2360. usage: PROG [-h] [--foo | --spam SPAM | badger ...]
  2361. '''
  2362. usage_when_required = '''\
  2363. usage: PROG [-h] (--foo | --spam SPAM | badger ...)
  2364. '''
  2365. help = '''\
  2366. positional arguments:
  2367. badger BADGER
  2368. options:
  2369. -h, --help show this help message and exit
  2370. --foo FOO
  2371. --spam SPAM SPAM
  2372. '''
  2373. class TestMutuallyExclusiveOptionalsMixed(MEMixin, TestCase):
  2374. def get_parser(self, required):
  2375. parser = ErrorRaisingArgumentParser(prog='PROG')
  2376. parser.add_argument('-x', action='store_true', help='x help')
  2377. group = parser.add_mutually_exclusive_group(required=required)
  2378. group.add_argument('-a', action='store_true', help='a help')
  2379. group.add_argument('-b', action='store_true', help='b help')
  2380. parser.add_argument('-y', action='store_true', help='y help')
  2381. group.add_argument('-c', action='store_true', help='c help')
  2382. return parser
  2383. failures = ['-a -b', '-b -c', '-a -c', '-a -b -c']
  2384. successes = [
  2385. ('-a', NS(a=True, b=False, c=False, x=False, y=False)),
  2386. ('-b', NS(a=False, b=True, c=False, x=False, y=False)),
  2387. ('-c', NS(a=False, b=False, c=True, x=False, y=False)),
  2388. ('-a -x', NS(a=True, b=False, c=False, x=True, y=False)),
  2389. ('-y -b', NS(a=False, b=True, c=False, x=False, y=True)),
  2390. ('-x -y -c', NS(a=False, b=False, c=True, x=True, y=True)),
  2391. ]
  2392. successes_when_not_required = [
  2393. ('', NS(a=False, b=False, c=False, x=False, y=False)),
  2394. ('-x', NS(a=False, b=False, c=False, x=True, y=False)),
  2395. ('-y', NS(a=False, b=False, c=False, x=False, y=True)),
  2396. ]
  2397. usage_when_required = usage_when_not_required = '''\
  2398. usage: PROG [-h] [-x] [-a] [-b] [-y] [-c]
  2399. '''
  2400. help = '''\
  2401. options:
  2402. -h, --help show this help message and exit
  2403. -x x help
  2404. -a a help
  2405. -b b help
  2406. -y y help
  2407. -c c help
  2408. '''
  2409. class TestMutuallyExclusiveInGroup(MEMixin, TestCase):
  2410. def get_parser(self, required=None):
  2411. parser = ErrorRaisingArgumentParser(prog='PROG')
  2412. titled_group = parser.add_argument_group(
  2413. title='Titled group', description='Group description')
  2414. mutex_group = \
  2415. titled_group.add_mutually_exclusive_group(required=required)
  2416. mutex_group.add_argument('--bar', help='bar help')
  2417. mutex_group.add_argument('--baz', help='baz help')
  2418. return parser
  2419. failures = ['--bar X --baz Y', '--baz X --bar Y']
  2420. successes = [
  2421. ('--bar X', NS(bar='X', baz=None)),
  2422. ('--baz Y', NS(bar=None, baz='Y')),
  2423. ]
  2424. successes_when_not_required = [
  2425. ('', NS(bar=None, baz=None)),
  2426. ]
  2427. usage_when_not_required = '''\
  2428. usage: PROG [-h] [--bar BAR | --baz BAZ]
  2429. '''
  2430. usage_when_required = '''\
  2431. usage: PROG [-h] (--bar BAR | --baz BAZ)
  2432. '''
  2433. help = '''\
  2434. options:
  2435. -h, --help show this help message and exit
  2436. Titled group:
  2437. Group description
  2438. --bar BAR bar help
  2439. --baz BAZ baz help
  2440. '''
  2441. class TestMutuallyExclusiveOptionalsAndPositionalsMixed(MEMixin, TestCase):
  2442. def get_parser(self, required):
  2443. parser = ErrorRaisingArgumentParser(prog='PROG')
  2444. parser.add_argument('x', help='x help')
  2445. parser.add_argument('-y', action='store_true', help='y help')
  2446. group = parser.add_mutually_exclusive_group(required=required)
  2447. group.add_argument('a', nargs='?', help='a help')
  2448. group.add_argument('-b', action='store_true', help='b help')
  2449. group.add_argument('-c', action='store_true', help='c help')
  2450. return parser
  2451. failures = ['X A -b', '-b -c', '-c X A']
  2452. successes = [
  2453. ('X A', NS(a='A', b=False, c=False, x='X', y=False)),
  2454. ('X -b', NS(a=None, b=True, c=False, x='X', y=False)),
  2455. ('X -c', NS(a=None, b=False, c=True, x='X', y=False)),
  2456. ('X A -y', NS(a='A', b=False, c=False, x='X', y=True)),
  2457. ('X -y -b', NS(a=None, b=True, c=False, x='X', y=True)),
  2458. ]
  2459. successes_when_not_required = [
  2460. ('X', NS(a=None, b=False, c=False, x='X', y=False)),
  2461. ('X -y', NS(a=None, b=False, c=False, x='X', y=True)),
  2462. ]
  2463. usage_when_required = usage_when_not_required = '''\
  2464. usage: PROG [-h] [-y] [-b] [-c] x [a]
  2465. '''
  2466. help = '''\
  2467. positional arguments:
  2468. x x help
  2469. a a help
  2470. options:
  2471. -h, --help show this help message and exit
  2472. -y y help
  2473. -b b help
  2474. -c c help
  2475. '''
  2476. class TestMutuallyExclusiveNested(MEMixin, TestCase):
  2477. # Nesting mutually exclusive groups is an undocumented feature
  2478. # that came about by accident through inheritance and has been
  2479. # the source of many bugs. It is deprecated and this test should
  2480. # eventually be removed along with it.
  2481. def get_parser(self, required):
  2482. parser = ErrorRaisingArgumentParser(prog='PROG')
  2483. group = parser.add_mutually_exclusive_group(required=required)
  2484. group.add_argument('-a')
  2485. group.add_argument('-b')
  2486. with warnings.catch_warnings():
  2487. warnings.simplefilter('ignore', DeprecationWarning)
  2488. group2 = group.add_mutually_exclusive_group(required=required)
  2489. group2.add_argument('-c')
  2490. group2.add_argument('-d')
  2491. with warnings.catch_warnings():
  2492. warnings.simplefilter('ignore', DeprecationWarning)
  2493. group3 = group2.add_mutually_exclusive_group(required=required)
  2494. group3.add_argument('-e')
  2495. group3.add_argument('-f')
  2496. return parser
  2497. usage_when_not_required = '''\
  2498. usage: PROG [-h] [-a A | -b B | [-c C | -d D | [-e E | -f F]]]
  2499. '''
  2500. usage_when_required = '''\
  2501. usage: PROG [-h] (-a A | -b B | (-c C | -d D | (-e E | -f F)))
  2502. '''
  2503. help = '''\
  2504. options:
  2505. -h, --help show this help message and exit
  2506. -a A
  2507. -b B
  2508. -c C
  2509. -d D
  2510. -e E
  2511. -f F
  2512. '''
  2513. # We are only interested in testing the behavior of format_usage().
  2514. test_failures_when_not_required = None
  2515. test_failures_when_required = None
  2516. test_successes_when_not_required = None
  2517. test_successes_when_required = None
  2518. # =================================================
  2519. # Mutually exclusive group in parent parser tests
  2520. # =================================================
  2521. class MEPBase(object):
  2522. def get_parser(self, required=None):
  2523. parent = super(MEPBase, self).get_parser(required=required)
  2524. parser = ErrorRaisingArgumentParser(
  2525. prog=parent.prog, add_help=False, parents=[parent])
  2526. return parser
  2527. class TestMutuallyExclusiveGroupErrorsParent(
  2528. MEPBase, TestMutuallyExclusiveGroupErrors):
  2529. pass
  2530. class TestMutuallyExclusiveSimpleParent(
  2531. MEPBase, TestMutuallyExclusiveSimple):
  2532. pass
  2533. class TestMutuallyExclusiveLongParent(
  2534. MEPBase, TestMutuallyExclusiveLong):
  2535. pass
  2536. class TestMutuallyExclusiveFirstSuppressedParent(
  2537. MEPBase, TestMutuallyExclusiveFirstSuppressed):
  2538. pass
  2539. class TestMutuallyExclusiveManySuppressedParent(
  2540. MEPBase, TestMutuallyExclusiveManySuppressed):
  2541. pass
  2542. class TestMutuallyExclusiveOptionalAndPositionalParent(
  2543. MEPBase, TestMutuallyExclusiveOptionalAndPositional):
  2544. pass
  2545. class TestMutuallyExclusiveOptionalsMixedParent(
  2546. MEPBase, TestMutuallyExclusiveOptionalsMixed):
  2547. pass
  2548. class TestMutuallyExclusiveOptionalsAndPositionalsMixedParent(
  2549. MEPBase, TestMutuallyExclusiveOptionalsAndPositionalsMixed):
  2550. pass
  2551. # =================
  2552. # Set default tests
  2553. # =================
  2554. class TestSetDefaults(TestCase):
  2555. def test_set_defaults_no_args(self):
  2556. parser = ErrorRaisingArgumentParser()
  2557. parser.set_defaults(x='foo')
  2558. parser.set_defaults(y='bar', z=1)
  2559. self.assertEqual(NS(x='foo', y='bar', z=1),
  2560. parser.parse_args([]))
  2561. self.assertEqual(NS(x='foo', y='bar', z=1),
  2562. parser.parse_args([], NS()))
  2563. self.assertEqual(NS(x='baz', y='bar', z=1),
  2564. parser.parse_args([], NS(x='baz')))
  2565. self.assertEqual(NS(x='baz', y='bar', z=2),
  2566. parser.parse_args([], NS(x='baz', z=2)))
  2567. def test_set_defaults_with_args(self):
  2568. parser = ErrorRaisingArgumentParser()
  2569. parser.set_defaults(x='foo', y='bar')
  2570. parser.add_argument('-x', default='xfoox')
  2571. self.assertEqual(NS(x='xfoox', y='bar'),
  2572. parser.parse_args([]))
  2573. self.assertEqual(NS(x='xfoox', y='bar'),
  2574. parser.parse_args([], NS()))
  2575. self.assertEqual(NS(x='baz', y='bar'),
  2576. parser.parse_args([], NS(x='baz')))
  2577. self.assertEqual(NS(x='1', y='bar'),
  2578. parser.parse_args('-x 1'.split()))
  2579. self.assertEqual(NS(x='1', y='bar'),
  2580. parser.parse_args('-x 1'.split(), NS()))
  2581. self.assertEqual(NS(x='1', y='bar'),
  2582. parser.parse_args('-x 1'.split(), NS(x='baz')))
  2583. def test_set_defaults_subparsers(self):
  2584. parser = ErrorRaisingArgumentParser()
  2585. parser.set_defaults(x='foo')
  2586. subparsers = parser.add_subparsers()
  2587. parser_a = subparsers.add_parser('a')
  2588. parser_a.set_defaults(y='bar')
  2589. self.assertEqual(NS(x='foo', y='bar'),
  2590. parser.parse_args('a'.split()))
  2591. def test_set_defaults_parents(self):
  2592. parent = ErrorRaisingArgumentParser(add_help=False)
  2593. parent.set_defaults(x='foo')
  2594. parser = ErrorRaisingArgumentParser(parents=[parent])
  2595. self.assertEqual(NS(x='foo'), parser.parse_args([]))
  2596. def test_set_defaults_on_parent_and_subparser(self):
  2597. parser = argparse.ArgumentParser()
  2598. xparser = parser.add_subparsers().add_parser('X')
  2599. parser.set_defaults(foo=1)
  2600. xparser.set_defaults(foo=2)
  2601. self.assertEqual(NS(foo=2), parser.parse_args(['X']))
  2602. def test_set_defaults_same_as_add_argument(self):
  2603. parser = ErrorRaisingArgumentParser()
  2604. parser.set_defaults(w='W', x='X', y='Y', z='Z')
  2605. parser.add_argument('-w')
  2606. parser.add_argument('-x', default='XX')
  2607. parser.add_argument('y', nargs='?')
  2608. parser.add_argument('z', nargs='?', default='ZZ')
  2609. # defaults set previously
  2610. self.assertEqual(NS(w='W', x='XX', y='Y', z='ZZ'),
  2611. parser.parse_args([]))
  2612. # reset defaults
  2613. parser.set_defaults(w='WW', x='X', y='YY', z='Z')
  2614. self.assertEqual(NS(w='WW', x='X', y='YY', z='Z'),
  2615. parser.parse_args([]))
  2616. def test_set_defaults_same_as_add_argument_group(self):
  2617. parser = ErrorRaisingArgumentParser()
  2618. parser.set_defaults(w='W', x='X', y='Y', z='Z')
  2619. group = parser.add_argument_group('foo')
  2620. group.add_argument('-w')
  2621. group.add_argument('-x', default='XX')
  2622. group.add_argument('y', nargs='?')
  2623. group.add_argument('z', nargs='?', default='ZZ')
  2624. # defaults set previously
  2625. self.assertEqual(NS(w='W', x='XX', y='Y', z='ZZ'),
  2626. parser.parse_args([]))
  2627. # reset defaults
  2628. parser.set_defaults(w='WW', x='X', y='YY', z='Z')
  2629. self.assertEqual(NS(w='WW', x='X', y='YY', z='Z'),
  2630. parser.parse_args([]))
  2631. # =================
  2632. # Get default tests
  2633. # =================
  2634. class TestGetDefault(TestCase):
  2635. def test_get_default(self):
  2636. parser = ErrorRaisingArgumentParser()
  2637. self.assertIsNone(parser.get_default("foo"))
  2638. self.assertIsNone(parser.get_default("bar"))
  2639. parser.add_argument("--foo")
  2640. self.assertIsNone(parser.get_default("foo"))
  2641. self.assertIsNone(parser.get_default("bar"))
  2642. parser.add_argument("--bar", type=int, default=42)
  2643. self.assertIsNone(parser.get_default("foo"))
  2644. self.assertEqual(42, parser.get_default("bar"))
  2645. parser.set_defaults(foo="badger")
  2646. self.assertEqual("badger", parser.get_default("foo"))
  2647. self.assertEqual(42, parser.get_default("bar"))
  2648. # ==========================
  2649. # Namespace 'contains' tests
  2650. # ==========================
  2651. class TestNamespaceContainsSimple(TestCase):
  2652. def test_empty(self):
  2653. ns = argparse.Namespace()
  2654. self.assertNotIn('', ns)
  2655. self.assertNotIn('x', ns)
  2656. def test_non_empty(self):
  2657. ns = argparse.Namespace(x=1, y=2)
  2658. self.assertNotIn('', ns)
  2659. self.assertIn('x', ns)
  2660. self.assertIn('y', ns)
  2661. self.assertNotIn('xx', ns)
  2662. self.assertNotIn('z', ns)
  2663. # =====================
  2664. # Help formatting tests
  2665. # =====================
  2666. class TestHelpFormattingMetaclass(type):
  2667. def __init__(cls, name, bases, bodydict):
  2668. if name == 'HelpTestCase':
  2669. return
  2670. class AddTests(object):
  2671. def __init__(self, test_class, func_suffix, std_name):
  2672. self.func_suffix = func_suffix
  2673. self.std_name = std_name
  2674. for test_func in [self.test_format,
  2675. self.test_print,
  2676. self.test_print_file]:
  2677. test_name = '%s_%s' % (test_func.__name__, func_suffix)
  2678. def test_wrapper(self, test_func=test_func):
  2679. test_func(self)
  2680. try:
  2681. test_wrapper.__name__ = test_name
  2682. except TypeError:
  2683. pass
  2684. setattr(test_class, test_name, test_wrapper)
  2685. def _get_parser(self, tester):
  2686. parser = argparse.ArgumentParser(
  2687. *tester.parser_signature.args,
  2688. **tester.parser_signature.kwargs)
  2689. for argument_sig in getattr(tester, 'argument_signatures', []):
  2690. parser.add_argument(*argument_sig.args,
  2691. **argument_sig.kwargs)
  2692. group_sigs = getattr(tester, 'argument_group_signatures', [])
  2693. for group_sig, argument_sigs in group_sigs:
  2694. group = parser.add_argument_group(*group_sig.args,
  2695. **group_sig.kwargs)
  2696. for argument_sig in argument_sigs:
  2697. group.add_argument(*argument_sig.args,
  2698. **argument_sig.kwargs)
  2699. subparsers_sigs = getattr(tester, 'subparsers_signatures', [])
  2700. if subparsers_sigs:
  2701. subparsers = parser.add_subparsers()
  2702. for subparser_sig in subparsers_sigs:
  2703. subparsers.add_parser(*subparser_sig.args,
  2704. **subparser_sig.kwargs)
  2705. return parser
  2706. def _test(self, tester, parser_text):
  2707. expected_text = getattr(tester, self.func_suffix)
  2708. expected_text = textwrap.dedent(expected_text)
  2709. tester.maxDiff = None
  2710. tester.assertEqual(expected_text, parser_text)
  2711. def test_format(self, tester):
  2712. parser = self._get_parser(tester)
  2713. format = getattr(parser, 'format_%s' % self.func_suffix)
  2714. self._test(tester, format())
  2715. def test_print(self, tester):
  2716. parser = self._get_parser(tester)
  2717. print_ = getattr(parser, 'print_%s' % self.func_suffix)
  2718. old_stream = getattr(sys, self.std_name)
  2719. setattr(sys, self.std_name, StdIOBuffer())
  2720. try:
  2721. print_()
  2722. parser_text = getattr(sys, self.std_name).getvalue()
  2723. finally:
  2724. setattr(sys, self.std_name, old_stream)
  2725. self._test(tester, parser_text)
  2726. def test_print_file(self, tester):
  2727. parser = self._get_parser(tester)
  2728. print_ = getattr(parser, 'print_%s' % self.func_suffix)
  2729. sfile = StdIOBuffer()
  2730. print_(sfile)
  2731. parser_text = sfile.getvalue()
  2732. self._test(tester, parser_text)
  2733. # add tests for {format,print}_{usage,help}
  2734. for func_suffix, std_name in [('usage', 'stdout'),
  2735. ('help', 'stdout')]:
  2736. AddTests(cls, func_suffix, std_name)
  2737. bases = TestCase,
  2738. HelpTestCase = TestHelpFormattingMetaclass('HelpTestCase', bases, {})
  2739. class TestHelpBiggerOptionals(HelpTestCase):
  2740. """Make sure that argument help aligns when options are longer"""
  2741. parser_signature = Sig(prog='PROG', description='DESCRIPTION',
  2742. epilog='EPILOG')
  2743. argument_signatures = [
  2744. Sig('-v', '--version', action='version', version='0.1'),
  2745. Sig('-x', action='store_true', help='X HELP'),
  2746. Sig('--y', help='Y HELP'),
  2747. Sig('foo', help='FOO HELP'),
  2748. Sig('bar', help='BAR HELP'),
  2749. ]
  2750. argument_group_signatures = []
  2751. usage = '''\
  2752. usage: PROG [-h] [-v] [-x] [--y Y] foo bar
  2753. '''
  2754. help = usage + '''\
  2755. DESCRIPTION
  2756. positional arguments:
  2757. foo FOO HELP
  2758. bar BAR HELP
  2759. options:
  2760. -h, --help show this help message and exit
  2761. -v, --version show program's version number and exit
  2762. -x X HELP
  2763. --y Y Y HELP
  2764. EPILOG
  2765. '''
  2766. version = '''\
  2767. 0.1
  2768. '''
  2769. class TestShortColumns(HelpTestCase):
  2770. '''Test extremely small number of columns.
  2771. TestCase prevents "COLUMNS" from being too small in the tests themselves,
  2772. but we don't want any exceptions thrown in such cases. Only ugly representation.
  2773. '''
  2774. def setUp(self):
  2775. env = self.enterContext(os_helper.EnvironmentVarGuard())
  2776. env.set("COLUMNS", '15')
  2777. parser_signature = TestHelpBiggerOptionals.parser_signature
  2778. argument_signatures = TestHelpBiggerOptionals.argument_signatures
  2779. argument_group_signatures = TestHelpBiggerOptionals.argument_group_signatures
  2780. usage = '''\
  2781. usage: PROG
  2782. [-h]
  2783. [-v]
  2784. [-x]
  2785. [--y Y]
  2786. foo
  2787. bar
  2788. '''
  2789. help = usage + '''\
  2790. DESCRIPTION
  2791. positional arguments:
  2792. foo
  2793. FOO HELP
  2794. bar
  2795. BAR HELP
  2796. options:
  2797. -h, --help
  2798. show this
  2799. help
  2800. message and
  2801. exit
  2802. -v, --version
  2803. show
  2804. program's
  2805. version
  2806. number and
  2807. exit
  2808. -x
  2809. X HELP
  2810. --y Y
  2811. Y HELP
  2812. EPILOG
  2813. '''
  2814. version = TestHelpBiggerOptionals.version
  2815. class TestHelpBiggerOptionalGroups(HelpTestCase):
  2816. """Make sure that argument help aligns when options are longer"""
  2817. parser_signature = Sig(prog='PROG', description='DESCRIPTION',
  2818. epilog='EPILOG')
  2819. argument_signatures = [
  2820. Sig('-v', '--version', action='version', version='0.1'),
  2821. Sig('-x', action='store_true', help='X HELP'),
  2822. Sig('--y', help='Y HELP'),
  2823. Sig('foo', help='FOO HELP'),
  2824. Sig('bar', help='BAR HELP'),
  2825. ]
  2826. argument_group_signatures = [
  2827. (Sig('GROUP TITLE', description='GROUP DESCRIPTION'), [
  2828. Sig('baz', help='BAZ HELP'),
  2829. Sig('-z', nargs='+', help='Z HELP')]),
  2830. ]
  2831. usage = '''\
  2832. usage: PROG [-h] [-v] [-x] [--y Y] [-z Z [Z ...]] foo bar baz
  2833. '''
  2834. help = usage + '''\
  2835. DESCRIPTION
  2836. positional arguments:
  2837. foo FOO HELP
  2838. bar BAR HELP
  2839. options:
  2840. -h, --help show this help message and exit
  2841. -v, --version show program's version number and exit
  2842. -x X HELP
  2843. --y Y Y HELP
  2844. GROUP TITLE:
  2845. GROUP DESCRIPTION
  2846. baz BAZ HELP
  2847. -z Z [Z ...] Z HELP
  2848. EPILOG
  2849. '''
  2850. version = '''\
  2851. 0.1
  2852. '''
  2853. class TestHelpBiggerPositionals(HelpTestCase):
  2854. """Make sure that help aligns when arguments are longer"""
  2855. parser_signature = Sig(usage='USAGE', description='DESCRIPTION')
  2856. argument_signatures = [
  2857. Sig('-x', action='store_true', help='X HELP'),
  2858. Sig('--y', help='Y HELP'),
  2859. Sig('ekiekiekifekang', help='EKI HELP'),
  2860. Sig('bar', help='BAR HELP'),
  2861. ]
  2862. argument_group_signatures = []
  2863. usage = '''\
  2864. usage: USAGE
  2865. '''
  2866. help = usage + '''\
  2867. DESCRIPTION
  2868. positional arguments:
  2869. ekiekiekifekang EKI HELP
  2870. bar BAR HELP
  2871. options:
  2872. -h, --help show this help message and exit
  2873. -x X HELP
  2874. --y Y Y HELP
  2875. '''
  2876. version = ''
  2877. class TestHelpReformatting(HelpTestCase):
  2878. """Make sure that text after short names starts on the first line"""
  2879. parser_signature = Sig(
  2880. prog='PROG',
  2881. description=' oddly formatted\n'
  2882. 'description\n'
  2883. '\n'
  2884. 'that is so long that it should go onto multiple '
  2885. 'lines when wrapped')
  2886. argument_signatures = [
  2887. Sig('-x', metavar='XX', help='oddly\n'
  2888. ' formatted -x help'),
  2889. Sig('y', metavar='yyy', help='normal y help'),
  2890. ]
  2891. argument_group_signatures = [
  2892. (Sig('title', description='\n'
  2893. ' oddly formatted group\n'
  2894. '\n'
  2895. 'description'),
  2896. [Sig('-a', action='store_true',
  2897. help=' oddly \n'
  2898. 'formatted -a help \n'
  2899. ' again, so long that it should be wrapped over '
  2900. 'multiple lines')]),
  2901. ]
  2902. usage = '''\
  2903. usage: PROG [-h] [-x XX] [-a] yyy
  2904. '''
  2905. help = usage + '''\
  2906. oddly formatted description that is so long that it should go onto \
  2907. multiple
  2908. lines when wrapped
  2909. positional arguments:
  2910. yyy normal y help
  2911. options:
  2912. -h, --help show this help message and exit
  2913. -x XX oddly formatted -x help
  2914. title:
  2915. oddly formatted group description
  2916. -a oddly formatted -a help again, so long that it should \
  2917. be wrapped
  2918. over multiple lines
  2919. '''
  2920. version = ''
  2921. class TestHelpWrappingShortNames(HelpTestCase):
  2922. """Make sure that text after short names starts on the first line"""
  2923. parser_signature = Sig(prog='PROG', description= 'D\nD' * 30)
  2924. argument_signatures = [
  2925. Sig('-x', metavar='XX', help='XHH HX' * 20),
  2926. Sig('y', metavar='yyy', help='YH YH' * 20),
  2927. ]
  2928. argument_group_signatures = [
  2929. (Sig('ALPHAS'), [
  2930. Sig('-a', action='store_true', help='AHHH HHA' * 10)]),
  2931. ]
  2932. usage = '''\
  2933. usage: PROG [-h] [-x XX] [-a] yyy
  2934. '''
  2935. help = usage + '''\
  2936. D DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD \
  2937. DD DD DD
  2938. DD DD DD DD D
  2939. positional arguments:
  2940. yyy YH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH \
  2941. YHYH YHYH
  2942. YHYH YHYH YHYH YHYH YHYH YHYH YHYH YH
  2943. options:
  2944. -h, --help show this help message and exit
  2945. -x XX XHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH \
  2946. HXXHH HXXHH
  2947. HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HX
  2948. ALPHAS:
  2949. -a AHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH \
  2950. HHAAHHH
  2951. HHAAHHH HHAAHHH HHA
  2952. '''
  2953. version = ''
  2954. class TestHelpWrappingLongNames(HelpTestCase):
  2955. """Make sure that text after long names starts on the next line"""
  2956. parser_signature = Sig(usage='USAGE', description= 'D D' * 30)
  2957. argument_signatures = [
  2958. Sig('-v', '--version', action='version', version='V V' * 30),
  2959. Sig('-x', metavar='X' * 25, help='XH XH' * 20),
  2960. Sig('y', metavar='y' * 25, help='YH YH' * 20),
  2961. ]
  2962. argument_group_signatures = [
  2963. (Sig('ALPHAS'), [
  2964. Sig('-a', metavar='A' * 25, help='AH AH' * 20),
  2965. Sig('z', metavar='z' * 25, help='ZH ZH' * 20)]),
  2966. ]
  2967. usage = '''\
  2968. usage: USAGE
  2969. '''
  2970. help = usage + '''\
  2971. D DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD \
  2972. DD DD DD
  2973. DD DD DD DD D
  2974. positional arguments:
  2975. yyyyyyyyyyyyyyyyyyyyyyyyy
  2976. YH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH \
  2977. YHYH YHYH
  2978. YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YH
  2979. options:
  2980. -h, --help show this help message and exit
  2981. -v, --version show program's version number and exit
  2982. -x XXXXXXXXXXXXXXXXXXXXXXXXX
  2983. XH XHXH XHXH XHXH XHXH XHXH XHXH XHXH XHXH \
  2984. XHXH XHXH
  2985. XHXH XHXH XHXH XHXH XHXH XHXH XHXH XHXH XHXH XH
  2986. ALPHAS:
  2987. -a AAAAAAAAAAAAAAAAAAAAAAAAA
  2988. AH AHAH AHAH AHAH AHAH AHAH AHAH AHAH AHAH \
  2989. AHAH AHAH
  2990. AHAH AHAH AHAH AHAH AHAH AHAH AHAH AHAH AHAH AH
  2991. zzzzzzzzzzzzzzzzzzzzzzzzz
  2992. ZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH \
  2993. ZHZH ZHZH
  2994. ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZH
  2995. '''
  2996. version = '''\
  2997. V VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV \
  2998. VV VV VV
  2999. VV VV VV VV V
  3000. '''
  3001. class TestHelpUsage(HelpTestCase):
  3002. """Test basic usage messages"""
  3003. parser_signature = Sig(prog='PROG')
  3004. argument_signatures = [
  3005. Sig('-w', nargs='+', help='w'),
  3006. Sig('-x', nargs='*', help='x'),
  3007. Sig('a', help='a'),
  3008. Sig('b', help='b', nargs=2),
  3009. Sig('c', help='c', nargs='?'),
  3010. Sig('--foo', help='Whether to foo', action=argparse.BooleanOptionalAction),
  3011. Sig('--bar', help='Whether to bar', default=True,
  3012. action=argparse.BooleanOptionalAction),
  3013. Sig('-f', '--foobar', '--barfoo', action=argparse.BooleanOptionalAction),
  3014. Sig('--bazz', action=argparse.BooleanOptionalAction,
  3015. default=argparse.SUPPRESS, help='Bazz!'),
  3016. ]
  3017. argument_group_signatures = [
  3018. (Sig('group'), [
  3019. Sig('-y', nargs='?', help='y'),
  3020. Sig('-z', nargs=3, help='z'),
  3021. Sig('d', help='d', nargs='*'),
  3022. Sig('e', help='e', nargs='+'),
  3023. ])
  3024. ]
  3025. usage = '''\
  3026. usage: PROG [-h] [-w W [W ...]] [-x [X ...]] [--foo | --no-foo]
  3027. [--bar | --no-bar]
  3028. [-f | --foobar | --no-foobar | --barfoo | --no-barfoo]
  3029. [--bazz | --no-bazz] [-y [Y]] [-z Z Z Z]
  3030. a b b [c] [d ...] e [e ...]
  3031. '''
  3032. help = usage + '''\
  3033. positional arguments:
  3034. a a
  3035. b b
  3036. c c
  3037. options:
  3038. -h, --help show this help message and exit
  3039. -w W [W ...] w
  3040. -x [X ...] x
  3041. --foo, --no-foo Whether to foo
  3042. --bar, --no-bar Whether to bar
  3043. -f, --foobar, --no-foobar, --barfoo, --no-barfoo
  3044. --bazz, --no-bazz Bazz!
  3045. group:
  3046. -y [Y] y
  3047. -z Z Z Z z
  3048. d d
  3049. e e
  3050. '''
  3051. version = ''
  3052. class TestHelpOnlyUserGroups(HelpTestCase):
  3053. """Test basic usage messages"""
  3054. parser_signature = Sig(prog='PROG', add_help=False)
  3055. argument_signatures = []
  3056. argument_group_signatures = [
  3057. (Sig('xxxx'), [
  3058. Sig('-x', help='x'),
  3059. Sig('a', help='a'),
  3060. ]),
  3061. (Sig('yyyy'), [
  3062. Sig('b', help='b'),
  3063. Sig('-y', help='y'),
  3064. ]),
  3065. ]
  3066. usage = '''\
  3067. usage: PROG [-x X] [-y Y] a b
  3068. '''
  3069. help = usage + '''\
  3070. xxxx:
  3071. -x X x
  3072. a a
  3073. yyyy:
  3074. b b
  3075. -y Y y
  3076. '''
  3077. version = ''
  3078. class TestHelpUsageLongProg(HelpTestCase):
  3079. """Test usage messages where the prog is long"""
  3080. parser_signature = Sig(prog='P' * 60)
  3081. argument_signatures = [
  3082. Sig('-w', metavar='W'),
  3083. Sig('-x', metavar='X'),
  3084. Sig('a'),
  3085. Sig('b'),
  3086. ]
  3087. argument_group_signatures = []
  3088. usage = '''\
  3089. usage: PPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPP
  3090. [-h] [-w W] [-x X] a b
  3091. '''
  3092. help = usage + '''\
  3093. positional arguments:
  3094. a
  3095. b
  3096. options:
  3097. -h, --help show this help message and exit
  3098. -w W
  3099. -x X
  3100. '''
  3101. version = ''
  3102. class TestHelpUsageLongProgOptionsWrap(HelpTestCase):
  3103. """Test usage messages where the prog is long and the optionals wrap"""
  3104. parser_signature = Sig(prog='P' * 60)
  3105. argument_signatures = [
  3106. Sig('-w', metavar='W' * 25),
  3107. Sig('-x', metavar='X' * 25),
  3108. Sig('-y', metavar='Y' * 25),
  3109. Sig('-z', metavar='Z' * 25),
  3110. Sig('a'),
  3111. Sig('b'),
  3112. ]
  3113. argument_group_signatures = []
  3114. usage = '''\
  3115. usage: PPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPP
  3116. [-h] [-w WWWWWWWWWWWWWWWWWWWWWWWWW] \
  3117. [-x XXXXXXXXXXXXXXXXXXXXXXXXX]
  3118. [-y YYYYYYYYYYYYYYYYYYYYYYYYY] [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
  3119. a b
  3120. '''
  3121. help = usage + '''\
  3122. positional arguments:
  3123. a
  3124. b
  3125. options:
  3126. -h, --help show this help message and exit
  3127. -w WWWWWWWWWWWWWWWWWWWWWWWWW
  3128. -x XXXXXXXXXXXXXXXXXXXXXXXXX
  3129. -y YYYYYYYYYYYYYYYYYYYYYYYYY
  3130. -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
  3131. '''
  3132. version = ''
  3133. class TestHelpUsageLongProgPositionalsWrap(HelpTestCase):
  3134. """Test usage messages where the prog is long and the positionals wrap"""
  3135. parser_signature = Sig(prog='P' * 60, add_help=False)
  3136. argument_signatures = [
  3137. Sig('a' * 25),
  3138. Sig('b' * 25),
  3139. Sig('c' * 25),
  3140. ]
  3141. argument_group_signatures = []
  3142. usage = '''\
  3143. usage: PPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPP
  3144. aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
  3145. ccccccccccccccccccccccccc
  3146. '''
  3147. help = usage + '''\
  3148. positional arguments:
  3149. aaaaaaaaaaaaaaaaaaaaaaaaa
  3150. bbbbbbbbbbbbbbbbbbbbbbbbb
  3151. ccccccccccccccccccccccccc
  3152. '''
  3153. version = ''
  3154. class TestHelpUsageOptionalsWrap(HelpTestCase):
  3155. """Test usage messages where the optionals wrap"""
  3156. parser_signature = Sig(prog='PROG')
  3157. argument_signatures = [
  3158. Sig('-w', metavar='W' * 25),
  3159. Sig('-x', metavar='X' * 25),
  3160. Sig('-y', metavar='Y' * 25),
  3161. Sig('-z', metavar='Z' * 25),
  3162. Sig('a'),
  3163. Sig('b'),
  3164. Sig('c'),
  3165. ]
  3166. argument_group_signatures = []
  3167. usage = '''\
  3168. usage: PROG [-h] [-w WWWWWWWWWWWWWWWWWWWWWWWWW] \
  3169. [-x XXXXXXXXXXXXXXXXXXXXXXXXX]
  3170. [-y YYYYYYYYYYYYYYYYYYYYYYYYY] \
  3171. [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
  3172. a b c
  3173. '''
  3174. help = usage + '''\
  3175. positional arguments:
  3176. a
  3177. b
  3178. c
  3179. options:
  3180. -h, --help show this help message and exit
  3181. -w WWWWWWWWWWWWWWWWWWWWWWWWW
  3182. -x XXXXXXXXXXXXXXXXXXXXXXXXX
  3183. -y YYYYYYYYYYYYYYYYYYYYYYYYY
  3184. -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
  3185. '''
  3186. version = ''
  3187. class TestHelpUsagePositionalsWrap(HelpTestCase):
  3188. """Test usage messages where the positionals wrap"""
  3189. parser_signature = Sig(prog='PROG')
  3190. argument_signatures = [
  3191. Sig('-x'),
  3192. Sig('-y'),
  3193. Sig('-z'),
  3194. Sig('a' * 25),
  3195. Sig('b' * 25),
  3196. Sig('c' * 25),
  3197. ]
  3198. argument_group_signatures = []
  3199. usage = '''\
  3200. usage: PROG [-h] [-x X] [-y Y] [-z Z]
  3201. aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
  3202. ccccccccccccccccccccccccc
  3203. '''
  3204. help = usage + '''\
  3205. positional arguments:
  3206. aaaaaaaaaaaaaaaaaaaaaaaaa
  3207. bbbbbbbbbbbbbbbbbbbbbbbbb
  3208. ccccccccccccccccccccccccc
  3209. options:
  3210. -h, --help show this help message and exit
  3211. -x X
  3212. -y Y
  3213. -z Z
  3214. '''
  3215. version = ''
  3216. class TestHelpUsageOptionalsPositionalsWrap(HelpTestCase):
  3217. """Test usage messages where the optionals and positionals wrap"""
  3218. parser_signature = Sig(prog='PROG')
  3219. argument_signatures = [
  3220. Sig('-x', metavar='X' * 25),
  3221. Sig('-y', metavar='Y' * 25),
  3222. Sig('-z', metavar='Z' * 25),
  3223. Sig('a' * 25),
  3224. Sig('b' * 25),
  3225. Sig('c' * 25),
  3226. ]
  3227. argument_group_signatures = []
  3228. usage = '''\
  3229. usage: PROG [-h] [-x XXXXXXXXXXXXXXXXXXXXXXXXX] \
  3230. [-y YYYYYYYYYYYYYYYYYYYYYYYYY]
  3231. [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
  3232. aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
  3233. ccccccccccccccccccccccccc
  3234. '''
  3235. help = usage + '''\
  3236. positional arguments:
  3237. aaaaaaaaaaaaaaaaaaaaaaaaa
  3238. bbbbbbbbbbbbbbbbbbbbbbbbb
  3239. ccccccccccccccccccccccccc
  3240. options:
  3241. -h, --help show this help message and exit
  3242. -x XXXXXXXXXXXXXXXXXXXXXXXXX
  3243. -y YYYYYYYYYYYYYYYYYYYYYYYYY
  3244. -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
  3245. '''
  3246. version = ''
  3247. class TestHelpUsageOptionalsOnlyWrap(HelpTestCase):
  3248. """Test usage messages where there are only optionals and they wrap"""
  3249. parser_signature = Sig(prog='PROG')
  3250. argument_signatures = [
  3251. Sig('-x', metavar='X' * 25),
  3252. Sig('-y', metavar='Y' * 25),
  3253. Sig('-z', metavar='Z' * 25),
  3254. ]
  3255. argument_group_signatures = []
  3256. usage = '''\
  3257. usage: PROG [-h] [-x XXXXXXXXXXXXXXXXXXXXXXXXX] \
  3258. [-y YYYYYYYYYYYYYYYYYYYYYYYYY]
  3259. [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
  3260. '''
  3261. help = usage + '''\
  3262. options:
  3263. -h, --help show this help message and exit
  3264. -x XXXXXXXXXXXXXXXXXXXXXXXXX
  3265. -y YYYYYYYYYYYYYYYYYYYYYYYYY
  3266. -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
  3267. '''
  3268. version = ''
  3269. class TestHelpUsagePositionalsOnlyWrap(HelpTestCase):
  3270. """Test usage messages where there are only positionals and they wrap"""
  3271. parser_signature = Sig(prog='PROG', add_help=False)
  3272. argument_signatures = [
  3273. Sig('a' * 25),
  3274. Sig('b' * 25),
  3275. Sig('c' * 25),
  3276. ]
  3277. argument_group_signatures = []
  3278. usage = '''\
  3279. usage: PROG aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
  3280. ccccccccccccccccccccccccc
  3281. '''
  3282. help = usage + '''\
  3283. positional arguments:
  3284. aaaaaaaaaaaaaaaaaaaaaaaaa
  3285. bbbbbbbbbbbbbbbbbbbbbbbbb
  3286. ccccccccccccccccccccccccc
  3287. '''
  3288. version = ''
  3289. class TestHelpVariableExpansion(HelpTestCase):
  3290. """Test that variables are expanded properly in help messages"""
  3291. parser_signature = Sig(prog='PROG')
  3292. argument_signatures = [
  3293. Sig('-x', type=int,
  3294. help='x %(prog)s %(default)s %(type)s %%'),
  3295. Sig('-y', action='store_const', default=42, const='XXX',
  3296. help='y %(prog)s %(default)s %(const)s'),
  3297. Sig('--foo', choices='abc',
  3298. help='foo %(prog)s %(default)s %(choices)s'),
  3299. Sig('--bar', default='baz', choices=[1, 2], metavar='BBB',
  3300. help='bar %(prog)s %(default)s %(dest)s'),
  3301. Sig('spam', help='spam %(prog)s %(default)s'),
  3302. Sig('badger', default=0.5, help='badger %(prog)s %(default)s'),
  3303. ]
  3304. argument_group_signatures = [
  3305. (Sig('group'), [
  3306. Sig('-a', help='a %(prog)s %(default)s'),
  3307. Sig('-b', default=-1, help='b %(prog)s %(default)s'),
  3308. ])
  3309. ]
  3310. usage = ('''\
  3311. usage: PROG [-h] [-x X] [-y] [--foo {a,b,c}] [--bar BBB] [-a A] [-b B]
  3312. spam badger
  3313. ''')
  3314. help = usage + '''\
  3315. positional arguments:
  3316. spam spam PROG None
  3317. badger badger PROG 0.5
  3318. options:
  3319. -h, --help show this help message and exit
  3320. -x X x PROG None int %
  3321. -y y PROG 42 XXX
  3322. --foo {a,b,c} foo PROG None a, b, c
  3323. --bar BBB bar PROG baz bar
  3324. group:
  3325. -a A a PROG None
  3326. -b B b PROG -1
  3327. '''
  3328. version = ''
  3329. class TestHelpVariableExpansionUsageSupplied(HelpTestCase):
  3330. """Test that variables are expanded properly when usage= is present"""
  3331. parser_signature = Sig(prog='PROG', usage='%(prog)s FOO')
  3332. argument_signatures = []
  3333. argument_group_signatures = []
  3334. usage = ('''\
  3335. usage: PROG FOO
  3336. ''')
  3337. help = usage + '''\
  3338. options:
  3339. -h, --help show this help message and exit
  3340. '''
  3341. version = ''
  3342. class TestHelpVariableExpansionNoArguments(HelpTestCase):
  3343. """Test that variables are expanded properly with no arguments"""
  3344. parser_signature = Sig(prog='PROG', add_help=False)
  3345. argument_signatures = []
  3346. argument_group_signatures = []
  3347. usage = ('''\
  3348. usage: PROG
  3349. ''')
  3350. help = usage
  3351. version = ''
  3352. class TestHelpSuppressUsage(HelpTestCase):
  3353. """Test that items can be suppressed in usage messages"""
  3354. parser_signature = Sig(prog='PROG', usage=argparse.SUPPRESS)
  3355. argument_signatures = [
  3356. Sig('--foo', help='foo help'),
  3357. Sig('spam', help='spam help'),
  3358. ]
  3359. argument_group_signatures = []
  3360. help = '''\
  3361. positional arguments:
  3362. spam spam help
  3363. options:
  3364. -h, --help show this help message and exit
  3365. --foo FOO foo help
  3366. '''
  3367. usage = ''
  3368. version = ''
  3369. class TestHelpSuppressOptional(HelpTestCase):
  3370. """Test that optional arguments can be suppressed in help messages"""
  3371. parser_signature = Sig(prog='PROG', add_help=False)
  3372. argument_signatures = [
  3373. Sig('--foo', help=argparse.SUPPRESS),
  3374. Sig('spam', help='spam help'),
  3375. ]
  3376. argument_group_signatures = []
  3377. usage = '''\
  3378. usage: PROG spam
  3379. '''
  3380. help = usage + '''\
  3381. positional arguments:
  3382. spam spam help
  3383. '''
  3384. version = ''
  3385. class TestHelpSuppressOptionalGroup(HelpTestCase):
  3386. """Test that optional groups can be suppressed in help messages"""
  3387. parser_signature = Sig(prog='PROG')
  3388. argument_signatures = [
  3389. Sig('--foo', help='foo help'),
  3390. Sig('spam', help='spam help'),
  3391. ]
  3392. argument_group_signatures = [
  3393. (Sig('group'), [Sig('--bar', help=argparse.SUPPRESS)]),
  3394. ]
  3395. usage = '''\
  3396. usage: PROG [-h] [--foo FOO] spam
  3397. '''
  3398. help = usage + '''\
  3399. positional arguments:
  3400. spam spam help
  3401. options:
  3402. -h, --help show this help message and exit
  3403. --foo FOO foo help
  3404. '''
  3405. version = ''
  3406. class TestHelpSuppressPositional(HelpTestCase):
  3407. """Test that positional arguments can be suppressed in help messages"""
  3408. parser_signature = Sig(prog='PROG')
  3409. argument_signatures = [
  3410. Sig('--foo', help='foo help'),
  3411. Sig('spam', help=argparse.SUPPRESS),
  3412. ]
  3413. argument_group_signatures = []
  3414. usage = '''\
  3415. usage: PROG [-h] [--foo FOO]
  3416. '''
  3417. help = usage + '''\
  3418. options:
  3419. -h, --help show this help message and exit
  3420. --foo FOO foo help
  3421. '''
  3422. version = ''
  3423. class TestHelpRequiredOptional(HelpTestCase):
  3424. """Test that required options don't look optional"""
  3425. parser_signature = Sig(prog='PROG')
  3426. argument_signatures = [
  3427. Sig('--foo', required=True, help='foo help'),
  3428. ]
  3429. argument_group_signatures = []
  3430. usage = '''\
  3431. usage: PROG [-h] --foo FOO
  3432. '''
  3433. help = usage + '''\
  3434. options:
  3435. -h, --help show this help message and exit
  3436. --foo FOO foo help
  3437. '''
  3438. version = ''
  3439. class TestHelpAlternatePrefixChars(HelpTestCase):
  3440. """Test that options display with different prefix characters"""
  3441. parser_signature = Sig(prog='PROG', prefix_chars='^;', add_help=False)
  3442. argument_signatures = [
  3443. Sig('^^foo', action='store_true', help='foo help'),
  3444. Sig(';b', ';;bar', help='bar help'),
  3445. ]
  3446. argument_group_signatures = []
  3447. usage = '''\
  3448. usage: PROG [^^foo] [;b BAR]
  3449. '''
  3450. help = usage + '''\
  3451. options:
  3452. ^^foo foo help
  3453. ;b BAR, ;;bar BAR bar help
  3454. '''
  3455. version = ''
  3456. class TestHelpNoHelpOptional(HelpTestCase):
  3457. """Test that the --help argument can be suppressed help messages"""
  3458. parser_signature = Sig(prog='PROG', add_help=False)
  3459. argument_signatures = [
  3460. Sig('--foo', help='foo help'),
  3461. Sig('spam', help='spam help'),
  3462. ]
  3463. argument_group_signatures = []
  3464. usage = '''\
  3465. usage: PROG [--foo FOO] spam
  3466. '''
  3467. help = usage + '''\
  3468. positional arguments:
  3469. spam spam help
  3470. options:
  3471. --foo FOO foo help
  3472. '''
  3473. version = ''
  3474. class TestHelpNone(HelpTestCase):
  3475. """Test that no errors occur if no help is specified"""
  3476. parser_signature = Sig(prog='PROG')
  3477. argument_signatures = [
  3478. Sig('--foo'),
  3479. Sig('spam'),
  3480. ]
  3481. argument_group_signatures = []
  3482. usage = '''\
  3483. usage: PROG [-h] [--foo FOO] spam
  3484. '''
  3485. help = usage + '''\
  3486. positional arguments:
  3487. spam
  3488. options:
  3489. -h, --help show this help message and exit
  3490. --foo FOO
  3491. '''
  3492. version = ''
  3493. class TestHelpTupleMetavar(HelpTestCase):
  3494. """Test specifying metavar as a tuple"""
  3495. parser_signature = Sig(prog='PROG')
  3496. argument_signatures = [
  3497. Sig('-w', help='w', nargs='+', metavar=('W1', 'W2')),
  3498. Sig('-x', help='x', nargs='*', metavar=('X1', 'X2')),
  3499. Sig('-y', help='y', nargs=3, metavar=('Y1', 'Y2', 'Y3')),
  3500. Sig('-z', help='z', nargs='?', metavar=('Z1', )),
  3501. ]
  3502. argument_group_signatures = []
  3503. usage = '''\
  3504. usage: PROG [-h] [-w W1 [W2 ...]] [-x [X1 [X2 ...]]] [-y Y1 Y2 Y3] \
  3505. [-z [Z1]]
  3506. '''
  3507. help = usage + '''\
  3508. options:
  3509. -h, --help show this help message and exit
  3510. -w W1 [W2 ...] w
  3511. -x [X1 [X2 ...]] x
  3512. -y Y1 Y2 Y3 y
  3513. -z [Z1] z
  3514. '''
  3515. version = ''
  3516. class TestHelpRawText(HelpTestCase):
  3517. """Test the RawTextHelpFormatter"""
  3518. parser_signature = Sig(
  3519. prog='PROG', formatter_class=argparse.RawTextHelpFormatter,
  3520. description='Keep the formatting\n'
  3521. ' exactly as it is written\n'
  3522. '\n'
  3523. 'here\n')
  3524. argument_signatures = [
  3525. Sig('--foo', help=' foo help should also\n'
  3526. 'appear as given here'),
  3527. Sig('spam', help='spam help'),
  3528. ]
  3529. argument_group_signatures = [
  3530. (Sig('title', description=' This text\n'
  3531. ' should be indented\n'
  3532. ' exactly like it is here\n'),
  3533. [Sig('--bar', help='bar help')]),
  3534. ]
  3535. usage = '''\
  3536. usage: PROG [-h] [--foo FOO] [--bar BAR] spam
  3537. '''
  3538. help = usage + '''\
  3539. Keep the formatting
  3540. exactly as it is written
  3541. here
  3542. positional arguments:
  3543. spam spam help
  3544. options:
  3545. -h, --help show this help message and exit
  3546. --foo FOO foo help should also
  3547. appear as given here
  3548. title:
  3549. This text
  3550. should be indented
  3551. exactly like it is here
  3552. --bar BAR bar help
  3553. '''
  3554. version = ''
  3555. class TestHelpRawDescription(HelpTestCase):
  3556. """Test the RawTextHelpFormatter"""
  3557. parser_signature = Sig(
  3558. prog='PROG', formatter_class=argparse.RawDescriptionHelpFormatter,
  3559. description='Keep the formatting\n'
  3560. ' exactly as it is written\n'
  3561. '\n'
  3562. 'here\n')
  3563. argument_signatures = [
  3564. Sig('--foo', help=' foo help should not\n'
  3565. ' retain this odd formatting'),
  3566. Sig('spam', help='spam help'),
  3567. ]
  3568. argument_group_signatures = [
  3569. (Sig('title', description=' This text\n'
  3570. ' should be indented\n'
  3571. ' exactly like it is here\n'),
  3572. [Sig('--bar', help='bar help')]),
  3573. ]
  3574. usage = '''\
  3575. usage: PROG [-h] [--foo FOO] [--bar BAR] spam
  3576. '''
  3577. help = usage + '''\
  3578. Keep the formatting
  3579. exactly as it is written
  3580. here
  3581. positional arguments:
  3582. spam spam help
  3583. options:
  3584. -h, --help show this help message and exit
  3585. --foo FOO foo help should not retain this odd formatting
  3586. title:
  3587. This text
  3588. should be indented
  3589. exactly like it is here
  3590. --bar BAR bar help
  3591. '''
  3592. version = ''
  3593. class TestHelpArgumentDefaults(HelpTestCase):
  3594. """Test the ArgumentDefaultsHelpFormatter"""
  3595. parser_signature = Sig(
  3596. prog='PROG', formatter_class=argparse.ArgumentDefaultsHelpFormatter,
  3597. description='description')
  3598. argument_signatures = [
  3599. Sig('--foo', help='foo help - oh and by the way, %(default)s'),
  3600. Sig('--bar', action='store_true', help='bar help'),
  3601. Sig('--taz', action=argparse.BooleanOptionalAction,
  3602. help='Whether to taz it', default=True),
  3603. Sig('--corge', action=argparse.BooleanOptionalAction,
  3604. help='Whether to corge it', default=argparse.SUPPRESS),
  3605. Sig('--quux', help="Set the quux", default=42),
  3606. Sig('spam', help='spam help'),
  3607. Sig('badger', nargs='?', default='wooden', help='badger help'),
  3608. ]
  3609. argument_group_signatures = [
  3610. (Sig('title', description='description'),
  3611. [Sig('--baz', type=int, default=42, help='baz help')]),
  3612. ]
  3613. usage = '''\
  3614. usage: PROG [-h] [--foo FOO] [--bar] [--taz | --no-taz] [--corge | --no-corge]
  3615. [--quux QUUX] [--baz BAZ]
  3616. spam [badger]
  3617. '''
  3618. help = usage + '''\
  3619. description
  3620. positional arguments:
  3621. spam spam help
  3622. badger badger help (default: wooden)
  3623. options:
  3624. -h, --help show this help message and exit
  3625. --foo FOO foo help - oh and by the way, None
  3626. --bar bar help (default: False)
  3627. --taz, --no-taz Whether to taz it (default: True)
  3628. --corge, --no-corge Whether to corge it
  3629. --quux QUUX Set the quux (default: 42)
  3630. title:
  3631. description
  3632. --baz BAZ baz help (default: 42)
  3633. '''
  3634. version = ''
  3635. class TestHelpVersionAction(HelpTestCase):
  3636. """Test the default help for the version action"""
  3637. parser_signature = Sig(prog='PROG', description='description')
  3638. argument_signatures = [Sig('-V', '--version', action='version', version='3.6')]
  3639. argument_group_signatures = []
  3640. usage = '''\
  3641. usage: PROG [-h] [-V]
  3642. '''
  3643. help = usage + '''\
  3644. description
  3645. options:
  3646. -h, --help show this help message and exit
  3647. -V, --version show program's version number and exit
  3648. '''
  3649. version = ''
  3650. class TestHelpVersionActionSuppress(HelpTestCase):
  3651. """Test that the --version argument can be suppressed in help messages"""
  3652. parser_signature = Sig(prog='PROG')
  3653. argument_signatures = [
  3654. Sig('-v', '--version', action='version', version='1.0',
  3655. help=argparse.SUPPRESS),
  3656. Sig('--foo', help='foo help'),
  3657. Sig('spam', help='spam help'),
  3658. ]
  3659. argument_group_signatures = []
  3660. usage = '''\
  3661. usage: PROG [-h] [--foo FOO] spam
  3662. '''
  3663. help = usage + '''\
  3664. positional arguments:
  3665. spam spam help
  3666. options:
  3667. -h, --help show this help message and exit
  3668. --foo FOO foo help
  3669. '''
  3670. class TestHelpSubparsersOrdering(HelpTestCase):
  3671. """Test ordering of subcommands in help matches the code"""
  3672. parser_signature = Sig(prog='PROG',
  3673. description='display some subcommands')
  3674. argument_signatures = [Sig('-v', '--version', action='version', version='0.1')]
  3675. subparsers_signatures = [Sig(name=name)
  3676. for name in ('a', 'b', 'c', 'd', 'e')]
  3677. usage = '''\
  3678. usage: PROG [-h] [-v] {a,b,c,d,e} ...
  3679. '''
  3680. help = usage + '''\
  3681. display some subcommands
  3682. positional arguments:
  3683. {a,b,c,d,e}
  3684. options:
  3685. -h, --help show this help message and exit
  3686. -v, --version show program's version number and exit
  3687. '''
  3688. version = '''\
  3689. 0.1
  3690. '''
  3691. class TestHelpSubparsersWithHelpOrdering(HelpTestCase):
  3692. """Test ordering of subcommands in help matches the code"""
  3693. parser_signature = Sig(prog='PROG',
  3694. description='display some subcommands')
  3695. argument_signatures = [Sig('-v', '--version', action='version', version='0.1')]
  3696. subcommand_data = (('a', 'a subcommand help'),
  3697. ('b', 'b subcommand help'),
  3698. ('c', 'c subcommand help'),
  3699. ('d', 'd subcommand help'),
  3700. ('e', 'e subcommand help'),
  3701. )
  3702. subparsers_signatures = [Sig(name=name, help=help)
  3703. for name, help in subcommand_data]
  3704. usage = '''\
  3705. usage: PROG [-h] [-v] {a,b,c,d,e} ...
  3706. '''
  3707. help = usage + '''\
  3708. display some subcommands
  3709. positional arguments:
  3710. {a,b,c,d,e}
  3711. a a subcommand help
  3712. b b subcommand help
  3713. c c subcommand help
  3714. d d subcommand help
  3715. e e subcommand help
  3716. options:
  3717. -h, --help show this help message and exit
  3718. -v, --version show program's version number and exit
  3719. '''
  3720. version = '''\
  3721. 0.1
  3722. '''
  3723. class TestHelpMetavarTypeFormatter(HelpTestCase):
  3724. def custom_type(string):
  3725. return string
  3726. parser_signature = Sig(prog='PROG', description='description',
  3727. formatter_class=argparse.MetavarTypeHelpFormatter)
  3728. argument_signatures = [Sig('a', type=int),
  3729. Sig('-b', type=custom_type),
  3730. Sig('-c', type=float, metavar='SOME FLOAT')]
  3731. argument_group_signatures = []
  3732. usage = '''\
  3733. usage: PROG [-h] [-b custom_type] [-c SOME FLOAT] int
  3734. '''
  3735. help = usage + '''\
  3736. description
  3737. positional arguments:
  3738. int
  3739. options:
  3740. -h, --help show this help message and exit
  3741. -b custom_type
  3742. -c SOME FLOAT
  3743. '''
  3744. version = ''
  3745. # =====================================
  3746. # Optional/Positional constructor tests
  3747. # =====================================
  3748. class TestInvalidArgumentConstructors(TestCase):
  3749. """Test a bunch of invalid Argument constructors"""
  3750. def assertTypeError(self, *args, **kwargs):
  3751. parser = argparse.ArgumentParser()
  3752. self.assertRaises(TypeError, parser.add_argument,
  3753. *args, **kwargs)
  3754. def assertValueError(self, *args, **kwargs):
  3755. parser = argparse.ArgumentParser()
  3756. self.assertRaises(ValueError, parser.add_argument,
  3757. *args, **kwargs)
  3758. def test_invalid_keyword_arguments(self):
  3759. self.assertTypeError('-x', bar=None)
  3760. self.assertTypeError('-y', callback='foo')
  3761. self.assertTypeError('-y', callback_args=())
  3762. self.assertTypeError('-y', callback_kwargs={})
  3763. def test_missing_destination(self):
  3764. self.assertTypeError()
  3765. for action in ['append', 'store']:
  3766. self.assertTypeError(action=action)
  3767. def test_invalid_option_strings(self):
  3768. self.assertValueError('--')
  3769. self.assertValueError('---')
  3770. def test_invalid_type(self):
  3771. self.assertValueError('--foo', type='int')
  3772. self.assertValueError('--foo', type=(int, float))
  3773. def test_invalid_action(self):
  3774. self.assertValueError('-x', action='foo')
  3775. self.assertValueError('foo', action='baz')
  3776. self.assertValueError('--foo', action=('store', 'append'))
  3777. parser = argparse.ArgumentParser()
  3778. with self.assertRaises(ValueError) as cm:
  3779. parser.add_argument("--foo", action="store-true")
  3780. self.assertIn('unknown action', str(cm.exception))
  3781. def test_multiple_dest(self):
  3782. parser = argparse.ArgumentParser()
  3783. parser.add_argument(dest='foo')
  3784. with self.assertRaises(ValueError) as cm:
  3785. parser.add_argument('bar', dest='baz')
  3786. self.assertIn('dest supplied twice for positional argument',
  3787. str(cm.exception))
  3788. def test_no_argument_actions(self):
  3789. for action in ['store_const', 'store_true', 'store_false',
  3790. 'append_const', 'count']:
  3791. for attrs in [dict(type=int), dict(nargs='+'),
  3792. dict(choices='ab')]:
  3793. self.assertTypeError('-x', action=action, **attrs)
  3794. def test_no_argument_no_const_actions(self):
  3795. # options with zero arguments
  3796. for action in ['store_true', 'store_false', 'count']:
  3797. # const is always disallowed
  3798. self.assertTypeError('-x', const='foo', action=action)
  3799. # nargs is always disallowed
  3800. self.assertTypeError('-x', nargs='*', action=action)
  3801. def test_more_than_one_argument_actions(self):
  3802. for action in ['store', 'append']:
  3803. # nargs=0 is disallowed
  3804. self.assertValueError('-x', nargs=0, action=action)
  3805. self.assertValueError('spam', nargs=0, action=action)
  3806. # const is disallowed with non-optional arguments
  3807. for nargs in [1, '*', '+']:
  3808. self.assertValueError('-x', const='foo',
  3809. nargs=nargs, action=action)
  3810. self.assertValueError('spam', const='foo',
  3811. nargs=nargs, action=action)
  3812. def test_required_const_actions(self):
  3813. for action in ['store_const', 'append_const']:
  3814. # nargs is always disallowed
  3815. self.assertTypeError('-x', nargs='+', action=action)
  3816. def test_parsers_action_missing_params(self):
  3817. self.assertTypeError('command', action='parsers')
  3818. self.assertTypeError('command', action='parsers', prog='PROG')
  3819. self.assertTypeError('command', action='parsers',
  3820. parser_class=argparse.ArgumentParser)
  3821. def test_required_positional(self):
  3822. self.assertTypeError('foo', required=True)
  3823. def test_user_defined_action(self):
  3824. class Success(Exception):
  3825. pass
  3826. class Action(object):
  3827. def __init__(self,
  3828. option_strings,
  3829. dest,
  3830. const,
  3831. default,
  3832. required=False):
  3833. if dest == 'spam':
  3834. if const is Success:
  3835. if default is Success:
  3836. raise Success()
  3837. def __call__(self, *args, **kwargs):
  3838. pass
  3839. parser = argparse.ArgumentParser()
  3840. self.assertRaises(Success, parser.add_argument, '--spam',
  3841. action=Action, default=Success, const=Success)
  3842. self.assertRaises(Success, parser.add_argument, 'spam',
  3843. action=Action, default=Success, const=Success)
  3844. # ================================
  3845. # Actions returned by add_argument
  3846. # ================================
  3847. class TestActionsReturned(TestCase):
  3848. def test_dest(self):
  3849. parser = argparse.ArgumentParser()
  3850. action = parser.add_argument('--foo')
  3851. self.assertEqual(action.dest, 'foo')
  3852. action = parser.add_argument('-b', '--bar')
  3853. self.assertEqual(action.dest, 'bar')
  3854. action = parser.add_argument('-x', '-y')
  3855. self.assertEqual(action.dest, 'x')
  3856. def test_misc(self):
  3857. parser = argparse.ArgumentParser()
  3858. action = parser.add_argument('--foo', nargs='?', const=42,
  3859. default=84, type=int, choices=[1, 2],
  3860. help='FOO', metavar='BAR', dest='baz')
  3861. self.assertEqual(action.nargs, '?')
  3862. self.assertEqual(action.const, 42)
  3863. self.assertEqual(action.default, 84)
  3864. self.assertEqual(action.type, int)
  3865. self.assertEqual(action.choices, [1, 2])
  3866. self.assertEqual(action.help, 'FOO')
  3867. self.assertEqual(action.metavar, 'BAR')
  3868. self.assertEqual(action.dest, 'baz')
  3869. # ================================
  3870. # Argument conflict handling tests
  3871. # ================================
  3872. class TestConflictHandling(TestCase):
  3873. def test_bad_type(self):
  3874. self.assertRaises(ValueError, argparse.ArgumentParser,
  3875. conflict_handler='foo')
  3876. def test_conflict_error(self):
  3877. parser = argparse.ArgumentParser()
  3878. parser.add_argument('-x')
  3879. self.assertRaises(argparse.ArgumentError,
  3880. parser.add_argument, '-x')
  3881. parser.add_argument('--spam')
  3882. self.assertRaises(argparse.ArgumentError,
  3883. parser.add_argument, '--spam')
  3884. def test_resolve_error(self):
  3885. get_parser = argparse.ArgumentParser
  3886. parser = get_parser(prog='PROG', conflict_handler='resolve')
  3887. parser.add_argument('-x', help='OLD X')
  3888. parser.add_argument('-x', help='NEW X')
  3889. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  3890. usage: PROG [-h] [-x X]
  3891. options:
  3892. -h, --help show this help message and exit
  3893. -x X NEW X
  3894. '''))
  3895. parser.add_argument('--spam', metavar='OLD_SPAM')
  3896. parser.add_argument('--spam', metavar='NEW_SPAM')
  3897. self.assertEqual(parser.format_help(), textwrap.dedent('''\
  3898. usage: PROG [-h] [-x X] [--spam NEW_SPAM]
  3899. options:
  3900. -h, --help show this help message and exit
  3901. -x X NEW X
  3902. --spam NEW_SPAM
  3903. '''))
  3904. def test_subparser_conflict(self):
  3905. parser = argparse.ArgumentParser()
  3906. sp = parser.add_subparsers()
  3907. sp.add_parser('fullname', aliases=['alias'])
  3908. self.assertRaises(argparse.ArgumentError,
  3909. sp.add_parser, 'fullname')
  3910. self.assertRaises(argparse.ArgumentError,
  3911. sp.add_parser, 'alias')
  3912. self.assertRaises(argparse.ArgumentError,
  3913. sp.add_parser, 'other', aliases=['fullname'])
  3914. self.assertRaises(argparse.ArgumentError,
  3915. sp.add_parser, 'other', aliases=['alias'])
  3916. # =============================
  3917. # Help and Version option tests
  3918. # =============================
  3919. class TestOptionalsHelpVersionActions(TestCase):
  3920. """Test the help and version actions"""
  3921. def assertPrintHelpExit(self, parser, args_str):
  3922. with self.assertRaises(ArgumentParserError) as cm:
  3923. parser.parse_args(args_str.split())
  3924. self.assertEqual(parser.format_help(), cm.exception.stdout)
  3925. def assertArgumentParserError(self, parser, *args):
  3926. self.assertRaises(ArgumentParserError, parser.parse_args, args)
  3927. def test_version(self):
  3928. parser = ErrorRaisingArgumentParser()
  3929. parser.add_argument('-v', '--version', action='version', version='1.0')
  3930. self.assertPrintHelpExit(parser, '-h')
  3931. self.assertPrintHelpExit(parser, '--help')
  3932. self.assertRaises(AttributeError, getattr, parser, 'format_version')
  3933. def test_version_format(self):
  3934. parser = ErrorRaisingArgumentParser(prog='PPP')
  3935. parser.add_argument('-v', '--version', action='version', version='%(prog)s 3.5')
  3936. with self.assertRaises(ArgumentParserError) as cm:
  3937. parser.parse_args(['-v'])
  3938. self.assertEqual('PPP 3.5\n', cm.exception.stdout)
  3939. def test_version_no_help(self):
  3940. parser = ErrorRaisingArgumentParser(add_help=False)
  3941. parser.add_argument('-v', '--version', action='version', version='1.0')
  3942. self.assertArgumentParserError(parser, '-h')
  3943. self.assertArgumentParserError(parser, '--help')
  3944. self.assertRaises(AttributeError, getattr, parser, 'format_version')
  3945. def test_version_action(self):
  3946. parser = ErrorRaisingArgumentParser(prog='XXX')
  3947. parser.add_argument('-V', action='version', version='%(prog)s 3.7')
  3948. with self.assertRaises(ArgumentParserError) as cm:
  3949. parser.parse_args(['-V'])
  3950. self.assertEqual('XXX 3.7\n', cm.exception.stdout)
  3951. def test_no_help(self):
  3952. parser = ErrorRaisingArgumentParser(add_help=False)
  3953. self.assertArgumentParserError(parser, '-h')
  3954. self.assertArgumentParserError(parser, '--help')
  3955. self.assertArgumentParserError(parser, '-v')
  3956. self.assertArgumentParserError(parser, '--version')
  3957. def test_alternate_help_version(self):
  3958. parser = ErrorRaisingArgumentParser()
  3959. parser.add_argument('-x', action='help')
  3960. parser.add_argument('-y', action='version')
  3961. self.assertPrintHelpExit(parser, '-x')
  3962. self.assertArgumentParserError(parser, '-v')
  3963. self.assertArgumentParserError(parser, '--version')
  3964. self.assertRaises(AttributeError, getattr, parser, 'format_version')
  3965. def test_help_version_extra_arguments(self):
  3966. parser = ErrorRaisingArgumentParser()
  3967. parser.add_argument('--version', action='version', version='1.0')
  3968. parser.add_argument('-x', action='store_true')
  3969. parser.add_argument('y')
  3970. # try all combinations of valid prefixes and suffixes
  3971. valid_prefixes = ['', '-x', 'foo', '-x bar', 'baz -x']
  3972. valid_suffixes = valid_prefixes + ['--bad-option', 'foo bar baz']
  3973. for prefix in valid_prefixes:
  3974. for suffix in valid_suffixes:
  3975. format = '%s %%s %s' % (prefix, suffix)
  3976. self.assertPrintHelpExit(parser, format % '-h')
  3977. self.assertPrintHelpExit(parser, format % '--help')
  3978. self.assertRaises(AttributeError, getattr, parser, 'format_version')
  3979. # ======================
  3980. # str() and repr() tests
  3981. # ======================
  3982. class TestStrings(TestCase):
  3983. """Test str() and repr() on Optionals and Positionals"""
  3984. def assertStringEqual(self, obj, result_string):
  3985. for func in [str, repr]:
  3986. self.assertEqual(func(obj), result_string)
  3987. def test_optional(self):
  3988. option = argparse.Action(
  3989. option_strings=['--foo', '-a', '-b'],
  3990. dest='b',
  3991. type='int',
  3992. nargs='+',
  3993. default=42,
  3994. choices=[1, 2, 3],
  3995. required=False,
  3996. help='HELP',
  3997. metavar='METAVAR')
  3998. string = (
  3999. "Action(option_strings=['--foo', '-a', '-b'], dest='b', "
  4000. "nargs='+', const=None, default=42, type='int', "
  4001. "choices=[1, 2, 3], required=False, help='HELP', metavar='METAVAR')")
  4002. self.assertStringEqual(option, string)
  4003. def test_argument(self):
  4004. argument = argparse.Action(
  4005. option_strings=[],
  4006. dest='x',
  4007. type=float,
  4008. nargs='?',
  4009. default=2.5,
  4010. choices=[0.5, 1.5, 2.5],
  4011. required=True,
  4012. help='H HH H',
  4013. metavar='MV MV MV')
  4014. string = (
  4015. "Action(option_strings=[], dest='x', nargs='?', "
  4016. "const=None, default=2.5, type=%r, choices=[0.5, 1.5, 2.5], "
  4017. "required=True, help='H HH H', metavar='MV MV MV')" % float)
  4018. self.assertStringEqual(argument, string)
  4019. def test_namespace(self):
  4020. ns = argparse.Namespace(foo=42, bar='spam')
  4021. string = "Namespace(foo=42, bar='spam')"
  4022. self.assertStringEqual(ns, string)
  4023. def test_namespace_starkwargs_notidentifier(self):
  4024. ns = argparse.Namespace(**{'"': 'quote'})
  4025. string = """Namespace(**{'"': 'quote'})"""
  4026. self.assertStringEqual(ns, string)
  4027. def test_namespace_kwargs_and_starkwargs_notidentifier(self):
  4028. ns = argparse.Namespace(a=1, **{'"': 'quote'})
  4029. string = """Namespace(a=1, **{'"': 'quote'})"""
  4030. self.assertStringEqual(ns, string)
  4031. def test_namespace_starkwargs_identifier(self):
  4032. ns = argparse.Namespace(**{'valid': True})
  4033. string = "Namespace(valid=True)"
  4034. self.assertStringEqual(ns, string)
  4035. def test_parser(self):
  4036. parser = argparse.ArgumentParser(prog='PROG')
  4037. string = (
  4038. "ArgumentParser(prog='PROG', usage=None, description=None, "
  4039. "formatter_class=%r, conflict_handler='error', "
  4040. "add_help=True)" % argparse.HelpFormatter)
  4041. self.assertStringEqual(parser, string)
  4042. # ===============
  4043. # Namespace tests
  4044. # ===============
  4045. class TestNamespace(TestCase):
  4046. def test_constructor(self):
  4047. ns = argparse.Namespace()
  4048. self.assertRaises(AttributeError, getattr, ns, 'x')
  4049. ns = argparse.Namespace(a=42, b='spam')
  4050. self.assertEqual(ns.a, 42)
  4051. self.assertEqual(ns.b, 'spam')
  4052. def test_equality(self):
  4053. ns1 = argparse.Namespace(a=1, b=2)
  4054. ns2 = argparse.Namespace(b=2, a=1)
  4055. ns3 = argparse.Namespace(a=1)
  4056. ns4 = argparse.Namespace(b=2)
  4057. self.assertEqual(ns1, ns2)
  4058. self.assertNotEqual(ns1, ns3)
  4059. self.assertNotEqual(ns1, ns4)
  4060. self.assertNotEqual(ns2, ns3)
  4061. self.assertNotEqual(ns2, ns4)
  4062. self.assertTrue(ns1 != ns3)
  4063. self.assertTrue(ns1 != ns4)
  4064. self.assertTrue(ns2 != ns3)
  4065. self.assertTrue(ns2 != ns4)
  4066. def test_equality_returns_notimplemented(self):
  4067. # See issue 21481
  4068. ns = argparse.Namespace(a=1, b=2)
  4069. self.assertIs(ns.__eq__(None), NotImplemented)
  4070. self.assertIs(ns.__ne__(None), NotImplemented)
  4071. # ===================
  4072. # File encoding tests
  4073. # ===================
  4074. class TestEncoding(TestCase):
  4075. def _test_module_encoding(self, path):
  4076. path, _ = os.path.splitext(path)
  4077. path += ".py"
  4078. with open(path, 'r', encoding='utf-8') as f:
  4079. f.read()
  4080. def test_argparse_module_encoding(self):
  4081. self._test_module_encoding(argparse.__file__)
  4082. def test_test_argparse_module_encoding(self):
  4083. self._test_module_encoding(__file__)
  4084. # ===================
  4085. # ArgumentError tests
  4086. # ===================
  4087. class TestArgumentError(TestCase):
  4088. def test_argument_error(self):
  4089. msg = "my error here"
  4090. error = argparse.ArgumentError(None, msg)
  4091. self.assertEqual(str(error), msg)
  4092. # =======================
  4093. # ArgumentTypeError tests
  4094. # =======================
  4095. class TestArgumentTypeError(TestCase):
  4096. def test_argument_type_error(self):
  4097. def spam(string):
  4098. raise argparse.ArgumentTypeError('spam!')
  4099. parser = ErrorRaisingArgumentParser(prog='PROG', add_help=False)
  4100. parser.add_argument('x', type=spam)
  4101. with self.assertRaises(ArgumentParserError) as cm:
  4102. parser.parse_args(['XXX'])
  4103. self.assertEqual('usage: PROG x\nPROG: error: argument x: spam!\n',
  4104. cm.exception.stderr)
  4105. # =========================
  4106. # MessageContentError tests
  4107. # =========================
  4108. class TestMessageContentError(TestCase):
  4109. def test_missing_argument_name_in_message(self):
  4110. parser = ErrorRaisingArgumentParser(prog='PROG', usage='')
  4111. parser.add_argument('req_pos', type=str)
  4112. parser.add_argument('-req_opt', type=int, required=True)
  4113. parser.add_argument('need_one', type=str, nargs='+')
  4114. with self.assertRaises(ArgumentParserError) as cm:
  4115. parser.parse_args([])
  4116. msg = str(cm.exception)
  4117. self.assertRegex(msg, 'req_pos')
  4118. self.assertRegex(msg, 'req_opt')
  4119. self.assertRegex(msg, 'need_one')
  4120. with self.assertRaises(ArgumentParserError) as cm:
  4121. parser.parse_args(['myXargument'])
  4122. msg = str(cm.exception)
  4123. self.assertNotIn(msg, 'req_pos')
  4124. self.assertRegex(msg, 'req_opt')
  4125. self.assertRegex(msg, 'need_one')
  4126. with self.assertRaises(ArgumentParserError) as cm:
  4127. parser.parse_args(['myXargument', '-req_opt=1'])
  4128. msg = str(cm.exception)
  4129. self.assertNotIn(msg, 'req_pos')
  4130. self.assertNotIn(msg, 'req_opt')
  4131. self.assertRegex(msg, 'need_one')
  4132. def test_optional_optional_not_in_message(self):
  4133. parser = ErrorRaisingArgumentParser(prog='PROG', usage='')
  4134. parser.add_argument('req_pos', type=str)
  4135. parser.add_argument('--req_opt', type=int, required=True)
  4136. parser.add_argument('--opt_opt', type=bool, nargs='?',
  4137. default=True)
  4138. with self.assertRaises(ArgumentParserError) as cm:
  4139. parser.parse_args([])
  4140. msg = str(cm.exception)
  4141. self.assertRegex(msg, 'req_pos')
  4142. self.assertRegex(msg, 'req_opt')
  4143. self.assertNotIn(msg, 'opt_opt')
  4144. with self.assertRaises(ArgumentParserError) as cm:
  4145. parser.parse_args(['--req_opt=1'])
  4146. msg = str(cm.exception)
  4147. self.assertRegex(msg, 'req_pos')
  4148. self.assertNotIn(msg, 'req_opt')
  4149. self.assertNotIn(msg, 'opt_opt')
  4150. def test_optional_positional_not_in_message(self):
  4151. parser = ErrorRaisingArgumentParser(prog='PROG', usage='')
  4152. parser.add_argument('req_pos')
  4153. parser.add_argument('optional_positional', nargs='?', default='eggs')
  4154. with self.assertRaises(ArgumentParserError) as cm:
  4155. parser.parse_args([])
  4156. msg = str(cm.exception)
  4157. self.assertRegex(msg, 'req_pos')
  4158. self.assertNotIn(msg, 'optional_positional')
  4159. # ================================================
  4160. # Check that the type function is called only once
  4161. # ================================================
  4162. class TestTypeFunctionCallOnlyOnce(TestCase):
  4163. def test_type_function_call_only_once(self):
  4164. def spam(string_to_convert):
  4165. self.assertEqual(string_to_convert, 'spam!')
  4166. return 'foo_converted'
  4167. parser = argparse.ArgumentParser()
  4168. parser.add_argument('--foo', type=spam, default='bar')
  4169. args = parser.parse_args('--foo spam!'.split())
  4170. self.assertEqual(NS(foo='foo_converted'), args)
  4171. # ==================================================================
  4172. # Check semantics regarding the default argument and type conversion
  4173. # ==================================================================
  4174. class TestTypeFunctionCalledOnDefault(TestCase):
  4175. def test_type_function_call_with_non_string_default(self):
  4176. def spam(int_to_convert):
  4177. self.assertEqual(int_to_convert, 0)
  4178. return 'foo_converted'
  4179. parser = argparse.ArgumentParser()
  4180. parser.add_argument('--foo', type=spam, default=0)
  4181. args = parser.parse_args([])
  4182. # foo should *not* be converted because its default is not a string.
  4183. self.assertEqual(NS(foo=0), args)
  4184. def test_type_function_call_with_string_default(self):
  4185. def spam(int_to_convert):
  4186. return 'foo_converted'
  4187. parser = argparse.ArgumentParser()
  4188. parser.add_argument('--foo', type=spam, default='0')
  4189. args = parser.parse_args([])
  4190. # foo is converted because its default is a string.
  4191. self.assertEqual(NS(foo='foo_converted'), args)
  4192. def test_no_double_type_conversion_of_default(self):
  4193. def extend(str_to_convert):
  4194. return str_to_convert + '*'
  4195. parser = argparse.ArgumentParser()
  4196. parser.add_argument('--test', type=extend, default='*')
  4197. args = parser.parse_args([])
  4198. # The test argument will be two stars, one coming from the default
  4199. # value and one coming from the type conversion being called exactly
  4200. # once.
  4201. self.assertEqual(NS(test='**'), args)
  4202. def test_issue_15906(self):
  4203. # Issue #15906: When action='append', type=str, default=[] are
  4204. # providing, the dest value was the string representation "[]" when it
  4205. # should have been an empty list.
  4206. parser = argparse.ArgumentParser()
  4207. parser.add_argument('--test', dest='test', type=str,
  4208. default=[], action='append')
  4209. args = parser.parse_args([])
  4210. self.assertEqual(args.test, [])
  4211. # ======================
  4212. # parse_known_args tests
  4213. # ======================
  4214. class TestParseKnownArgs(TestCase):
  4215. def test_arguments_tuple(self):
  4216. parser = argparse.ArgumentParser()
  4217. parser.parse_args(())
  4218. def test_arguments_list(self):
  4219. parser = argparse.ArgumentParser()
  4220. parser.parse_args([])
  4221. def test_arguments_tuple_positional(self):
  4222. parser = argparse.ArgumentParser()
  4223. parser.add_argument('x')
  4224. parser.parse_args(('x',))
  4225. def test_arguments_list_positional(self):
  4226. parser = argparse.ArgumentParser()
  4227. parser.add_argument('x')
  4228. parser.parse_args(['x'])
  4229. def test_optionals(self):
  4230. parser = argparse.ArgumentParser()
  4231. parser.add_argument('--foo')
  4232. args, extras = parser.parse_known_args('--foo F --bar --baz'.split())
  4233. self.assertEqual(NS(foo='F'), args)
  4234. self.assertEqual(['--bar', '--baz'], extras)
  4235. def test_mixed(self):
  4236. parser = argparse.ArgumentParser()
  4237. parser.add_argument('-v', nargs='?', const=1, type=int)
  4238. parser.add_argument('--spam', action='store_false')
  4239. parser.add_argument('badger')
  4240. argv = ["B", "C", "--foo", "-v", "3", "4"]
  4241. args, extras = parser.parse_known_args(argv)
  4242. self.assertEqual(NS(v=3, spam=True, badger="B"), args)
  4243. self.assertEqual(["C", "--foo", "4"], extras)
  4244. # ===========================
  4245. # parse_intermixed_args tests
  4246. # ===========================
  4247. class TestIntermixedArgs(TestCase):
  4248. def test_basic(self):
  4249. # test parsing intermixed optionals and positionals
  4250. parser = argparse.ArgumentParser(prog='PROG')
  4251. parser.add_argument('--foo', dest='foo')
  4252. bar = parser.add_argument('--bar', dest='bar', required=True)
  4253. parser.add_argument('cmd')
  4254. parser.add_argument('rest', nargs='*', type=int)
  4255. argv = 'cmd --foo x 1 --bar y 2 3'.split()
  4256. args = parser.parse_intermixed_args(argv)
  4257. # rest gets [1,2,3] despite the foo and bar strings
  4258. self.assertEqual(NS(bar='y', cmd='cmd', foo='x', rest=[1, 2, 3]), args)
  4259. args, extras = parser.parse_known_args(argv)
  4260. # cannot parse the '1,2,3'
  4261. self.assertEqual(NS(bar='y', cmd='cmd', foo='x', rest=[]), args)
  4262. self.assertEqual(["1", "2", "3"], extras)
  4263. argv = 'cmd --foo x 1 --error 2 --bar y 3'.split()
  4264. args, extras = parser.parse_known_intermixed_args(argv)
  4265. # unknown optionals go into extras
  4266. self.assertEqual(NS(bar='y', cmd='cmd', foo='x', rest=[1]), args)
  4267. self.assertEqual(['--error', '2', '3'], extras)
  4268. # restores attributes that were temporarily changed
  4269. self.assertIsNone(parser.usage)
  4270. self.assertEqual(bar.required, True)
  4271. def test_remainder(self):
  4272. # Intermixed and remainder are incompatible
  4273. parser = ErrorRaisingArgumentParser(prog='PROG')
  4274. parser.add_argument('-z')
  4275. parser.add_argument('x')
  4276. parser.add_argument('y', nargs='...')
  4277. argv = 'X A B -z Z'.split()
  4278. # intermixed fails with '...' (also 'A...')
  4279. # self.assertRaises(TypeError, parser.parse_intermixed_args, argv)
  4280. with self.assertRaises(TypeError) as cm:
  4281. parser.parse_intermixed_args(argv)
  4282. self.assertRegex(str(cm.exception), r'\.\.\.')
  4283. def test_exclusive(self):
  4284. # mutually exclusive group; intermixed works fine
  4285. parser = ErrorRaisingArgumentParser(prog='PROG')
  4286. group = parser.add_mutually_exclusive_group(required=True)
  4287. group.add_argument('--foo', action='store_true', help='FOO')
  4288. group.add_argument('--spam', help='SPAM')
  4289. parser.add_argument('badger', nargs='*', default='X', help='BADGER')
  4290. args = parser.parse_intermixed_args('1 --foo 2'.split())
  4291. self.assertEqual(NS(badger=['1', '2'], foo=True, spam=None), args)
  4292. self.assertRaises(ArgumentParserError, parser.parse_intermixed_args, '1 2'.split())
  4293. self.assertEqual(group.required, True)
  4294. def test_exclusive_incompatible(self):
  4295. # mutually exclusive group including positional - fail
  4296. parser = ErrorRaisingArgumentParser(prog='PROG')
  4297. group = parser.add_mutually_exclusive_group(required=True)
  4298. group.add_argument('--foo', action='store_true', help='FOO')
  4299. group.add_argument('--spam', help='SPAM')
  4300. group.add_argument('badger', nargs='*', default='X', help='BADGER')
  4301. self.assertRaises(TypeError, parser.parse_intermixed_args, [])
  4302. self.assertEqual(group.required, True)
  4303. class TestIntermixedMessageContentError(TestCase):
  4304. # case where Intermixed gives different error message
  4305. # error is raised by 1st parsing step
  4306. def test_missing_argument_name_in_message(self):
  4307. parser = ErrorRaisingArgumentParser(prog='PROG', usage='')
  4308. parser.add_argument('req_pos', type=str)
  4309. parser.add_argument('-req_opt', type=int, required=True)
  4310. with self.assertRaises(ArgumentParserError) as cm:
  4311. parser.parse_args([])
  4312. msg = str(cm.exception)
  4313. self.assertRegex(msg, 'req_pos')
  4314. self.assertRegex(msg, 'req_opt')
  4315. with self.assertRaises(ArgumentParserError) as cm:
  4316. parser.parse_intermixed_args([])
  4317. msg = str(cm.exception)
  4318. self.assertNotRegex(msg, 'req_pos')
  4319. self.assertRegex(msg, 'req_opt')
  4320. # ==========================
  4321. # add_argument metavar tests
  4322. # ==========================
  4323. class TestAddArgumentMetavar(TestCase):
  4324. EXPECTED_MESSAGE = "length of metavar tuple does not match nargs"
  4325. def do_test_no_exception(self, nargs, metavar):
  4326. parser = argparse.ArgumentParser()
  4327. parser.add_argument("--foo", nargs=nargs, metavar=metavar)
  4328. def do_test_exception(self, nargs, metavar):
  4329. parser = argparse.ArgumentParser()
  4330. with self.assertRaises(ValueError) as cm:
  4331. parser.add_argument("--foo", nargs=nargs, metavar=metavar)
  4332. self.assertEqual(cm.exception.args[0], self.EXPECTED_MESSAGE)
  4333. # Unit tests for different values of metavar when nargs=None
  4334. def test_nargs_None_metavar_string(self):
  4335. self.do_test_no_exception(nargs=None, metavar="1")
  4336. def test_nargs_None_metavar_length0(self):
  4337. self.do_test_exception(nargs=None, metavar=tuple())
  4338. def test_nargs_None_metavar_length1(self):
  4339. self.do_test_no_exception(nargs=None, metavar=("1",))
  4340. def test_nargs_None_metavar_length2(self):
  4341. self.do_test_exception(nargs=None, metavar=("1", "2"))
  4342. def test_nargs_None_metavar_length3(self):
  4343. self.do_test_exception(nargs=None, metavar=("1", "2", "3"))
  4344. # Unit tests for different values of metavar when nargs=?
  4345. def test_nargs_optional_metavar_string(self):
  4346. self.do_test_no_exception(nargs="?", metavar="1")
  4347. def test_nargs_optional_metavar_length0(self):
  4348. self.do_test_exception(nargs="?", metavar=tuple())
  4349. def test_nargs_optional_metavar_length1(self):
  4350. self.do_test_no_exception(nargs="?", metavar=("1",))
  4351. def test_nargs_optional_metavar_length2(self):
  4352. self.do_test_exception(nargs="?", metavar=("1", "2"))
  4353. def test_nargs_optional_metavar_length3(self):
  4354. self.do_test_exception(nargs="?", metavar=("1", "2", "3"))
  4355. # Unit tests for different values of metavar when nargs=*
  4356. def test_nargs_zeroormore_metavar_string(self):
  4357. self.do_test_no_exception(nargs="*", metavar="1")
  4358. def test_nargs_zeroormore_metavar_length0(self):
  4359. self.do_test_exception(nargs="*", metavar=tuple())
  4360. def test_nargs_zeroormore_metavar_length1(self):
  4361. self.do_test_no_exception(nargs="*", metavar=("1",))
  4362. def test_nargs_zeroormore_metavar_length2(self):
  4363. self.do_test_no_exception(nargs="*", metavar=("1", "2"))
  4364. def test_nargs_zeroormore_metavar_length3(self):
  4365. self.do_test_exception(nargs="*", metavar=("1", "2", "3"))
  4366. # Unit tests for different values of metavar when nargs=+
  4367. def test_nargs_oneormore_metavar_string(self):
  4368. self.do_test_no_exception(nargs="+", metavar="1")
  4369. def test_nargs_oneormore_metavar_length0(self):
  4370. self.do_test_exception(nargs="+", metavar=tuple())
  4371. def test_nargs_oneormore_metavar_length1(self):
  4372. self.do_test_exception(nargs="+", metavar=("1",))
  4373. def test_nargs_oneormore_metavar_length2(self):
  4374. self.do_test_no_exception(nargs="+", metavar=("1", "2"))
  4375. def test_nargs_oneormore_metavar_length3(self):
  4376. self.do_test_exception(nargs="+", metavar=("1", "2", "3"))
  4377. # Unit tests for different values of metavar when nargs=...
  4378. def test_nargs_remainder_metavar_string(self):
  4379. self.do_test_no_exception(nargs="...", metavar="1")
  4380. def test_nargs_remainder_metavar_length0(self):
  4381. self.do_test_no_exception(nargs="...", metavar=tuple())
  4382. def test_nargs_remainder_metavar_length1(self):
  4383. self.do_test_no_exception(nargs="...", metavar=("1",))
  4384. def test_nargs_remainder_metavar_length2(self):
  4385. self.do_test_no_exception(nargs="...", metavar=("1", "2"))
  4386. def test_nargs_remainder_metavar_length3(self):
  4387. self.do_test_no_exception(nargs="...", metavar=("1", "2", "3"))
  4388. # Unit tests for different values of metavar when nargs=A...
  4389. def test_nargs_parser_metavar_string(self):
  4390. self.do_test_no_exception(nargs="A...", metavar="1")
  4391. def test_nargs_parser_metavar_length0(self):
  4392. self.do_test_exception(nargs="A...", metavar=tuple())
  4393. def test_nargs_parser_metavar_length1(self):
  4394. self.do_test_no_exception(nargs="A...", metavar=("1",))
  4395. def test_nargs_parser_metavar_length2(self):
  4396. self.do_test_exception(nargs="A...", metavar=("1", "2"))
  4397. def test_nargs_parser_metavar_length3(self):
  4398. self.do_test_exception(nargs="A...", metavar=("1", "2", "3"))
  4399. # Unit tests for different values of metavar when nargs=1
  4400. def test_nargs_1_metavar_string(self):
  4401. self.do_test_no_exception(nargs=1, metavar="1")
  4402. def test_nargs_1_metavar_length0(self):
  4403. self.do_test_exception(nargs=1, metavar=tuple())
  4404. def test_nargs_1_metavar_length1(self):
  4405. self.do_test_no_exception(nargs=1, metavar=("1",))
  4406. def test_nargs_1_metavar_length2(self):
  4407. self.do_test_exception(nargs=1, metavar=("1", "2"))
  4408. def test_nargs_1_metavar_length3(self):
  4409. self.do_test_exception(nargs=1, metavar=("1", "2", "3"))
  4410. # Unit tests for different values of metavar when nargs=2
  4411. def test_nargs_2_metavar_string(self):
  4412. self.do_test_no_exception(nargs=2, metavar="1")
  4413. def test_nargs_2_metavar_length0(self):
  4414. self.do_test_exception(nargs=2, metavar=tuple())
  4415. def test_nargs_2_metavar_length1(self):
  4416. self.do_test_exception(nargs=2, metavar=("1",))
  4417. def test_nargs_2_metavar_length2(self):
  4418. self.do_test_no_exception(nargs=2, metavar=("1", "2"))
  4419. def test_nargs_2_metavar_length3(self):
  4420. self.do_test_exception(nargs=2, metavar=("1", "2", "3"))
  4421. # Unit tests for different values of metavar when nargs=3
  4422. def test_nargs_3_metavar_string(self):
  4423. self.do_test_no_exception(nargs=3, metavar="1")
  4424. def test_nargs_3_metavar_length0(self):
  4425. self.do_test_exception(nargs=3, metavar=tuple())
  4426. def test_nargs_3_metavar_length1(self):
  4427. self.do_test_exception(nargs=3, metavar=("1",))
  4428. def test_nargs_3_metavar_length2(self):
  4429. self.do_test_exception(nargs=3, metavar=("1", "2"))
  4430. def test_nargs_3_metavar_length3(self):
  4431. self.do_test_no_exception(nargs=3, metavar=("1", "2", "3"))
  4432. class TestInvalidNargs(TestCase):
  4433. EXPECTED_INVALID_MESSAGE = "invalid nargs value"
  4434. EXPECTED_RANGE_MESSAGE = ("nargs for store actions must be != 0; if you "
  4435. "have nothing to store, actions such as store "
  4436. "true or store const may be more appropriate")
  4437. def do_test_range_exception(self, nargs):
  4438. parser = argparse.ArgumentParser()
  4439. with self.assertRaises(ValueError) as cm:
  4440. parser.add_argument("--foo", nargs=nargs)
  4441. self.assertEqual(cm.exception.args[0], self.EXPECTED_RANGE_MESSAGE)
  4442. def do_test_invalid_exception(self, nargs):
  4443. parser = argparse.ArgumentParser()
  4444. with self.assertRaises(ValueError) as cm:
  4445. parser.add_argument("--foo", nargs=nargs)
  4446. self.assertEqual(cm.exception.args[0], self.EXPECTED_INVALID_MESSAGE)
  4447. # Unit tests for different values of nargs
  4448. def test_nargs_alphabetic(self):
  4449. self.do_test_invalid_exception(nargs='a')
  4450. self.do_test_invalid_exception(nargs="abcd")
  4451. def test_nargs_zero(self):
  4452. self.do_test_range_exception(nargs=0)
  4453. # ============================
  4454. # from argparse import * tests
  4455. # ============================
  4456. class TestImportStar(TestCase):
  4457. def test(self):
  4458. for name in argparse.__all__:
  4459. self.assertTrue(hasattr(argparse, name))
  4460. def test_all_exports_everything_but_modules(self):
  4461. items = [
  4462. name
  4463. for name, value in vars(argparse).items()
  4464. if not (name.startswith("_") or name == 'ngettext')
  4465. if not inspect.ismodule(value)
  4466. ]
  4467. self.assertEqual(sorted(items), sorted(argparse.__all__))
  4468. class TestWrappingMetavar(TestCase):
  4469. def setUp(self):
  4470. super().setUp()
  4471. self.parser = ErrorRaisingArgumentParser(
  4472. 'this_is_spammy_prog_with_a_long_name_sorry_about_the_name'
  4473. )
  4474. # this metavar was triggering library assertion errors due to usage
  4475. # message formatting incorrectly splitting on the ] chars within
  4476. metavar = '<http[s]://example:1234>'
  4477. self.parser.add_argument('--proxy', metavar=metavar)
  4478. def test_help_with_metavar(self):
  4479. help_text = self.parser.format_help()
  4480. self.assertEqual(help_text, textwrap.dedent('''\
  4481. usage: this_is_spammy_prog_with_a_long_name_sorry_about_the_name
  4482. [-h] [--proxy <http[s]://example:1234>]
  4483. options:
  4484. -h, --help show this help message and exit
  4485. --proxy <http[s]://example:1234>
  4486. '''))
  4487. class TestExitOnError(TestCase):
  4488. def setUp(self):
  4489. self.parser = argparse.ArgumentParser(exit_on_error=False)
  4490. self.parser.add_argument('--integers', metavar='N', type=int)
  4491. def test_exit_on_error_with_good_args(self):
  4492. ns = self.parser.parse_args('--integers 4'.split())
  4493. self.assertEqual(ns, argparse.Namespace(integers=4))
  4494. def test_exit_on_error_with_bad_args(self):
  4495. with self.assertRaises(argparse.ArgumentError):
  4496. self.parser.parse_args('--integers a'.split())
  4497. def tearDownModule():
  4498. # Remove global references to avoid looking like we have refleaks.
  4499. RFile.seen = {}
  4500. WFile.seen = set()
  4501. if __name__ == '__main__':
  4502. unittest.main()