| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993499449954996499749984999500050015002500350045005500650075008500950105011501250135014501550165017501850195020502150225023502450255026502750285029503050315032503350345035503650375038503950405041504250435044504550465047504850495050505150525053505450555056505750585059506050615062506350645065506650675068506950705071507250735074507550765077507850795080508150825083508450855086508750885089509050915092509350945095509650975098509951005101510251035104510551065107510851095110511151125113511451155116511751185119512051215122512351245125512651275128512951305131513251335134513551365137513851395140514151425143514451455146514751485149515051515152515351545155515651575158515951605161516251635164516551665167516851695170517151725173517451755176517751785179518051815182518351845185518651875188518951905191519251935194519551965197519851995200520152025203520452055206520752085209521052115212521352145215521652175218521952205221522252235224522552265227522852295230523152325233523452355236523752385239524052415242524352445245524652475248524952505251525252535254525552565257525852595260526152625263526452655266526752685269527052715272527352745275527652775278527952805281528252835284528552865287528852895290529152925293529452955296529752985299530053015302530353045305530653075308530953105311531253135314531553165317531853195320532153225323532453255326532753285329533053315332533353345335533653375338533953405341534253435344534553465347534853495350535153525353535453555356535753585359536053615362536353645365536653675368536953705371537253735374537553765377537853795380538153825383538453855386538753885389539053915392539353945395539653975398539954005401540254035404540554065407540854095410541154125413541454155416541754185419542054215422542354245425542654275428542954305431543254335434543554365437543854395440544154425443544454455446544754485449545054515452545354545455545654575458545954605461546254635464546554665467546854695470547154725473547454755476547754785479548054815482548354845485548654875488548954905491549254935494549554965497549854995500550155025503550455055506550755085509551055115512551355145515551655175518551955205521552255235524552555265527552855295530553155325533553455355536553755385539554055415542554355445545554655475548554955505551555255535554555555565557555855595560556155625563556455655566556755685569557055715572557355745575 |
- # Author: Steven J. Bethard <steven.bethard@gmail.com>.
- import inspect
- import io
- import operator
- import os
- import shutil
- import stat
- import sys
- import textwrap
- import tempfile
- import unittest
- import argparse
- import warnings
- from test.support import os_helper
- from unittest import mock
- class StdIOBuffer(io.TextIOWrapper):
- '''Replacement for writable io.StringIO that behaves more like real file
- Unlike StringIO, provides a buffer attribute that holds the underlying
- binary data, allowing it to replace sys.stdout/sys.stderr in more
- contexts.
- '''
- def __init__(self, initial_value='', newline='\n'):
- initial_value = initial_value.encode('utf-8')
- super().__init__(io.BufferedWriter(io.BytesIO(initial_value)),
- 'utf-8', newline=newline)
- def getvalue(self):
- self.flush()
- return self.buffer.raw.getvalue().decode('utf-8')
- class TestCase(unittest.TestCase):
- def setUp(self):
- # The tests assume that line wrapping occurs at 80 columns, but this
- # behaviour can be overridden by setting the COLUMNS environment
- # variable. To ensure that this width is used, set COLUMNS to 80.
- env = self.enterContext(os_helper.EnvironmentVarGuard())
- env['COLUMNS'] = '80'
- @os_helper.skip_unless_working_chmod
- class TempDirMixin(object):
- def setUp(self):
- self.temp_dir = tempfile.mkdtemp()
- self.old_dir = os.getcwd()
- os.chdir(self.temp_dir)
- def tearDown(self):
- os.chdir(self.old_dir)
- for root, dirs, files in os.walk(self.temp_dir, topdown=False):
- for name in files:
- os.chmod(os.path.join(self.temp_dir, name), stat.S_IWRITE)
- shutil.rmtree(self.temp_dir, True)
- def create_writable_file(self, filename):
- file_path = os.path.join(self.temp_dir, filename)
- with open(file_path, 'w', encoding="utf-8") as file:
- file.write(filename)
- return file_path
- def create_readonly_file(self, filename):
- os.chmod(self.create_writable_file(filename), stat.S_IREAD)
- class Sig(object):
- def __init__(self, *args, **kwargs):
- self.args = args
- self.kwargs = kwargs
- class NS(object):
- def __init__(self, **kwargs):
- self.__dict__.update(kwargs)
- def __repr__(self):
- sorted_items = sorted(self.__dict__.items())
- kwarg_str = ', '.join(['%s=%r' % tup for tup in sorted_items])
- return '%s(%s)' % (type(self).__name__, kwarg_str)
- def __eq__(self, other):
- return vars(self) == vars(other)
- class ArgumentParserError(Exception):
- def __init__(self, message, stdout=None, stderr=None, error_code=None):
- Exception.__init__(self, message, stdout, stderr)
- self.message = message
- self.stdout = stdout
- self.stderr = stderr
- self.error_code = error_code
- def stderr_to_parser_error(parse_args, *args, **kwargs):
- # if this is being called recursively and stderr or stdout is already being
- # redirected, simply call the function and let the enclosing function
- # catch the exception
- if isinstance(sys.stderr, StdIOBuffer) or isinstance(sys.stdout, StdIOBuffer):
- return parse_args(*args, **kwargs)
- # if this is not being called recursively, redirect stderr and
- # use it as the ArgumentParserError message
- old_stdout = sys.stdout
- old_stderr = sys.stderr
- sys.stdout = StdIOBuffer()
- sys.stderr = StdIOBuffer()
- try:
- try:
- result = parse_args(*args, **kwargs)
- for key in list(vars(result)):
- attr = getattr(result, key)
- if attr is sys.stdout:
- setattr(result, key, old_stdout)
- elif attr is sys.stdout.buffer:
- setattr(result, key, getattr(old_stdout, 'buffer', BIN_STDOUT_SENTINEL))
- elif attr is sys.stderr:
- setattr(result, key, old_stderr)
- elif attr is sys.stderr.buffer:
- setattr(result, key, getattr(old_stderr, 'buffer', BIN_STDERR_SENTINEL))
- return result
- except SystemExit as e:
- code = e.code
- stdout = sys.stdout.getvalue()
- stderr = sys.stderr.getvalue()
- raise ArgumentParserError(
- "SystemExit", stdout, stderr, code) from None
- finally:
- sys.stdout = old_stdout
- sys.stderr = old_stderr
- class ErrorRaisingArgumentParser(argparse.ArgumentParser):
- def parse_args(self, *args, **kwargs):
- parse_args = super(ErrorRaisingArgumentParser, self).parse_args
- return stderr_to_parser_error(parse_args, *args, **kwargs)
- def exit(self, *args, **kwargs):
- exit = super(ErrorRaisingArgumentParser, self).exit
- return stderr_to_parser_error(exit, *args, **kwargs)
- def error(self, *args, **kwargs):
- error = super(ErrorRaisingArgumentParser, self).error
- return stderr_to_parser_error(error, *args, **kwargs)
- class ParserTesterMetaclass(type):
- """Adds parser tests using the class attributes.
- Classes of this type should specify the following attributes:
- argument_signatures -- a list of Sig objects which specify
- the signatures of Argument objects to be created
- failures -- a list of args lists that should cause the parser
- to fail
- successes -- a list of (initial_args, options, remaining_args) tuples
- where initial_args specifies the string args to be parsed,
- options is a dict that should match the vars() of the options
- parsed out of initial_args, and remaining_args should be any
- remaining unparsed arguments
- """
- def __init__(cls, name, bases, bodydict):
- if name == 'ParserTestCase':
- return
- # default parser signature is empty
- if not hasattr(cls, 'parser_signature'):
- cls.parser_signature = Sig()
- if not hasattr(cls, 'parser_class'):
- cls.parser_class = ErrorRaisingArgumentParser
- # ---------------------------------------
- # functions for adding optional arguments
- # ---------------------------------------
- def no_groups(parser, argument_signatures):
- """Add all arguments directly to the parser"""
- for sig in argument_signatures:
- parser.add_argument(*sig.args, **sig.kwargs)
- def one_group(parser, argument_signatures):
- """Add all arguments under a single group in the parser"""
- group = parser.add_argument_group('foo')
- for sig in argument_signatures:
- group.add_argument(*sig.args, **sig.kwargs)
- def many_groups(parser, argument_signatures):
- """Add each argument in its own group to the parser"""
- for i, sig in enumerate(argument_signatures):
- group = parser.add_argument_group('foo:%i' % i)
- group.add_argument(*sig.args, **sig.kwargs)
- # --------------------------
- # functions for parsing args
- # --------------------------
- def listargs(parser, args):
- """Parse the args by passing in a list"""
- return parser.parse_args(args)
- def sysargs(parser, args):
- """Parse the args by defaulting to sys.argv"""
- old_sys_argv = sys.argv
- sys.argv = [old_sys_argv[0]] + args
- try:
- return parser.parse_args()
- finally:
- sys.argv = old_sys_argv
- # class that holds the combination of one optional argument
- # addition method and one arg parsing method
- class AddTests(object):
- def __init__(self, tester_cls, add_arguments, parse_args):
- self._add_arguments = add_arguments
- self._parse_args = parse_args
- add_arguments_name = self._add_arguments.__name__
- parse_args_name = self._parse_args.__name__
- for test_func in [self.test_failures, self.test_successes]:
- func_name = test_func.__name__
- names = func_name, add_arguments_name, parse_args_name
- test_name = '_'.join(names)
- def wrapper(self, test_func=test_func):
- test_func(self)
- try:
- wrapper.__name__ = test_name
- except TypeError:
- pass
- setattr(tester_cls, test_name, wrapper)
- def _get_parser(self, tester):
- args = tester.parser_signature.args
- kwargs = tester.parser_signature.kwargs
- parser = tester.parser_class(*args, **kwargs)
- self._add_arguments(parser, tester.argument_signatures)
- return parser
- def test_failures(self, tester):
- parser = self._get_parser(tester)
- for args_str in tester.failures:
- args = args_str.split()
- with tester.assertRaises(ArgumentParserError, msg=args):
- parser.parse_args(args)
- def test_successes(self, tester):
- parser = self._get_parser(tester)
- for args, expected_ns in tester.successes:
- if isinstance(args, str):
- args = args.split()
- result_ns = self._parse_args(parser, args)
- tester.assertEqual(expected_ns, result_ns)
- # add tests for each combination of an optionals adding method
- # and an arg parsing method
- for add_arguments in [no_groups, one_group, many_groups]:
- for parse_args in [listargs, sysargs]:
- AddTests(cls, add_arguments, parse_args)
- bases = TestCase,
- ParserTestCase = ParserTesterMetaclass('ParserTestCase', bases, {})
- # ===============
- # Optionals tests
- # ===============
- class TestOptionalsSingleDash(ParserTestCase):
- """Test an Optional with a single-dash option string"""
- argument_signatures = [Sig('-x')]
- failures = ['-x', 'a', '--foo', '-x --foo', '-x -y']
- successes = [
- ('', NS(x=None)),
- ('-x a', NS(x='a')),
- ('-xa', NS(x='a')),
- ('-x -1', NS(x='-1')),
- ('-x-1', NS(x='-1')),
- ]
- class TestOptionalsSingleDashCombined(ParserTestCase):
- """Test an Optional with a single-dash option string"""
- argument_signatures = [
- Sig('-x', action='store_true'),
- Sig('-yyy', action='store_const', const=42),
- Sig('-z'),
- ]
- failures = ['a', '--foo', '-xa', '-x --foo', '-x -z', '-z -x',
- '-yx', '-yz a', '-yyyx', '-yyyza', '-xyza', '-x=']
- successes = [
- ('', NS(x=False, yyy=None, z=None)),
- ('-x', NS(x=True, yyy=None, z=None)),
- ('-za', NS(x=False, yyy=None, z='a')),
- ('-z a', NS(x=False, yyy=None, z='a')),
- ('-xza', NS(x=True, yyy=None, z='a')),
- ('-xz a', NS(x=True, yyy=None, z='a')),
- ('-x -za', NS(x=True, yyy=None, z='a')),
- ('-x -z a', NS(x=True, yyy=None, z='a')),
- ('-y', NS(x=False, yyy=42, z=None)),
- ('-yyy', NS(x=False, yyy=42, z=None)),
- ('-x -yyy -za', NS(x=True, yyy=42, z='a')),
- ('-x -yyy -z a', NS(x=True, yyy=42, z='a')),
- ]
- class TestOptionalsSingleDashLong(ParserTestCase):
- """Test an Optional with a multi-character single-dash option string"""
- argument_signatures = [Sig('-foo')]
- failures = ['-foo', 'a', '--foo', '-foo --foo', '-foo -y', '-fooa']
- successes = [
- ('', NS(foo=None)),
- ('-foo a', NS(foo='a')),
- ('-foo -1', NS(foo='-1')),
- ('-fo a', NS(foo='a')),
- ('-f a', NS(foo='a')),
- ]
- class TestOptionalsSingleDashSubsetAmbiguous(ParserTestCase):
- """Test Optionals where option strings are subsets of each other"""
- argument_signatures = [Sig('-f'), Sig('-foobar'), Sig('-foorab')]
- failures = ['-f', '-foo', '-fo', '-foo b', '-foob', '-fooba', '-foora']
- successes = [
- ('', NS(f=None, foobar=None, foorab=None)),
- ('-f a', NS(f='a', foobar=None, foorab=None)),
- ('-fa', NS(f='a', foobar=None, foorab=None)),
- ('-foa', NS(f='oa', foobar=None, foorab=None)),
- ('-fooa', NS(f='ooa', foobar=None, foorab=None)),
- ('-foobar a', NS(f=None, foobar='a', foorab=None)),
- ('-foorab a', NS(f=None, foobar=None, foorab='a')),
- ]
- class TestOptionalsSingleDashAmbiguous(ParserTestCase):
- """Test Optionals that partially match but are not subsets"""
- argument_signatures = [Sig('-foobar'), Sig('-foorab')]
- failures = ['-f', '-f a', '-fa', '-foa', '-foo', '-fo', '-foo b']
- successes = [
- ('', NS(foobar=None, foorab=None)),
- ('-foob a', NS(foobar='a', foorab=None)),
- ('-foor a', NS(foobar=None, foorab='a')),
- ('-fooba a', NS(foobar='a', foorab=None)),
- ('-foora a', NS(foobar=None, foorab='a')),
- ('-foobar a', NS(foobar='a', foorab=None)),
- ('-foorab a', NS(foobar=None, foorab='a')),
- ]
- class TestOptionalsNumeric(ParserTestCase):
- """Test an Optional with a short opt string"""
- argument_signatures = [Sig('-1', dest='one')]
- failures = ['-1', 'a', '-1 --foo', '-1 -y', '-1 -1', '-1 -2']
- successes = [
- ('', NS(one=None)),
- ('-1 a', NS(one='a')),
- ('-1a', NS(one='a')),
- ('-1-2', NS(one='-2')),
- ]
- class TestOptionalsDoubleDash(ParserTestCase):
- """Test an Optional with a double-dash option string"""
- argument_signatures = [Sig('--foo')]
- failures = ['--foo', '-f', '-f a', 'a', '--foo -x', '--foo --bar']
- successes = [
- ('', NS(foo=None)),
- ('--foo a', NS(foo='a')),
- ('--foo=a', NS(foo='a')),
- ('--foo -2.5', NS(foo='-2.5')),
- ('--foo=-2.5', NS(foo='-2.5')),
- ]
- class TestOptionalsDoubleDashPartialMatch(ParserTestCase):
- """Tests partial matching with a double-dash option string"""
- argument_signatures = [
- Sig('--badger', action='store_true'),
- Sig('--bat'),
- ]
- failures = ['--bar', '--b', '--ba', '--b=2', '--ba=4', '--badge 5']
- successes = [
- ('', NS(badger=False, bat=None)),
- ('--bat X', NS(badger=False, bat='X')),
- ('--bad', NS(badger=True, bat=None)),
- ('--badg', NS(badger=True, bat=None)),
- ('--badge', NS(badger=True, bat=None)),
- ('--badger', NS(badger=True, bat=None)),
- ]
- class TestOptionalsDoubleDashPrefixMatch(ParserTestCase):
- """Tests when one double-dash option string is a prefix of another"""
- argument_signatures = [
- Sig('--badger', action='store_true'),
- Sig('--ba'),
- ]
- failures = ['--bar', '--b', '--ba', '--b=2', '--badge 5']
- successes = [
- ('', NS(badger=False, ba=None)),
- ('--ba X', NS(badger=False, ba='X')),
- ('--ba=X', NS(badger=False, ba='X')),
- ('--bad', NS(badger=True, ba=None)),
- ('--badg', NS(badger=True, ba=None)),
- ('--badge', NS(badger=True, ba=None)),
- ('--badger', NS(badger=True, ba=None)),
- ]
- class TestOptionalsSingleDoubleDash(ParserTestCase):
- """Test an Optional with single- and double-dash option strings"""
- argument_signatures = [
- Sig('-f', action='store_true'),
- Sig('--bar'),
- Sig('-baz', action='store_const', const=42),
- ]
- failures = ['--bar', '-fbar', '-fbaz', '-bazf', '-b B', 'B']
- successes = [
- ('', NS(f=False, bar=None, baz=None)),
- ('-f', NS(f=True, bar=None, baz=None)),
- ('--ba B', NS(f=False, bar='B', baz=None)),
- ('-f --bar B', NS(f=True, bar='B', baz=None)),
- ('-f -b', NS(f=True, bar=None, baz=42)),
- ('-ba -f', NS(f=True, bar=None, baz=42)),
- ]
- class TestOptionalsAlternatePrefixChars(ParserTestCase):
- """Test an Optional with option strings with custom prefixes"""
- parser_signature = Sig(prefix_chars='+:/', add_help=False)
- argument_signatures = [
- Sig('+f', action='store_true'),
- Sig('::bar'),
- Sig('/baz', action='store_const', const=42),
- ]
- failures = ['--bar', '-fbar', '-b B', 'B', '-f', '--bar B', '-baz', '-h', '--help', '+h', '::help', '/help']
- successes = [
- ('', NS(f=False, bar=None, baz=None)),
- ('+f', NS(f=True, bar=None, baz=None)),
- ('::ba B', NS(f=False, bar='B', baz=None)),
- ('+f ::bar B', NS(f=True, bar='B', baz=None)),
- ('+f /b', NS(f=True, bar=None, baz=42)),
- ('/ba +f', NS(f=True, bar=None, baz=42)),
- ]
- class TestOptionalsAlternatePrefixCharsAddedHelp(ParserTestCase):
- """When ``-`` not in prefix_chars, default operators created for help
- should use the prefix_chars in use rather than - or --
- http://bugs.python.org/issue9444"""
- parser_signature = Sig(prefix_chars='+:/', add_help=True)
- argument_signatures = [
- Sig('+f', action='store_true'),
- Sig('::bar'),
- Sig('/baz', action='store_const', const=42),
- ]
- failures = ['--bar', '-fbar', '-b B', 'B', '-f', '--bar B', '-baz']
- successes = [
- ('', NS(f=False, bar=None, baz=None)),
- ('+f', NS(f=True, bar=None, baz=None)),
- ('::ba B', NS(f=False, bar='B', baz=None)),
- ('+f ::bar B', NS(f=True, bar='B', baz=None)),
- ('+f /b', NS(f=True, bar=None, baz=42)),
- ('/ba +f', NS(f=True, bar=None, baz=42))
- ]
- class TestOptionalsAlternatePrefixCharsMultipleShortArgs(ParserTestCase):
- """Verify that Optionals must be called with their defined prefixes"""
- parser_signature = Sig(prefix_chars='+-', add_help=False)
- argument_signatures = [
- Sig('-x', action='store_true'),
- Sig('+y', action='store_true'),
- Sig('+z', action='store_true'),
- ]
- failures = ['-w',
- '-xyz',
- '+x',
- '-y',
- '+xyz',
- ]
- successes = [
- ('', NS(x=False, y=False, z=False)),
- ('-x', NS(x=True, y=False, z=False)),
- ('+y -x', NS(x=True, y=True, z=False)),
- ('+yz -x', NS(x=True, y=True, z=True)),
- ]
- class TestOptionalsShortLong(ParserTestCase):
- """Test a combination of single- and double-dash option strings"""
- argument_signatures = [
- Sig('-v', '--verbose', '-n', '--noisy', action='store_true'),
- ]
- failures = ['--x --verbose', '-N', 'a', '-v x']
- successes = [
- ('', NS(verbose=False)),
- ('-v', NS(verbose=True)),
- ('--verbose', NS(verbose=True)),
- ('-n', NS(verbose=True)),
- ('--noisy', NS(verbose=True)),
- ]
- class TestOptionalsDest(ParserTestCase):
- """Tests various means of setting destination"""
- argument_signatures = [Sig('--foo-bar'), Sig('--baz', dest='zabbaz')]
- failures = ['a']
- successes = [
- ('--foo-bar f', NS(foo_bar='f', zabbaz=None)),
- ('--baz g', NS(foo_bar=None, zabbaz='g')),
- ('--foo-bar h --baz i', NS(foo_bar='h', zabbaz='i')),
- ('--baz j --foo-bar k', NS(foo_bar='k', zabbaz='j')),
- ]
- class TestOptionalsDefault(ParserTestCase):
- """Tests specifying a default for an Optional"""
- argument_signatures = [Sig('-x'), Sig('-y', default=42)]
- failures = ['a']
- successes = [
- ('', NS(x=None, y=42)),
- ('-xx', NS(x='x', y=42)),
- ('-yy', NS(x=None, y='y')),
- ]
- class TestOptionalsNargsDefault(ParserTestCase):
- """Tests not specifying the number of args for an Optional"""
- argument_signatures = [Sig('-x')]
- failures = ['a', '-x']
- successes = [
- ('', NS(x=None)),
- ('-x a', NS(x='a')),
- ]
- class TestOptionalsNargs1(ParserTestCase):
- """Tests specifying 1 arg for an Optional"""
- argument_signatures = [Sig('-x', nargs=1)]
- failures = ['a', '-x']
- successes = [
- ('', NS(x=None)),
- ('-x a', NS(x=['a'])),
- ]
- class TestOptionalsNargs3(ParserTestCase):
- """Tests specifying 3 args for an Optional"""
- argument_signatures = [Sig('-x', nargs=3)]
- failures = ['a', '-x', '-x a', '-x a b', 'a -x', 'a -x b']
- successes = [
- ('', NS(x=None)),
- ('-x a b c', NS(x=['a', 'b', 'c'])),
- ]
- class TestOptionalsNargsOptional(ParserTestCase):
- """Tests specifying an Optional arg for an Optional"""
- argument_signatures = [
- Sig('-w', nargs='?'),
- Sig('-x', nargs='?', const=42),
- Sig('-y', nargs='?', default='spam'),
- Sig('-z', nargs='?', type=int, const='42', default='84'),
- ]
- failures = ['2']
- successes = [
- ('', NS(w=None, x=None, y='spam', z=84)),
- ('-w', NS(w=None, x=None, y='spam', z=84)),
- ('-w 2', NS(w='2', x=None, y='spam', z=84)),
- ('-x', NS(w=None, x=42, y='spam', z=84)),
- ('-x 2', NS(w=None, x='2', y='spam', z=84)),
- ('-y', NS(w=None, x=None, y=None, z=84)),
- ('-y 2', NS(w=None, x=None, y='2', z=84)),
- ('-z', NS(w=None, x=None, y='spam', z=42)),
- ('-z 2', NS(w=None, x=None, y='spam', z=2)),
- ]
- class TestOptionalsNargsZeroOrMore(ParserTestCase):
- """Tests specifying args for an Optional that accepts zero or more"""
- argument_signatures = [
- Sig('-x', nargs='*'),
- Sig('-y', nargs='*', default='spam'),
- ]
- failures = ['a']
- successes = [
- ('', NS(x=None, y='spam')),
- ('-x', NS(x=[], y='spam')),
- ('-x a', NS(x=['a'], y='spam')),
- ('-x a b', NS(x=['a', 'b'], y='spam')),
- ('-y', NS(x=None, y=[])),
- ('-y a', NS(x=None, y=['a'])),
- ('-y a b', NS(x=None, y=['a', 'b'])),
- ]
- class TestOptionalsNargsOneOrMore(ParserTestCase):
- """Tests specifying args for an Optional that accepts one or more"""
- argument_signatures = [
- Sig('-x', nargs='+'),
- Sig('-y', nargs='+', default='spam'),
- ]
- failures = ['a', '-x', '-y', 'a -x', 'a -y b']
- successes = [
- ('', NS(x=None, y='spam')),
- ('-x a', NS(x=['a'], y='spam')),
- ('-x a b', NS(x=['a', 'b'], y='spam')),
- ('-y a', NS(x=None, y=['a'])),
- ('-y a b', NS(x=None, y=['a', 'b'])),
- ]
- class TestOptionalsChoices(ParserTestCase):
- """Tests specifying the choices for an Optional"""
- argument_signatures = [
- Sig('-f', choices='abc'),
- Sig('-g', type=int, choices=range(5))]
- failures = ['a', '-f d', '-fad', '-ga', '-g 6']
- successes = [
- ('', NS(f=None, g=None)),
- ('-f a', NS(f='a', g=None)),
- ('-f c', NS(f='c', g=None)),
- ('-g 0', NS(f=None, g=0)),
- ('-g 03', NS(f=None, g=3)),
- ('-fb -g4', NS(f='b', g=4)),
- ]
- class TestOptionalsRequired(ParserTestCase):
- """Tests an optional action that is required"""
- argument_signatures = [
- Sig('-x', type=int, required=True),
- ]
- failures = ['a', '']
- successes = [
- ('-x 1', NS(x=1)),
- ('-x42', NS(x=42)),
- ]
- class TestOptionalsActionStore(ParserTestCase):
- """Tests the store action for an Optional"""
- argument_signatures = [Sig('-x', action='store')]
- failures = ['a', 'a -x']
- successes = [
- ('', NS(x=None)),
- ('-xfoo', NS(x='foo')),
- ]
- class TestOptionalsActionStoreConst(ParserTestCase):
- """Tests the store_const action for an Optional"""
- argument_signatures = [Sig('-y', action='store_const', const=object)]
- failures = ['a']
- successes = [
- ('', NS(y=None)),
- ('-y', NS(y=object)),
- ]
- class TestOptionalsActionStoreFalse(ParserTestCase):
- """Tests the store_false action for an Optional"""
- argument_signatures = [Sig('-z', action='store_false')]
- failures = ['a', '-za', '-z a']
- successes = [
- ('', NS(z=True)),
- ('-z', NS(z=False)),
- ]
- class TestOptionalsActionStoreTrue(ParserTestCase):
- """Tests the store_true action for an Optional"""
- argument_signatures = [Sig('--apple', action='store_true')]
- failures = ['a', '--apple=b', '--apple b']
- successes = [
- ('', NS(apple=False)),
- ('--apple', NS(apple=True)),
- ]
- class TestBooleanOptionalAction(ParserTestCase):
- """Tests BooleanOptionalAction"""
- argument_signatures = [Sig('--foo', action=argparse.BooleanOptionalAction)]
- failures = ['--foo bar', '--foo=bar']
- successes = [
- ('', NS(foo=None)),
- ('--foo', NS(foo=True)),
- ('--no-foo', NS(foo=False)),
- ('--foo --no-foo', NS(foo=False)), # useful for aliases
- ('--no-foo --foo', NS(foo=True)),
- ]
- def test_const(self):
- # See bpo-40862
- parser = argparse.ArgumentParser()
- with self.assertRaises(TypeError) as cm:
- parser.add_argument('--foo', const=True, action=argparse.BooleanOptionalAction)
- self.assertIn("got an unexpected keyword argument 'const'", str(cm.exception))
- class TestBooleanOptionalActionRequired(ParserTestCase):
- """Tests BooleanOptionalAction required"""
- argument_signatures = [
- Sig('--foo', required=True, action=argparse.BooleanOptionalAction)
- ]
- failures = ['']
- successes = [
- ('--foo', NS(foo=True)),
- ('--no-foo', NS(foo=False)),
- ]
- class TestOptionalsActionAppend(ParserTestCase):
- """Tests the append action for an Optional"""
- argument_signatures = [Sig('--baz', action='append')]
- failures = ['a', '--baz', 'a --baz', '--baz a b']
- successes = [
- ('', NS(baz=None)),
- ('--baz a', NS(baz=['a'])),
- ('--baz a --baz b', NS(baz=['a', 'b'])),
- ]
- class TestOptionalsActionAppendWithDefault(ParserTestCase):
- """Tests the append action for an Optional"""
- argument_signatures = [Sig('--baz', action='append', default=['X'])]
- failures = ['a', '--baz', 'a --baz', '--baz a b']
- successes = [
- ('', NS(baz=['X'])),
- ('--baz a', NS(baz=['X', 'a'])),
- ('--baz a --baz b', NS(baz=['X', 'a', 'b'])),
- ]
- class TestConstActionsMissingConstKwarg(ParserTestCase):
- """Tests that const gets default value of None when not provided"""
- argument_signatures = [
- Sig('-f', action='append_const'),
- Sig('--foo', action='append_const'),
- Sig('-b', action='store_const'),
- Sig('--bar', action='store_const')
- ]
- failures = ['-f v', '--foo=bar', '--foo bar']
- successes = [
- ('', NS(f=None, foo=None, b=None, bar=None)),
- ('-f', NS(f=[None], foo=None, b=None, bar=None)),
- ('--foo', NS(f=None, foo=[None], b=None, bar=None)),
- ('-b', NS(f=None, foo=None, b=None, bar=None)),
- ('--bar', NS(f=None, foo=None, b=None, bar=None)),
- ]
- class TestOptionalsActionAppendConst(ParserTestCase):
- """Tests the append_const action for an Optional"""
- argument_signatures = [
- Sig('-b', action='append_const', const=Exception),
- Sig('-c', action='append', dest='b'),
- ]
- failures = ['a', '-c', 'a -c', '-bx', '-b x']
- successes = [
- ('', NS(b=None)),
- ('-b', NS(b=[Exception])),
- ('-b -cx -b -cyz', NS(b=[Exception, 'x', Exception, 'yz'])),
- ]
- class TestOptionalsActionAppendConstWithDefault(ParserTestCase):
- """Tests the append_const action for an Optional"""
- argument_signatures = [
- Sig('-b', action='append_const', const=Exception, default=['X']),
- Sig('-c', action='append', dest='b'),
- ]
- failures = ['a', '-c', 'a -c', '-bx', '-b x']
- successes = [
- ('', NS(b=['X'])),
- ('-b', NS(b=['X', Exception])),
- ('-b -cx -b -cyz', NS(b=['X', Exception, 'x', Exception, 'yz'])),
- ]
- class TestOptionalsActionCount(ParserTestCase):
- """Tests the count action for an Optional"""
- argument_signatures = [Sig('-x', action='count')]
- failures = ['a', '-x a', '-x b', '-x a -x b']
- successes = [
- ('', NS(x=None)),
- ('-x', NS(x=1)),
- ]
- class TestOptionalsAllowLongAbbreviation(ParserTestCase):
- """Allow long options to be abbreviated unambiguously"""
- argument_signatures = [
- Sig('--foo'),
- Sig('--foobaz'),
- Sig('--fooble', action='store_true'),
- ]
- failures = ['--foob 5', '--foob']
- successes = [
- ('', NS(foo=None, foobaz=None, fooble=False)),
- ('--foo 7', NS(foo='7', foobaz=None, fooble=False)),
- ('--fooba a', NS(foo=None, foobaz='a', fooble=False)),
- ('--foobl --foo g', NS(foo='g', foobaz=None, fooble=True)),
- ]
- class TestOptionalsDisallowLongAbbreviation(ParserTestCase):
- """Do not allow abbreviations of long options at all"""
- parser_signature = Sig(allow_abbrev=False)
- argument_signatures = [
- Sig('--foo'),
- Sig('--foodle', action='store_true'),
- Sig('--foonly'),
- ]
- failures = ['-foon 3', '--foon 3', '--food', '--food --foo 2']
- successes = [
- ('', NS(foo=None, foodle=False, foonly=None)),
- ('--foo 3', NS(foo='3', foodle=False, foonly=None)),
- ('--foonly 7 --foodle --foo 2', NS(foo='2', foodle=True, foonly='7')),
- ]
- class TestOptionalsDisallowLongAbbreviationPrefixChars(ParserTestCase):
- """Disallowing abbreviations works with alternative prefix characters"""
- parser_signature = Sig(prefix_chars='+', allow_abbrev=False)
- argument_signatures = [
- Sig('++foo'),
- Sig('++foodle', action='store_true'),
- Sig('++foonly'),
- ]
- failures = ['+foon 3', '++foon 3', '++food', '++food ++foo 2']
- successes = [
- ('', NS(foo=None, foodle=False, foonly=None)),
- ('++foo 3', NS(foo='3', foodle=False, foonly=None)),
- ('++foonly 7 ++foodle ++foo 2', NS(foo='2', foodle=True, foonly='7')),
- ]
- class TestDisallowLongAbbreviationAllowsShortGrouping(ParserTestCase):
- """Do not allow abbreviations of long options at all"""
- parser_signature = Sig(allow_abbrev=False)
- argument_signatures = [
- Sig('-r'),
- Sig('-c', action='count'),
- ]
- failures = ['-r', '-c -r']
- successes = [
- ('', NS(r=None, c=None)),
- ('-ra', NS(r='a', c=None)),
- ('-rcc', NS(r='cc', c=None)),
- ('-cc', NS(r=None, c=2)),
- ('-cc -ra', NS(r='a', c=2)),
- ('-ccrcc', NS(r='cc', c=2)),
- ]
- class TestDisallowLongAbbreviationAllowsShortGroupingPrefix(ParserTestCase):
- """Short option grouping works with custom prefix and allow_abbrev=False"""
- parser_signature = Sig(prefix_chars='+', allow_abbrev=False)
- argument_signatures = [
- Sig('+r'),
- Sig('+c', action='count'),
- ]
- failures = ['+r', '+c +r']
- successes = [
- ('', NS(r=None, c=None)),
- ('+ra', NS(r='a', c=None)),
- ('+rcc', NS(r='cc', c=None)),
- ('+cc', NS(r=None, c=2)),
- ('+cc +ra', NS(r='a', c=2)),
- ('+ccrcc', NS(r='cc', c=2)),
- ]
- # ================
- # Positional tests
- # ================
- class TestPositionalsNargsNone(ParserTestCase):
- """Test a Positional that doesn't specify nargs"""
- argument_signatures = [Sig('foo')]
- failures = ['', '-x', 'a b']
- successes = [
- ('a', NS(foo='a')),
- ]
- class TestPositionalsNargs1(ParserTestCase):
- """Test a Positional that specifies an nargs of 1"""
- argument_signatures = [Sig('foo', nargs=1)]
- failures = ['', '-x', 'a b']
- successes = [
- ('a', NS(foo=['a'])),
- ]
- class TestPositionalsNargs2(ParserTestCase):
- """Test a Positional that specifies an nargs of 2"""
- argument_signatures = [Sig('foo', nargs=2)]
- failures = ['', 'a', '-x', 'a b c']
- successes = [
- ('a b', NS(foo=['a', 'b'])),
- ]
- class TestPositionalsNargsZeroOrMore(ParserTestCase):
- """Test a Positional that specifies unlimited nargs"""
- argument_signatures = [Sig('foo', nargs='*')]
- failures = ['-x']
- successes = [
- ('', NS(foo=[])),
- ('a', NS(foo=['a'])),
- ('a b', NS(foo=['a', 'b'])),
- ]
- class TestPositionalsNargsZeroOrMoreDefault(ParserTestCase):
- """Test a Positional that specifies unlimited nargs and a default"""
- argument_signatures = [Sig('foo', nargs='*', default='bar')]
- failures = ['-x']
- successes = [
- ('', NS(foo='bar')),
- ('a', NS(foo=['a'])),
- ('a b', NS(foo=['a', 'b'])),
- ]
- class TestPositionalsNargsOneOrMore(ParserTestCase):
- """Test a Positional that specifies one or more nargs"""
- argument_signatures = [Sig('foo', nargs='+')]
- failures = ['', '-x']
- successes = [
- ('a', NS(foo=['a'])),
- ('a b', NS(foo=['a', 'b'])),
- ]
- class TestPositionalsNargsOptional(ParserTestCase):
- """Tests an Optional Positional"""
- argument_signatures = [Sig('foo', nargs='?')]
- failures = ['-x', 'a b']
- successes = [
- ('', NS(foo=None)),
- ('a', NS(foo='a')),
- ]
- class TestPositionalsNargsOptionalDefault(ParserTestCase):
- """Tests an Optional Positional with a default value"""
- argument_signatures = [Sig('foo', nargs='?', default=42)]
- failures = ['-x', 'a b']
- successes = [
- ('', NS(foo=42)),
- ('a', NS(foo='a')),
- ]
- class TestPositionalsNargsOptionalConvertedDefault(ParserTestCase):
- """Tests an Optional Positional with a default value
- that needs to be converted to the appropriate type.
- """
- argument_signatures = [
- Sig('foo', nargs='?', type=int, default='42'),
- ]
- failures = ['-x', 'a b', '1 2']
- successes = [
- ('', NS(foo=42)),
- ('1', NS(foo=1)),
- ]
- class TestPositionalsNargsNoneNone(ParserTestCase):
- """Test two Positionals that don't specify nargs"""
- argument_signatures = [Sig('foo'), Sig('bar')]
- failures = ['', '-x', 'a', 'a b c']
- successes = [
- ('a b', NS(foo='a', bar='b')),
- ]
- class TestPositionalsNargsNone1(ParserTestCase):
- """Test a Positional with no nargs followed by one with 1"""
- argument_signatures = [Sig('foo'), Sig('bar', nargs=1)]
- failures = ['', '--foo', 'a', 'a b c']
- successes = [
- ('a b', NS(foo='a', bar=['b'])),
- ]
- class TestPositionalsNargs2None(ParserTestCase):
- """Test a Positional with 2 nargs followed by one with none"""
- argument_signatures = [Sig('foo', nargs=2), Sig('bar')]
- failures = ['', '--foo', 'a', 'a b', 'a b c d']
- successes = [
- ('a b c', NS(foo=['a', 'b'], bar='c')),
- ]
- class TestPositionalsNargsNoneZeroOrMore(ParserTestCase):
- """Test a Positional with no nargs followed by one with unlimited"""
- argument_signatures = [Sig('foo'), Sig('bar', nargs='*')]
- failures = ['', '--foo']
- successes = [
- ('a', NS(foo='a', bar=[])),
- ('a b', NS(foo='a', bar=['b'])),
- ('a b c', NS(foo='a', bar=['b', 'c'])),
- ]
- class TestPositionalsNargsNoneOneOrMore(ParserTestCase):
- """Test a Positional with no nargs followed by one with one or more"""
- argument_signatures = [Sig('foo'), Sig('bar', nargs='+')]
- failures = ['', '--foo', 'a']
- successes = [
- ('a b', NS(foo='a', bar=['b'])),
- ('a b c', NS(foo='a', bar=['b', 'c'])),
- ]
- class TestPositionalsNargsNoneOptional(ParserTestCase):
- """Test a Positional with no nargs followed by one with an Optional"""
- argument_signatures = [Sig('foo'), Sig('bar', nargs='?')]
- failures = ['', '--foo', 'a b c']
- successes = [
- ('a', NS(foo='a', bar=None)),
- ('a b', NS(foo='a', bar='b')),
- ]
- class TestPositionalsNargsZeroOrMoreNone(ParserTestCase):
- """Test a Positional with unlimited nargs followed by one with none"""
- argument_signatures = [Sig('foo', nargs='*'), Sig('bar')]
- failures = ['', '--foo']
- successes = [
- ('a', NS(foo=[], bar='a')),
- ('a b', NS(foo=['a'], bar='b')),
- ('a b c', NS(foo=['a', 'b'], bar='c')),
- ]
- class TestPositionalsNargsOneOrMoreNone(ParserTestCase):
- """Test a Positional with one or more nargs followed by one with none"""
- argument_signatures = [Sig('foo', nargs='+'), Sig('bar')]
- failures = ['', '--foo', 'a']
- successes = [
- ('a b', NS(foo=['a'], bar='b')),
- ('a b c', NS(foo=['a', 'b'], bar='c')),
- ]
- class TestPositionalsNargsOptionalNone(ParserTestCase):
- """Test a Positional with an Optional nargs followed by one with none"""
- argument_signatures = [Sig('foo', nargs='?', default=42), Sig('bar')]
- failures = ['', '--foo', 'a b c']
- successes = [
- ('a', NS(foo=42, bar='a')),
- ('a b', NS(foo='a', bar='b')),
- ]
- class TestPositionalsNargs2ZeroOrMore(ParserTestCase):
- """Test a Positional with 2 nargs followed by one with unlimited"""
- argument_signatures = [Sig('foo', nargs=2), Sig('bar', nargs='*')]
- failures = ['', '--foo', 'a']
- successes = [
- ('a b', NS(foo=['a', 'b'], bar=[])),
- ('a b c', NS(foo=['a', 'b'], bar=['c'])),
- ]
- class TestPositionalsNargs2OneOrMore(ParserTestCase):
- """Test a Positional with 2 nargs followed by one with one or more"""
- argument_signatures = [Sig('foo', nargs=2), Sig('bar', nargs='+')]
- failures = ['', '--foo', 'a', 'a b']
- successes = [
- ('a b c', NS(foo=['a', 'b'], bar=['c'])),
- ]
- class TestPositionalsNargs2Optional(ParserTestCase):
- """Test a Positional with 2 nargs followed by one optional"""
- argument_signatures = [Sig('foo', nargs=2), Sig('bar', nargs='?')]
- failures = ['', '--foo', 'a', 'a b c d']
- successes = [
- ('a b', NS(foo=['a', 'b'], bar=None)),
- ('a b c', NS(foo=['a', 'b'], bar='c')),
- ]
- class TestPositionalsNargsZeroOrMore1(ParserTestCase):
- """Test a Positional with unlimited nargs followed by one with 1"""
- argument_signatures = [Sig('foo', nargs='*'), Sig('bar', nargs=1)]
- failures = ['', '--foo', ]
- successes = [
- ('a', NS(foo=[], bar=['a'])),
- ('a b', NS(foo=['a'], bar=['b'])),
- ('a b c', NS(foo=['a', 'b'], bar=['c'])),
- ]
- class TestPositionalsNargsOneOrMore1(ParserTestCase):
- """Test a Positional with one or more nargs followed by one with 1"""
- argument_signatures = [Sig('foo', nargs='+'), Sig('bar', nargs=1)]
- failures = ['', '--foo', 'a']
- successes = [
- ('a b', NS(foo=['a'], bar=['b'])),
- ('a b c', NS(foo=['a', 'b'], bar=['c'])),
- ]
- class TestPositionalsNargsOptional1(ParserTestCase):
- """Test a Positional with an Optional nargs followed by one with 1"""
- argument_signatures = [Sig('foo', nargs='?'), Sig('bar', nargs=1)]
- failures = ['', '--foo', 'a b c']
- successes = [
- ('a', NS(foo=None, bar=['a'])),
- ('a b', NS(foo='a', bar=['b'])),
- ]
- class TestPositionalsNargsNoneZeroOrMore1(ParserTestCase):
- """Test three Positionals: no nargs, unlimited nargs and 1 nargs"""
- argument_signatures = [
- Sig('foo'),
- Sig('bar', nargs='*'),
- Sig('baz', nargs=1),
- ]
- failures = ['', '--foo', 'a']
- successes = [
- ('a b', NS(foo='a', bar=[], baz=['b'])),
- ('a b c', NS(foo='a', bar=['b'], baz=['c'])),
- ]
- class TestPositionalsNargsNoneOneOrMore1(ParserTestCase):
- """Test three Positionals: no nargs, one or more nargs and 1 nargs"""
- argument_signatures = [
- Sig('foo'),
- Sig('bar', nargs='+'),
- Sig('baz', nargs=1),
- ]
- failures = ['', '--foo', 'a', 'b']
- successes = [
- ('a b c', NS(foo='a', bar=['b'], baz=['c'])),
- ('a b c d', NS(foo='a', bar=['b', 'c'], baz=['d'])),
- ]
- class TestPositionalsNargsNoneOptional1(ParserTestCase):
- """Test three Positionals: no nargs, optional narg and 1 nargs"""
- argument_signatures = [
- Sig('foo'),
- Sig('bar', nargs='?', default=0.625),
- Sig('baz', nargs=1),
- ]
- failures = ['', '--foo', 'a']
- successes = [
- ('a b', NS(foo='a', bar=0.625, baz=['b'])),
- ('a b c', NS(foo='a', bar='b', baz=['c'])),
- ]
- class TestPositionalsNargsOptionalOptional(ParserTestCase):
- """Test two optional nargs"""
- argument_signatures = [
- Sig('foo', nargs='?'),
- Sig('bar', nargs='?', default=42),
- ]
- failures = ['--foo', 'a b c']
- successes = [
- ('', NS(foo=None, bar=42)),
- ('a', NS(foo='a', bar=42)),
- ('a b', NS(foo='a', bar='b')),
- ]
- class TestPositionalsNargsOptionalZeroOrMore(ParserTestCase):
- """Test an Optional narg followed by unlimited nargs"""
- argument_signatures = [Sig('foo', nargs='?'), Sig('bar', nargs='*')]
- failures = ['--foo']
- successes = [
- ('', NS(foo=None, bar=[])),
- ('a', NS(foo='a', bar=[])),
- ('a b', NS(foo='a', bar=['b'])),
- ('a b c', NS(foo='a', bar=['b', 'c'])),
- ]
- class TestPositionalsNargsOptionalOneOrMore(ParserTestCase):
- """Test an Optional narg followed by one or more nargs"""
- argument_signatures = [Sig('foo', nargs='?'), Sig('bar', nargs='+')]
- failures = ['', '--foo']
- successes = [
- ('a', NS(foo=None, bar=['a'])),
- ('a b', NS(foo='a', bar=['b'])),
- ('a b c', NS(foo='a', bar=['b', 'c'])),
- ]
- class TestPositionalsChoicesString(ParserTestCase):
- """Test a set of single-character choices"""
- argument_signatures = [Sig('spam', choices=set('abcdefg'))]
- failures = ['', '--foo', 'h', '42', 'ef']
- successes = [
- ('a', NS(spam='a')),
- ('g', NS(spam='g')),
- ]
- class TestPositionalsChoicesInt(ParserTestCase):
- """Test a set of integer choices"""
- argument_signatures = [Sig('spam', type=int, choices=range(20))]
- failures = ['', '--foo', 'h', '42', 'ef']
- successes = [
- ('4', NS(spam=4)),
- ('15', NS(spam=15)),
- ]
- class TestPositionalsActionAppend(ParserTestCase):
- """Test the 'append' action"""
- argument_signatures = [
- Sig('spam', action='append'),
- Sig('spam', action='append', nargs=2),
- ]
- failures = ['', '--foo', 'a', 'a b', 'a b c d']
- successes = [
- ('a b c', NS(spam=['a', ['b', 'c']])),
- ]
- # ========================================
- # Combined optionals and positionals tests
- # ========================================
- class TestOptionalsNumericAndPositionals(ParserTestCase):
- """Tests negative number args when numeric options are present"""
- argument_signatures = [
- Sig('x', nargs='?'),
- Sig('-4', dest='y', action='store_true'),
- ]
- failures = ['-2', '-315']
- successes = [
- ('', NS(x=None, y=False)),
- ('a', NS(x='a', y=False)),
- ('-4', NS(x=None, y=True)),
- ('-4 a', NS(x='a', y=True)),
- ]
- class TestOptionalsAlmostNumericAndPositionals(ParserTestCase):
- """Tests negative number args when almost numeric options are present"""
- argument_signatures = [
- Sig('x', nargs='?'),
- Sig('-k4', dest='y', action='store_true'),
- ]
- failures = ['-k3']
- successes = [
- ('', NS(x=None, y=False)),
- ('-2', NS(x='-2', y=False)),
- ('a', NS(x='a', y=False)),
- ('-k4', NS(x=None, y=True)),
- ('-k4 a', NS(x='a', y=True)),
- ]
- class TestEmptyAndSpaceContainingArguments(ParserTestCase):
- argument_signatures = [
- Sig('x', nargs='?'),
- Sig('-y', '--yyy', dest='y'),
- ]
- failures = ['-y']
- successes = [
- ([''], NS(x='', y=None)),
- (['a badger'], NS(x='a badger', y=None)),
- (['-a badger'], NS(x='-a badger', y=None)),
- (['-y', ''], NS(x=None, y='')),
- (['-y', 'a badger'], NS(x=None, y='a badger')),
- (['-y', '-a badger'], NS(x=None, y='-a badger')),
- (['--yyy=a badger'], NS(x=None, y='a badger')),
- (['--yyy=-a badger'], NS(x=None, y='-a badger')),
- ]
- class TestPrefixCharacterOnlyArguments(ParserTestCase):
- parser_signature = Sig(prefix_chars='-+')
- argument_signatures = [
- Sig('-', dest='x', nargs='?', const='badger'),
- Sig('+', dest='y', type=int, default=42),
- Sig('-+-', dest='z', action='store_true'),
- ]
- failures = ['-y', '+ -']
- successes = [
- ('', NS(x=None, y=42, z=False)),
- ('-', NS(x='badger', y=42, z=False)),
- ('- X', NS(x='X', y=42, z=False)),
- ('+ -3', NS(x=None, y=-3, z=False)),
- ('-+-', NS(x=None, y=42, z=True)),
- ('- ===', NS(x='===', y=42, z=False)),
- ]
- class TestNargsZeroOrMore(ParserTestCase):
- """Tests specifying args for an Optional that accepts zero or more"""
- argument_signatures = [Sig('-x', nargs='*'), Sig('y', nargs='*')]
- failures = []
- successes = [
- ('', NS(x=None, y=[])),
- ('-x', NS(x=[], y=[])),
- ('-x a', NS(x=['a'], y=[])),
- ('-x a -- b', NS(x=['a'], y=['b'])),
- ('a', NS(x=None, y=['a'])),
- ('a -x', NS(x=[], y=['a'])),
- ('a -x b', NS(x=['b'], y=['a'])),
- ]
- class TestNargsRemainder(ParserTestCase):
- """Tests specifying a positional with nargs=REMAINDER"""
- argument_signatures = [Sig('x'), Sig('y', nargs='...'), Sig('-z')]
- failures = ['', '-z', '-z Z']
- successes = [
- ('X', NS(x='X', y=[], z=None)),
- ('-z Z X', NS(x='X', y=[], z='Z')),
- ('X A B -z Z', NS(x='X', y=['A', 'B', '-z', 'Z'], z=None)),
- ('X Y --foo', NS(x='X', y=['Y', '--foo'], z=None)),
- ]
- class TestOptionLike(ParserTestCase):
- """Tests options that may or may not be arguments"""
- argument_signatures = [
- Sig('-x', type=float),
- Sig('-3', type=float, dest='y'),
- Sig('z', nargs='*'),
- ]
- failures = ['-x', '-y2.5', '-xa', '-x -a',
- '-x -3', '-x -3.5', '-3 -3.5',
- '-x -2.5', '-x -2.5 a', '-3 -.5',
- 'a x -1', '-x -1 a', '-3 -1 a']
- successes = [
- ('', NS(x=None, y=None, z=[])),
- ('-x 2.5', NS(x=2.5, y=None, z=[])),
- ('-x 2.5 a', NS(x=2.5, y=None, z=['a'])),
- ('-3.5', NS(x=None, y=0.5, z=[])),
- ('-3-.5', NS(x=None, y=-0.5, z=[])),
- ('-3 .5', NS(x=None, y=0.5, z=[])),
- ('a -3.5', NS(x=None, y=0.5, z=['a'])),
- ('a', NS(x=None, y=None, z=['a'])),
- ('a -x 1', NS(x=1.0, y=None, z=['a'])),
- ('-x 1 a', NS(x=1.0, y=None, z=['a'])),
- ('-3 1 a', NS(x=None, y=1.0, z=['a'])),
- ]
- class TestDefaultSuppress(ParserTestCase):
- """Test actions with suppressed defaults"""
- argument_signatures = [
- Sig('foo', nargs='?', default=argparse.SUPPRESS),
- Sig('bar', nargs='*', default=argparse.SUPPRESS),
- Sig('--baz', action='store_true', default=argparse.SUPPRESS),
- ]
- failures = ['-x']
- successes = [
- ('', NS()),
- ('a', NS(foo='a')),
- ('a b', NS(foo='a', bar=['b'])),
- ('--baz', NS(baz=True)),
- ('a --baz', NS(foo='a', baz=True)),
- ('--baz a b', NS(foo='a', bar=['b'], baz=True)),
- ]
- class TestParserDefaultSuppress(ParserTestCase):
- """Test actions with a parser-level default of SUPPRESS"""
- parser_signature = Sig(argument_default=argparse.SUPPRESS)
- argument_signatures = [
- Sig('foo', nargs='?'),
- Sig('bar', nargs='*'),
- Sig('--baz', action='store_true'),
- ]
- failures = ['-x']
- successes = [
- ('', NS()),
- ('a', NS(foo='a')),
- ('a b', NS(foo='a', bar=['b'])),
- ('--baz', NS(baz=True)),
- ('a --baz', NS(foo='a', baz=True)),
- ('--baz a b', NS(foo='a', bar=['b'], baz=True)),
- ]
- class TestParserDefault42(ParserTestCase):
- """Test actions with a parser-level default of 42"""
- parser_signature = Sig(argument_default=42)
- argument_signatures = [
- Sig('--version', action='version', version='1.0'),
- Sig('foo', nargs='?'),
- Sig('bar', nargs='*'),
- Sig('--baz', action='store_true'),
- ]
- failures = ['-x']
- successes = [
- ('', NS(foo=42, bar=42, baz=42, version=42)),
- ('a', NS(foo='a', bar=42, baz=42, version=42)),
- ('a b', NS(foo='a', bar=['b'], baz=42, version=42)),
- ('--baz', NS(foo=42, bar=42, baz=True, version=42)),
- ('a --baz', NS(foo='a', bar=42, baz=True, version=42)),
- ('--baz a b', NS(foo='a', bar=['b'], baz=True, version=42)),
- ]
- class TestArgumentsFromFile(TempDirMixin, ParserTestCase):
- """Test reading arguments from a file"""
- def setUp(self):
- super(TestArgumentsFromFile, self).setUp()
- file_texts = [
- ('hello', 'hello world!\n'),
- ('recursive', '-a\n'
- 'A\n'
- '@hello'),
- ('invalid', '@no-such-path\n'),
- ]
- for path, text in file_texts:
- with open(path, 'w', encoding="utf-8") as file:
- file.write(text)
- parser_signature = Sig(fromfile_prefix_chars='@')
- argument_signatures = [
- Sig('-a'),
- Sig('x'),
- Sig('y', nargs='+'),
- ]
- failures = ['', '-b', 'X', '@invalid', '@missing']
- successes = [
- ('X Y', NS(a=None, x='X', y=['Y'])),
- ('X -a A Y Z', NS(a='A', x='X', y=['Y', 'Z'])),
- ('@hello X', NS(a=None, x='hello world!', y=['X'])),
- ('X @hello', NS(a=None, x='X', y=['hello world!'])),
- ('-a B @recursive Y Z', NS(a='A', x='hello world!', y=['Y', 'Z'])),
- ('X @recursive Z -a B', NS(a='B', x='X', y=['hello world!', 'Z'])),
- (["-a", "", "X", "Y"], NS(a='', x='X', y=['Y'])),
- ]
- class TestArgumentsFromFileConverter(TempDirMixin, ParserTestCase):
- """Test reading arguments from a file"""
- def setUp(self):
- super(TestArgumentsFromFileConverter, self).setUp()
- file_texts = [
- ('hello', 'hello world!\n'),
- ]
- for path, text in file_texts:
- with open(path, 'w', encoding="utf-8") as file:
- file.write(text)
- class FromFileConverterArgumentParser(ErrorRaisingArgumentParser):
- def convert_arg_line_to_args(self, arg_line):
- for arg in arg_line.split():
- if not arg.strip():
- continue
- yield arg
- parser_class = FromFileConverterArgumentParser
- parser_signature = Sig(fromfile_prefix_chars='@')
- argument_signatures = [
- Sig('y', nargs='+'),
- ]
- failures = []
- successes = [
- ('@hello X', NS(y=['hello', 'world!', 'X'])),
- ]
- # =====================
- # Type conversion tests
- # =====================
- class TestFileTypeRepr(TestCase):
- def test_r(self):
- type = argparse.FileType('r')
- self.assertEqual("FileType('r')", repr(type))
- def test_wb_1(self):
- type = argparse.FileType('wb', 1)
- self.assertEqual("FileType('wb', 1)", repr(type))
- def test_r_latin(self):
- type = argparse.FileType('r', encoding='latin_1')
- self.assertEqual("FileType('r', encoding='latin_1')", repr(type))
- def test_w_big5_ignore(self):
- type = argparse.FileType('w', encoding='big5', errors='ignore')
- self.assertEqual("FileType('w', encoding='big5', errors='ignore')",
- repr(type))
- def test_r_1_replace(self):
- type = argparse.FileType('r', 1, errors='replace')
- self.assertEqual("FileType('r', 1, errors='replace')", repr(type))
- BIN_STDOUT_SENTINEL = object()
- BIN_STDERR_SENTINEL = object()
- class StdStreamComparer:
- def __init__(self, attr):
- # We try to use the actual stdXXX.buffer attribute as our
- # marker, but but under some test environments,
- # sys.stdout/err are replaced by io.StringIO which won't have .buffer,
- # so we use a sentinel simply to show that the tests do the right thing
- # for any buffer supporting object
- self.getattr = operator.attrgetter(attr)
- if attr == 'stdout.buffer':
- self.backupattr = BIN_STDOUT_SENTINEL
- elif attr == 'stderr.buffer':
- self.backupattr = BIN_STDERR_SENTINEL
- else:
- self.backupattr = object() # Not equal to anything
- def __eq__(self, other):
- try:
- return other == self.getattr(sys)
- except AttributeError:
- return other == self.backupattr
- eq_stdin = StdStreamComparer('stdin')
- eq_stdout = StdStreamComparer('stdout')
- eq_stderr = StdStreamComparer('stderr')
- eq_bstdin = StdStreamComparer('stdin.buffer')
- eq_bstdout = StdStreamComparer('stdout.buffer')
- eq_bstderr = StdStreamComparer('stderr.buffer')
- class RFile(object):
- seen = {}
- def __init__(self, name):
- self.name = name
- def __eq__(self, other):
- if other in self.seen:
- text = self.seen[other]
- else:
- text = self.seen[other] = other.read()
- other.close()
- if not isinstance(text, str):
- text = text.decode('ascii')
- return self.name == other.name == text
- class TestFileTypeR(TempDirMixin, ParserTestCase):
- """Test the FileType option/argument type for reading files"""
- def setUp(self):
- super(TestFileTypeR, self).setUp()
- for file_name in ['foo', 'bar']:
- with open(os.path.join(self.temp_dir, file_name),
- 'w', encoding="utf-8") as file:
- file.write(file_name)
- self.create_readonly_file('readonly')
- argument_signatures = [
- Sig('-x', type=argparse.FileType()),
- Sig('spam', type=argparse.FileType('r')),
- ]
- failures = ['-x', '', 'non-existent-file.txt']
- successes = [
- ('foo', NS(x=None, spam=RFile('foo'))),
- ('-x foo bar', NS(x=RFile('foo'), spam=RFile('bar'))),
- ('bar -x foo', NS(x=RFile('foo'), spam=RFile('bar'))),
- ('-x - -', NS(x=eq_stdin, spam=eq_stdin)),
- ('readonly', NS(x=None, spam=RFile('readonly'))),
- ]
- class TestFileTypeDefaults(TempDirMixin, ParserTestCase):
- """Test that a file is not created unless the default is needed"""
- def setUp(self):
- super(TestFileTypeDefaults, self).setUp()
- file = open(os.path.join(self.temp_dir, 'good'), 'w', encoding="utf-8")
- file.write('good')
- file.close()
- argument_signatures = [
- Sig('-c', type=argparse.FileType('r'), default='no-file.txt'),
- ]
- # should provoke no such file error
- failures = ['']
- # should not provoke error because default file is created
- successes = [('-c good', NS(c=RFile('good')))]
- class TestFileTypeRB(TempDirMixin, ParserTestCase):
- """Test the FileType option/argument type for reading files"""
- def setUp(self):
- super(TestFileTypeRB, self).setUp()
- for file_name in ['foo', 'bar']:
- with open(os.path.join(self.temp_dir, file_name),
- 'w', encoding="utf-8") as file:
- file.write(file_name)
- argument_signatures = [
- Sig('-x', type=argparse.FileType('rb')),
- Sig('spam', type=argparse.FileType('rb')),
- ]
- failures = ['-x', '']
- successes = [
- ('foo', NS(x=None, spam=RFile('foo'))),
- ('-x foo bar', NS(x=RFile('foo'), spam=RFile('bar'))),
- ('bar -x foo', NS(x=RFile('foo'), spam=RFile('bar'))),
- ('-x - -', NS(x=eq_bstdin, spam=eq_bstdin)),
- ]
- class WFile(object):
- seen = set()
- def __init__(self, name):
- self.name = name
- def __eq__(self, other):
- if other not in self.seen:
- text = 'Check that file is writable.'
- if 'b' in other.mode:
- text = text.encode('ascii')
- other.write(text)
- other.close()
- self.seen.add(other)
- return self.name == other.name
- @os_helper.skip_if_dac_override
- class TestFileTypeW(TempDirMixin, ParserTestCase):
- """Test the FileType option/argument type for writing files"""
- def setUp(self):
- super().setUp()
- self.create_readonly_file('readonly')
- self.create_writable_file('writable')
- argument_signatures = [
- Sig('-x', type=argparse.FileType('w')),
- Sig('spam', type=argparse.FileType('w')),
- ]
- failures = ['-x', '', 'readonly']
- successes = [
- ('foo', NS(x=None, spam=WFile('foo'))),
- ('writable', NS(x=None, spam=WFile('writable'))),
- ('-x foo bar', NS(x=WFile('foo'), spam=WFile('bar'))),
- ('bar -x foo', NS(x=WFile('foo'), spam=WFile('bar'))),
- ('-x - -', NS(x=eq_stdout, spam=eq_stdout)),
- ]
- @os_helper.skip_if_dac_override
- class TestFileTypeX(TempDirMixin, ParserTestCase):
- """Test the FileType option/argument type for writing new files only"""
- def setUp(self):
- super().setUp()
- self.create_readonly_file('readonly')
- self.create_writable_file('writable')
- argument_signatures = [
- Sig('-x', type=argparse.FileType('x')),
- Sig('spam', type=argparse.FileType('x')),
- ]
- failures = ['-x', '', 'readonly', 'writable']
- successes = [
- ('-x foo bar', NS(x=WFile('foo'), spam=WFile('bar'))),
- ('-x - -', NS(x=eq_stdout, spam=eq_stdout)),
- ]
- @os_helper.skip_if_dac_override
- class TestFileTypeWB(TempDirMixin, ParserTestCase):
- """Test the FileType option/argument type for writing binary files"""
- argument_signatures = [
- Sig('-x', type=argparse.FileType('wb')),
- Sig('spam', type=argparse.FileType('wb')),
- ]
- failures = ['-x', '']
- successes = [
- ('foo', NS(x=None, spam=WFile('foo'))),
- ('-x foo bar', NS(x=WFile('foo'), spam=WFile('bar'))),
- ('bar -x foo', NS(x=WFile('foo'), spam=WFile('bar'))),
- ('-x - -', NS(x=eq_bstdout, spam=eq_bstdout)),
- ]
- @os_helper.skip_if_dac_override
- class TestFileTypeXB(TestFileTypeX):
- "Test the FileType option/argument type for writing new binary files only"
- argument_signatures = [
- Sig('-x', type=argparse.FileType('xb')),
- Sig('spam', type=argparse.FileType('xb')),
- ]
- successes = [
- ('-x foo bar', NS(x=WFile('foo'), spam=WFile('bar'))),
- ('-x - -', NS(x=eq_bstdout, spam=eq_bstdout)),
- ]
- class TestFileTypeOpenArgs(TestCase):
- """Test that open (the builtin) is correctly called"""
- def test_open_args(self):
- FT = argparse.FileType
- cases = [
- (FT('rb'), ('rb', -1, None, None)),
- (FT('w', 1), ('w', 1, None, None)),
- (FT('w', errors='replace'), ('w', -1, None, 'replace')),
- (FT('wb', encoding='big5'), ('wb', -1, 'big5', None)),
- (FT('w', 0, 'l1', 'strict'), ('w', 0, 'l1', 'strict')),
- ]
- with mock.patch('builtins.open') as m:
- for type, args in cases:
- type('foo')
- m.assert_called_with('foo', *args)
- class TestFileTypeMissingInitialization(TestCase):
- """
- Test that add_argument throws an error if FileType class
- object was passed instead of instance of FileType
- """
- def test(self):
- parser = argparse.ArgumentParser()
- with self.assertRaises(ValueError) as cm:
- parser.add_argument('-x', type=argparse.FileType)
- self.assertEqual(
- '%r is a FileType class object, instance of it must be passed'
- % (argparse.FileType,),
- str(cm.exception)
- )
- class TestTypeCallable(ParserTestCase):
- """Test some callables as option/argument types"""
- argument_signatures = [
- Sig('--eggs', type=complex),
- Sig('spam', type=float),
- ]
- failures = ['a', '42j', '--eggs a', '--eggs 2i']
- successes = [
- ('--eggs=42 42', NS(eggs=42, spam=42.0)),
- ('--eggs 2j -- -1.5', NS(eggs=2j, spam=-1.5)),
- ('1024.675', NS(eggs=None, spam=1024.675)),
- ]
- class TestTypeUserDefined(ParserTestCase):
- """Test a user-defined option/argument type"""
- class MyType(TestCase):
- def __init__(self, value):
- self.value = value
- def __eq__(self, other):
- return (type(self), self.value) == (type(other), other.value)
- argument_signatures = [
- Sig('-x', type=MyType),
- Sig('spam', type=MyType),
- ]
- failures = []
- successes = [
- ('a -x b', NS(x=MyType('b'), spam=MyType('a'))),
- ('-xf g', NS(x=MyType('f'), spam=MyType('g'))),
- ]
- class TestTypeClassicClass(ParserTestCase):
- """Test a classic class type"""
- class C:
- def __init__(self, value):
- self.value = value
- def __eq__(self, other):
- return (type(self), self.value) == (type(other), other.value)
- argument_signatures = [
- Sig('-x', type=C),
- Sig('spam', type=C),
- ]
- failures = []
- successes = [
- ('a -x b', NS(x=C('b'), spam=C('a'))),
- ('-xf g', NS(x=C('f'), spam=C('g'))),
- ]
- class TestTypeRegistration(TestCase):
- """Test a user-defined type by registering it"""
- def test(self):
- def get_my_type(string):
- return 'my_type{%s}' % string
- parser = argparse.ArgumentParser()
- parser.register('type', 'my_type', get_my_type)
- parser.add_argument('-x', type='my_type')
- parser.add_argument('y', type='my_type')
- self.assertEqual(parser.parse_args('1'.split()),
- NS(x=None, y='my_type{1}'))
- self.assertEqual(parser.parse_args('-x 1 42'.split()),
- NS(x='my_type{1}', y='my_type{42}'))
- # ============
- # Action tests
- # ============
- class TestActionUserDefined(ParserTestCase):
- """Test a user-defined option/argument action"""
- class OptionalAction(argparse.Action):
- def __call__(self, parser, namespace, value, option_string=None):
- try:
- # check destination and option string
- assert self.dest == 'spam', 'dest: %s' % self.dest
- assert option_string == '-s', 'flag: %s' % option_string
- # when option is before argument, badger=2, and when
- # option is after argument, badger=<whatever was set>
- expected_ns = NS(spam=0.25)
- if value in [0.125, 0.625]:
- expected_ns.badger = 2
- elif value in [2.0]:
- expected_ns.badger = 84
- else:
- raise AssertionError('value: %s' % value)
- assert expected_ns == namespace, ('expected %s, got %s' %
- (expected_ns, namespace))
- except AssertionError as e:
- raise ArgumentParserError('opt_action failed: %s' % e)
- setattr(namespace, 'spam', value)
- class PositionalAction(argparse.Action):
- def __call__(self, parser, namespace, value, option_string=None):
- try:
- assert option_string is None, ('option_string: %s' %
- option_string)
- # check destination
- assert self.dest == 'badger', 'dest: %s' % self.dest
- # when argument is before option, spam=0.25, and when
- # option is after argument, spam=<whatever was set>
- expected_ns = NS(badger=2)
- if value in [42, 84]:
- expected_ns.spam = 0.25
- elif value in [1]:
- expected_ns.spam = 0.625
- elif value in [2]:
- expected_ns.spam = 0.125
- else:
- raise AssertionError('value: %s' % value)
- assert expected_ns == namespace, ('expected %s, got %s' %
- (expected_ns, namespace))
- except AssertionError as e:
- raise ArgumentParserError('arg_action failed: %s' % e)
- setattr(namespace, 'badger', value)
- argument_signatures = [
- Sig('-s', dest='spam', action=OptionalAction,
- type=float, default=0.25),
- Sig('badger', action=PositionalAction,
- type=int, nargs='?', default=2),
- ]
- failures = []
- successes = [
- ('-s0.125', NS(spam=0.125, badger=2)),
- ('42', NS(spam=0.25, badger=42)),
- ('-s 0.625 1', NS(spam=0.625, badger=1)),
- ('84 -s2', NS(spam=2.0, badger=84)),
- ]
- class TestActionRegistration(TestCase):
- """Test a user-defined action supplied by registering it"""
- class MyAction(argparse.Action):
- def __call__(self, parser, namespace, values, option_string=None):
- setattr(namespace, self.dest, 'foo[%s]' % values)
- def test(self):
- parser = argparse.ArgumentParser()
- parser.register('action', 'my_action', self.MyAction)
- parser.add_argument('badger', action='my_action')
- self.assertEqual(parser.parse_args(['1']), NS(badger='foo[1]'))
- self.assertEqual(parser.parse_args(['42']), NS(badger='foo[42]'))
- class TestActionExtend(ParserTestCase):
- argument_signatures = [
- Sig('--foo', action="extend", nargs="+", type=str),
- ]
- failures = ()
- successes = [
- ('--foo f1 --foo f2 f3 f4', NS(foo=['f1', 'f2', 'f3', 'f4'])),
- ]
- # ================
- # Subparsers tests
- # ================
- class TestAddSubparsers(TestCase):
- """Test the add_subparsers method"""
- def assertArgumentParserError(self, *args, **kwargs):
- self.assertRaises(ArgumentParserError, *args, **kwargs)
- def _get_parser(self, subparser_help=False, prefix_chars=None,
- aliases=False):
- # create a parser with a subparsers argument
- if prefix_chars:
- parser = ErrorRaisingArgumentParser(
- prog='PROG', description='main description', prefix_chars=prefix_chars)
- parser.add_argument(
- prefix_chars[0] * 2 + 'foo', action='store_true', help='foo help')
- else:
- parser = ErrorRaisingArgumentParser(
- prog='PROG', description='main description')
- parser.add_argument(
- '--foo', action='store_true', help='foo help')
- parser.add_argument(
- 'bar', type=float, help='bar help')
- # check that only one subparsers argument can be added
- subparsers_kwargs = {'required': False}
- if aliases:
- subparsers_kwargs['metavar'] = 'COMMAND'
- subparsers_kwargs['title'] = 'commands'
- else:
- subparsers_kwargs['help'] = 'command help'
- subparsers = parser.add_subparsers(**subparsers_kwargs)
- self.assertArgumentParserError(parser.add_subparsers)
- # add first sub-parser
- parser1_kwargs = dict(description='1 description')
- if subparser_help:
- parser1_kwargs['help'] = '1 help'
- if aliases:
- parser1_kwargs['aliases'] = ['1alias1', '1alias2']
- parser1 = subparsers.add_parser('1', **parser1_kwargs)
- parser1.add_argument('-w', type=int, help='w help')
- parser1.add_argument('x', choices='abc', help='x help')
- # add second sub-parser
- parser2_kwargs = dict(description='2 description')
- if subparser_help:
- parser2_kwargs['help'] = '2 help'
- parser2 = subparsers.add_parser('2', **parser2_kwargs)
- parser2.add_argument('-y', choices='123', help='y help')
- parser2.add_argument('z', type=complex, nargs='*', help='z help')
- # add third sub-parser
- parser3_kwargs = dict(description='3 description')
- if subparser_help:
- parser3_kwargs['help'] = '3 help'
- parser3 = subparsers.add_parser('3', **parser3_kwargs)
- parser3.add_argument('t', type=int, help='t help')
- parser3.add_argument('u', nargs='...', help='u help')
- # return the main parser
- return parser
- def setUp(self):
- super().setUp()
- self.parser = self._get_parser()
- self.command_help_parser = self._get_parser(subparser_help=True)
- def test_parse_args_failures(self):
- # check some failure cases:
- for args_str in ['', 'a', 'a a', '0.5 a', '0.5 1',
- '0.5 1 -y', '0.5 2 -w']:
- args = args_str.split()
- self.assertArgumentParserError(self.parser.parse_args, args)
- def test_parse_args(self):
- # check some non-failure cases:
- self.assertEqual(
- self.parser.parse_args('0.5 1 b -w 7'.split()),
- NS(foo=False, bar=0.5, w=7, x='b'),
- )
- self.assertEqual(
- self.parser.parse_args('0.25 --foo 2 -y 2 3j -- -1j'.split()),
- NS(foo=True, bar=0.25, y='2', z=[3j, -1j]),
- )
- self.assertEqual(
- self.parser.parse_args('--foo 0.125 1 c'.split()),
- NS(foo=True, bar=0.125, w=None, x='c'),
- )
- self.assertEqual(
- self.parser.parse_args('-1.5 3 11 -- a --foo 7 -- b'.split()),
- NS(foo=False, bar=-1.5, t=11, u=['a', '--foo', '7', '--', 'b']),
- )
- def test_parse_known_args(self):
- self.assertEqual(
- self.parser.parse_known_args('0.5 1 b -w 7'.split()),
- (NS(foo=False, bar=0.5, w=7, x='b'), []),
- )
- self.assertEqual(
- self.parser.parse_known_args('0.5 -p 1 b -w 7'.split()),
- (NS(foo=False, bar=0.5, w=7, x='b'), ['-p']),
- )
- self.assertEqual(
- self.parser.parse_known_args('0.5 1 b -w 7 -p'.split()),
- (NS(foo=False, bar=0.5, w=7, x='b'), ['-p']),
- )
- self.assertEqual(
- self.parser.parse_known_args('0.5 1 b -q -rs -w 7'.split()),
- (NS(foo=False, bar=0.5, w=7, x='b'), ['-q', '-rs']),
- )
- self.assertEqual(
- self.parser.parse_known_args('0.5 -W 1 b -X Y -w 7 Z'.split()),
- (NS(foo=False, bar=0.5, w=7, x='b'), ['-W', '-X', 'Y', 'Z']),
- )
- def test_dest(self):
- parser = ErrorRaisingArgumentParser()
- parser.add_argument('--foo', action='store_true')
- subparsers = parser.add_subparsers(dest='bar')
- parser1 = subparsers.add_parser('1')
- parser1.add_argument('baz')
- self.assertEqual(NS(foo=False, bar='1', baz='2'),
- parser.parse_args('1 2'.split()))
- def _test_required_subparsers(self, parser):
- # Should parse the sub command
- ret = parser.parse_args(['run'])
- self.assertEqual(ret.command, 'run')
- # Error when the command is missing
- self.assertArgumentParserError(parser.parse_args, ())
- def test_required_subparsers_via_attribute(self):
- parser = ErrorRaisingArgumentParser()
- subparsers = parser.add_subparsers(dest='command')
- subparsers.required = True
- subparsers.add_parser('run')
- self._test_required_subparsers(parser)
- def test_required_subparsers_via_kwarg(self):
- parser = ErrorRaisingArgumentParser()
- subparsers = parser.add_subparsers(dest='command', required=True)
- subparsers.add_parser('run')
- self._test_required_subparsers(parser)
- def test_required_subparsers_default(self):
- parser = ErrorRaisingArgumentParser()
- subparsers = parser.add_subparsers(dest='command')
- subparsers.add_parser('run')
- # No error here
- ret = parser.parse_args(())
- self.assertIsNone(ret.command)
- def test_required_subparsers_no_destination_error(self):
- parser = ErrorRaisingArgumentParser()
- subparsers = parser.add_subparsers(required=True)
- subparsers.add_parser('foo')
- subparsers.add_parser('bar')
- with self.assertRaises(ArgumentParserError) as excinfo:
- parser.parse_args(())
- self.assertRegex(
- excinfo.exception.stderr,
- 'error: the following arguments are required: {foo,bar}\n$'
- )
- def test_wrong_argument_subparsers_no_destination_error(self):
- parser = ErrorRaisingArgumentParser()
- subparsers = parser.add_subparsers(required=True)
- subparsers.add_parser('foo')
- subparsers.add_parser('bar')
- with self.assertRaises(ArgumentParserError) as excinfo:
- parser.parse_args(('baz',))
- self.assertRegex(
- excinfo.exception.stderr,
- r"error: argument {foo,bar}: invalid choice: 'baz' \(choose from 'foo', 'bar'\)\n$"
- )
- def test_optional_subparsers(self):
- parser = ErrorRaisingArgumentParser()
- subparsers = parser.add_subparsers(dest='command', required=False)
- subparsers.add_parser('run')
- # No error here
- ret = parser.parse_args(())
- self.assertIsNone(ret.command)
- def test_help(self):
- self.assertEqual(self.parser.format_usage(),
- 'usage: PROG [-h] [--foo] bar {1,2,3} ...\n')
- self.assertEqual(self.parser.format_help(), textwrap.dedent('''\
- usage: PROG [-h] [--foo] bar {1,2,3} ...
- main description
- positional arguments:
- bar bar help
- {1,2,3} command help
- options:
- -h, --help show this help message and exit
- --foo foo help
- '''))
- def test_help_extra_prefix_chars(self):
- # Make sure - is still used for help if it is a non-first prefix char
- parser = self._get_parser(prefix_chars='+:-')
- self.assertEqual(parser.format_usage(),
- 'usage: PROG [-h] [++foo] bar {1,2,3} ...\n')
- self.assertEqual(parser.format_help(), textwrap.dedent('''\
- usage: PROG [-h] [++foo] bar {1,2,3} ...
- main description
- positional arguments:
- bar bar help
- {1,2,3} command help
- options:
- -h, --help show this help message and exit
- ++foo foo help
- '''))
- def test_help_non_breaking_spaces(self):
- parser = ErrorRaisingArgumentParser(
- prog='PROG', description='main description')
- parser.add_argument(
- "--non-breaking", action='store_false',
- help='help message containing non-breaking spaces shall not '
- 'wrap\N{NO-BREAK SPACE}at non-breaking spaces')
- self.assertEqual(parser.format_help(), textwrap.dedent('''\
- usage: PROG [-h] [--non-breaking]
- main description
- options:
- -h, --help show this help message and exit
- --non-breaking help message containing non-breaking spaces shall not
- wrap\N{NO-BREAK SPACE}at non-breaking spaces
- '''))
- def test_help_blank(self):
- # Issue 24444
- parser = ErrorRaisingArgumentParser(
- prog='PROG', description='main description')
- parser.add_argument(
- 'foo',
- help=' ')
- self.assertEqual(parser.format_help(), textwrap.dedent('''\
- usage: PROG [-h] foo
- main description
- positional arguments:
- foo \n
- options:
- -h, --help show this help message and exit
- '''))
- parser = ErrorRaisingArgumentParser(
- prog='PROG', description='main description')
- parser.add_argument(
- 'foo', choices=[],
- help='%(choices)s')
- self.assertEqual(parser.format_help(), textwrap.dedent('''\
- usage: PROG [-h] {}
- main description
- positional arguments:
- {} \n
- options:
- -h, --help show this help message and exit
- '''))
- def test_help_alternate_prefix_chars(self):
- parser = self._get_parser(prefix_chars='+:/')
- self.assertEqual(parser.format_usage(),
- 'usage: PROG [+h] [++foo] bar {1,2,3} ...\n')
- self.assertEqual(parser.format_help(), textwrap.dedent('''\
- usage: PROG [+h] [++foo] bar {1,2,3} ...
- main description
- positional arguments:
- bar bar help
- {1,2,3} command help
- options:
- +h, ++help show this help message and exit
- ++foo foo help
- '''))
- def test_parser_command_help(self):
- self.assertEqual(self.command_help_parser.format_usage(),
- 'usage: PROG [-h] [--foo] bar {1,2,3} ...\n')
- self.assertEqual(self.command_help_parser.format_help(),
- textwrap.dedent('''\
- usage: PROG [-h] [--foo] bar {1,2,3} ...
- main description
- positional arguments:
- bar bar help
- {1,2,3} command help
- 1 1 help
- 2 2 help
- 3 3 help
- options:
- -h, --help show this help message and exit
- --foo foo help
- '''))
- def test_subparser_title_help(self):
- parser = ErrorRaisingArgumentParser(prog='PROG',
- description='main description')
- parser.add_argument('--foo', action='store_true', help='foo help')
- parser.add_argument('bar', help='bar help')
- subparsers = parser.add_subparsers(title='subcommands',
- description='command help',
- help='additional text')
- parser1 = subparsers.add_parser('1')
- parser2 = subparsers.add_parser('2')
- self.assertEqual(parser.format_usage(),
- 'usage: PROG [-h] [--foo] bar {1,2} ...\n')
- self.assertEqual(parser.format_help(), textwrap.dedent('''\
- usage: PROG [-h] [--foo] bar {1,2} ...
- main description
- positional arguments:
- bar bar help
- options:
- -h, --help show this help message and exit
- --foo foo help
- subcommands:
- command help
- {1,2} additional text
- '''))
- def _test_subparser_help(self, args_str, expected_help):
- with self.assertRaises(ArgumentParserError) as cm:
- self.parser.parse_args(args_str.split())
- self.assertEqual(expected_help, cm.exception.stdout)
- def test_subparser1_help(self):
- self._test_subparser_help('5.0 1 -h', textwrap.dedent('''\
- usage: PROG bar 1 [-h] [-w W] {a,b,c}
- 1 description
- positional arguments:
- {a,b,c} x help
- options:
- -h, --help show this help message and exit
- -w W w help
- '''))
- def test_subparser2_help(self):
- self._test_subparser_help('5.0 2 -h', textwrap.dedent('''\
- usage: PROG bar 2 [-h] [-y {1,2,3}] [z ...]
- 2 description
- positional arguments:
- z z help
- options:
- -h, --help show this help message and exit
- -y {1,2,3} y help
- '''))
- def test_alias_invocation(self):
- parser = self._get_parser(aliases=True)
- self.assertEqual(
- parser.parse_known_args('0.5 1alias1 b'.split()),
- (NS(foo=False, bar=0.5, w=None, x='b'), []),
- )
- self.assertEqual(
- parser.parse_known_args('0.5 1alias2 b'.split()),
- (NS(foo=False, bar=0.5, w=None, x='b'), []),
- )
- def test_error_alias_invocation(self):
- parser = self._get_parser(aliases=True)
- self.assertArgumentParserError(parser.parse_args,
- '0.5 1alias3 b'.split())
- def test_alias_help(self):
- parser = self._get_parser(aliases=True, subparser_help=True)
- self.maxDiff = None
- self.assertEqual(parser.format_help(), textwrap.dedent("""\
- usage: PROG [-h] [--foo] bar COMMAND ...
- main description
- positional arguments:
- bar bar help
- options:
- -h, --help show this help message and exit
- --foo foo help
- commands:
- COMMAND
- 1 (1alias1, 1alias2)
- 1 help
- 2 2 help
- 3 3 help
- """))
- # ============
- # Groups tests
- # ============
- class TestPositionalsGroups(TestCase):
- """Tests that order of group positionals matches construction order"""
- def test_nongroup_first(self):
- parser = ErrorRaisingArgumentParser()
- parser.add_argument('foo')
- group = parser.add_argument_group('g')
- group.add_argument('bar')
- parser.add_argument('baz')
- expected = NS(foo='1', bar='2', baz='3')
- result = parser.parse_args('1 2 3'.split())
- self.assertEqual(expected, result)
- def test_group_first(self):
- parser = ErrorRaisingArgumentParser()
- group = parser.add_argument_group('xxx')
- group.add_argument('foo')
- parser.add_argument('bar')
- parser.add_argument('baz')
- expected = NS(foo='1', bar='2', baz='3')
- result = parser.parse_args('1 2 3'.split())
- self.assertEqual(expected, result)
- def test_interleaved_groups(self):
- parser = ErrorRaisingArgumentParser()
- group = parser.add_argument_group('xxx')
- parser.add_argument('foo')
- group.add_argument('bar')
- parser.add_argument('baz')
- group = parser.add_argument_group('yyy')
- group.add_argument('frell')
- expected = NS(foo='1', bar='2', baz='3', frell='4')
- result = parser.parse_args('1 2 3 4'.split())
- self.assertEqual(expected, result)
- # ===================
- # Parent parser tests
- # ===================
- class TestParentParsers(TestCase):
- """Tests that parsers can be created with parent parsers"""
- def assertArgumentParserError(self, *args, **kwargs):
- self.assertRaises(ArgumentParserError, *args, **kwargs)
- def setUp(self):
- super().setUp()
- self.wxyz_parent = ErrorRaisingArgumentParser(add_help=False)
- self.wxyz_parent.add_argument('--w')
- x_group = self.wxyz_parent.add_argument_group('x')
- x_group.add_argument('-y')
- self.wxyz_parent.add_argument('z')
- self.abcd_parent = ErrorRaisingArgumentParser(add_help=False)
- self.abcd_parent.add_argument('a')
- self.abcd_parent.add_argument('-b')
- c_group = self.abcd_parent.add_argument_group('c')
- c_group.add_argument('--d')
- self.w_parent = ErrorRaisingArgumentParser(add_help=False)
- self.w_parent.add_argument('--w')
- self.z_parent = ErrorRaisingArgumentParser(add_help=False)
- self.z_parent.add_argument('z')
- # parents with mutually exclusive groups
- self.ab_mutex_parent = ErrorRaisingArgumentParser(add_help=False)
- group = self.ab_mutex_parent.add_mutually_exclusive_group()
- group.add_argument('-a', action='store_true')
- group.add_argument('-b', action='store_true')
- self.main_program = os.path.basename(sys.argv[0])
- def test_single_parent(self):
- parser = ErrorRaisingArgumentParser(parents=[self.wxyz_parent])
- self.assertEqual(parser.parse_args('-y 1 2 --w 3'.split()),
- NS(w='3', y='1', z='2'))
- def test_single_parent_mutex(self):
- self._test_mutex_ab(self.ab_mutex_parent.parse_args)
- parser = ErrorRaisingArgumentParser(parents=[self.ab_mutex_parent])
- self._test_mutex_ab(parser.parse_args)
- def test_single_granparent_mutex(self):
- parents = [self.ab_mutex_parent]
- parser = ErrorRaisingArgumentParser(add_help=False, parents=parents)
- parser = ErrorRaisingArgumentParser(parents=[parser])
- self._test_mutex_ab(parser.parse_args)
- def _test_mutex_ab(self, parse_args):
- self.assertEqual(parse_args([]), NS(a=False, b=False))
- self.assertEqual(parse_args(['-a']), NS(a=True, b=False))
- self.assertEqual(parse_args(['-b']), NS(a=False, b=True))
- self.assertArgumentParserError(parse_args, ['-a', '-b'])
- self.assertArgumentParserError(parse_args, ['-b', '-a'])
- self.assertArgumentParserError(parse_args, ['-c'])
- self.assertArgumentParserError(parse_args, ['-a', '-c'])
- self.assertArgumentParserError(parse_args, ['-b', '-c'])
- def test_multiple_parents(self):
- parents = [self.abcd_parent, self.wxyz_parent]
- parser = ErrorRaisingArgumentParser(parents=parents)
- self.assertEqual(parser.parse_args('--d 1 --w 2 3 4'.split()),
- NS(a='3', b=None, d='1', w='2', y=None, z='4'))
- def test_multiple_parents_mutex(self):
- parents = [self.ab_mutex_parent, self.wxyz_parent]
- parser = ErrorRaisingArgumentParser(parents=parents)
- self.assertEqual(parser.parse_args('-a --w 2 3'.split()),
- NS(a=True, b=False, w='2', y=None, z='3'))
- self.assertArgumentParserError(
- parser.parse_args, '-a --w 2 3 -b'.split())
- self.assertArgumentParserError(
- parser.parse_args, '-a -b --w 2 3'.split())
- def test_conflicting_parents(self):
- self.assertRaises(
- argparse.ArgumentError,
- argparse.ArgumentParser,
- parents=[self.w_parent, self.wxyz_parent])
- def test_conflicting_parents_mutex(self):
- self.assertRaises(
- argparse.ArgumentError,
- argparse.ArgumentParser,
- parents=[self.abcd_parent, self.ab_mutex_parent])
- def test_same_argument_name_parents(self):
- parents = [self.wxyz_parent, self.z_parent]
- parser = ErrorRaisingArgumentParser(parents=parents)
- self.assertEqual(parser.parse_args('1 2'.split()),
- NS(w=None, y=None, z='2'))
- def test_subparser_parents(self):
- parser = ErrorRaisingArgumentParser()
- subparsers = parser.add_subparsers()
- abcde_parser = subparsers.add_parser('bar', parents=[self.abcd_parent])
- abcde_parser.add_argument('e')
- self.assertEqual(parser.parse_args('bar -b 1 --d 2 3 4'.split()),
- NS(a='3', b='1', d='2', e='4'))
- def test_subparser_parents_mutex(self):
- parser = ErrorRaisingArgumentParser()
- subparsers = parser.add_subparsers()
- parents = [self.ab_mutex_parent]
- abc_parser = subparsers.add_parser('foo', parents=parents)
- c_group = abc_parser.add_argument_group('c_group')
- c_group.add_argument('c')
- parents = [self.wxyz_parent, self.ab_mutex_parent]
- wxyzabe_parser = subparsers.add_parser('bar', parents=parents)
- wxyzabe_parser.add_argument('e')
- self.assertEqual(parser.parse_args('foo -a 4'.split()),
- NS(a=True, b=False, c='4'))
- self.assertEqual(parser.parse_args('bar -b --w 2 3 4'.split()),
- NS(a=False, b=True, w='2', y=None, z='3', e='4'))
- self.assertArgumentParserError(
- parser.parse_args, 'foo -a -b 4'.split())
- self.assertArgumentParserError(
- parser.parse_args, 'bar -b -a 4'.split())
- def test_parent_help(self):
- parents = [self.abcd_parent, self.wxyz_parent]
- parser = ErrorRaisingArgumentParser(parents=parents)
- parser_help = parser.format_help()
- progname = self.main_program
- self.assertEqual(parser_help, textwrap.dedent('''\
- usage: {}{}[-h] [-b B] [--d D] [--w W] [-y Y] a z
- positional arguments:
- a
- z
- options:
- -h, --help show this help message and exit
- -b B
- --w W
- c:
- --d D
- x:
- -y Y
- '''.format(progname, ' ' if progname else '' )))
- def test_groups_parents(self):
- parent = ErrorRaisingArgumentParser(add_help=False)
- g = parent.add_argument_group(title='g', description='gd')
- g.add_argument('-w')
- g.add_argument('-x')
- m = parent.add_mutually_exclusive_group()
- m.add_argument('-y')
- m.add_argument('-z')
- parser = ErrorRaisingArgumentParser(parents=[parent])
- self.assertRaises(ArgumentParserError, parser.parse_args,
- ['-y', 'Y', '-z', 'Z'])
- parser_help = parser.format_help()
- progname = self.main_program
- self.assertEqual(parser_help, textwrap.dedent('''\
- usage: {}{}[-h] [-w W] [-x X] [-y Y | -z Z]
- options:
- -h, --help show this help message and exit
- -y Y
- -z Z
- g:
- gd
- -w W
- -x X
- '''.format(progname, ' ' if progname else '' )))
- # ==============================
- # Mutually exclusive group tests
- # ==============================
- class TestMutuallyExclusiveGroupErrors(TestCase):
- def test_invalid_add_argument_group(self):
- parser = ErrorRaisingArgumentParser()
- raises = self.assertRaises
- raises(TypeError, parser.add_mutually_exclusive_group, title='foo')
- def test_invalid_add_argument(self):
- parser = ErrorRaisingArgumentParser()
- group = parser.add_mutually_exclusive_group()
- add_argument = group.add_argument
- raises = self.assertRaises
- raises(ValueError, add_argument, '--foo', required=True)
- raises(ValueError, add_argument, 'bar')
- raises(ValueError, add_argument, 'bar', nargs='+')
- raises(ValueError, add_argument, 'bar', nargs=1)
- raises(ValueError, add_argument, 'bar', nargs=argparse.PARSER)
- def test_help(self):
- parser = ErrorRaisingArgumentParser(prog='PROG')
- group1 = parser.add_mutually_exclusive_group()
- group1.add_argument('--foo', action='store_true')
- group1.add_argument('--bar', action='store_false')
- group2 = parser.add_mutually_exclusive_group()
- group2.add_argument('--soup', action='store_true')
- group2.add_argument('--nuts', action='store_false')
- expected = '''\
- usage: PROG [-h] [--foo | --bar] [--soup | --nuts]
- options:
- -h, --help show this help message and exit
- --foo
- --bar
- --soup
- --nuts
- '''
- self.assertEqual(parser.format_help(), textwrap.dedent(expected))
- def test_empty_group(self):
- # See issue 26952
- parser = argparse.ArgumentParser()
- group = parser.add_mutually_exclusive_group()
- with self.assertRaises(ValueError):
- parser.parse_args(['-h'])
- class MEMixin(object):
- def test_failures_when_not_required(self):
- parse_args = self.get_parser(required=False).parse_args
- error = ArgumentParserError
- for args_string in self.failures:
- self.assertRaises(error, parse_args, args_string.split())
- def test_failures_when_required(self):
- parse_args = self.get_parser(required=True).parse_args
- error = ArgumentParserError
- for args_string in self.failures + ['']:
- self.assertRaises(error, parse_args, args_string.split())
- def test_successes_when_not_required(self):
- parse_args = self.get_parser(required=False).parse_args
- successes = self.successes + self.successes_when_not_required
- for args_string, expected_ns in successes:
- actual_ns = parse_args(args_string.split())
- self.assertEqual(actual_ns, expected_ns)
- def test_successes_when_required(self):
- parse_args = self.get_parser(required=True).parse_args
- for args_string, expected_ns in self.successes:
- actual_ns = parse_args(args_string.split())
- self.assertEqual(actual_ns, expected_ns)
- def test_usage_when_not_required(self):
- format_usage = self.get_parser(required=False).format_usage
- expected_usage = self.usage_when_not_required
- self.assertEqual(format_usage(), textwrap.dedent(expected_usage))
- def test_usage_when_required(self):
- format_usage = self.get_parser(required=True).format_usage
- expected_usage = self.usage_when_required
- self.assertEqual(format_usage(), textwrap.dedent(expected_usage))
- def test_help_when_not_required(self):
- format_help = self.get_parser(required=False).format_help
- help = self.usage_when_not_required + self.help
- self.assertEqual(format_help(), textwrap.dedent(help))
- def test_help_when_required(self):
- format_help = self.get_parser(required=True).format_help
- help = self.usage_when_required + self.help
- self.assertEqual(format_help(), textwrap.dedent(help))
- class TestMutuallyExclusiveSimple(MEMixin, TestCase):
- def get_parser(self, required=None):
- parser = ErrorRaisingArgumentParser(prog='PROG')
- group = parser.add_mutually_exclusive_group(required=required)
- group.add_argument('--bar', help='bar help')
- group.add_argument('--baz', nargs='?', const='Z', help='baz help')
- return parser
- failures = ['--bar X --baz Y', '--bar X --baz']
- successes = [
- ('--bar X', NS(bar='X', baz=None)),
- ('--bar X --bar Z', NS(bar='Z', baz=None)),
- ('--baz Y', NS(bar=None, baz='Y')),
- ('--baz', NS(bar=None, baz='Z')),
- ]
- successes_when_not_required = [
- ('', NS(bar=None, baz=None)),
- ]
- usage_when_not_required = '''\
- usage: PROG [-h] [--bar BAR | --baz [BAZ]]
- '''
- usage_when_required = '''\
- usage: PROG [-h] (--bar BAR | --baz [BAZ])
- '''
- help = '''\
- options:
- -h, --help show this help message and exit
- --bar BAR bar help
- --baz [BAZ] baz help
- '''
- class TestMutuallyExclusiveLong(MEMixin, TestCase):
- def get_parser(self, required=None):
- parser = ErrorRaisingArgumentParser(prog='PROG')
- parser.add_argument('--abcde', help='abcde help')
- parser.add_argument('--fghij', help='fghij help')
- group = parser.add_mutually_exclusive_group(required=required)
- group.add_argument('--klmno', help='klmno help')
- group.add_argument('--pqrst', help='pqrst help')
- return parser
- failures = ['--klmno X --pqrst Y']
- successes = [
- ('--klmno X', NS(abcde=None, fghij=None, klmno='X', pqrst=None)),
- ('--abcde Y --klmno X',
- NS(abcde='Y', fghij=None, klmno='X', pqrst=None)),
- ('--pqrst X', NS(abcde=None, fghij=None, klmno=None, pqrst='X')),
- ('--pqrst X --fghij Y',
- NS(abcde=None, fghij='Y', klmno=None, pqrst='X')),
- ]
- successes_when_not_required = [
- ('', NS(abcde=None, fghij=None, klmno=None, pqrst=None)),
- ]
- usage_when_not_required = '''\
- usage: PROG [-h] [--abcde ABCDE] [--fghij FGHIJ]
- [--klmno KLMNO | --pqrst PQRST]
- '''
- usage_when_required = '''\
- usage: PROG [-h] [--abcde ABCDE] [--fghij FGHIJ]
- (--klmno KLMNO | --pqrst PQRST)
- '''
- help = '''\
- options:
- -h, --help show this help message and exit
- --abcde ABCDE abcde help
- --fghij FGHIJ fghij help
- --klmno KLMNO klmno help
- --pqrst PQRST pqrst help
- '''
- class TestMutuallyExclusiveFirstSuppressed(MEMixin, TestCase):
- def get_parser(self, required):
- parser = ErrorRaisingArgumentParser(prog='PROG')
- group = parser.add_mutually_exclusive_group(required=required)
- group.add_argument('-x', help=argparse.SUPPRESS)
- group.add_argument('-y', action='store_false', help='y help')
- return parser
- failures = ['-x X -y']
- successes = [
- ('-x X', NS(x='X', y=True)),
- ('-x X -x Y', NS(x='Y', y=True)),
- ('-y', NS(x=None, y=False)),
- ]
- successes_when_not_required = [
- ('', NS(x=None, y=True)),
- ]
- usage_when_not_required = '''\
- usage: PROG [-h] [-y]
- '''
- usage_when_required = '''\
- usage: PROG [-h] -y
- '''
- help = '''\
- options:
- -h, --help show this help message and exit
- -y y help
- '''
- class TestMutuallyExclusiveManySuppressed(MEMixin, TestCase):
- def get_parser(self, required):
- parser = ErrorRaisingArgumentParser(prog='PROG')
- group = parser.add_mutually_exclusive_group(required=required)
- add = group.add_argument
- add('--spam', action='store_true', help=argparse.SUPPRESS)
- add('--badger', action='store_false', help=argparse.SUPPRESS)
- add('--bladder', help=argparse.SUPPRESS)
- return parser
- failures = [
- '--spam --badger',
- '--badger --bladder B',
- '--bladder B --spam',
- ]
- successes = [
- ('--spam', NS(spam=True, badger=True, bladder=None)),
- ('--badger', NS(spam=False, badger=False, bladder=None)),
- ('--bladder B', NS(spam=False, badger=True, bladder='B')),
- ('--spam --spam', NS(spam=True, badger=True, bladder=None)),
- ]
- successes_when_not_required = [
- ('', NS(spam=False, badger=True, bladder=None)),
- ]
- usage_when_required = usage_when_not_required = '''\
- usage: PROG [-h]
- '''
- help = '''\
- options:
- -h, --help show this help message and exit
- '''
- class TestMutuallyExclusiveOptionalAndPositional(MEMixin, TestCase):
- def get_parser(self, required):
- parser = ErrorRaisingArgumentParser(prog='PROG')
- group = parser.add_mutually_exclusive_group(required=required)
- group.add_argument('--foo', action='store_true', help='FOO')
- group.add_argument('--spam', help='SPAM')
- group.add_argument('badger', nargs='*', default='X', help='BADGER')
- return parser
- failures = [
- '--foo --spam S',
- '--spam S X',
- 'X --foo',
- 'X Y Z --spam S',
- '--foo X Y',
- ]
- successes = [
- ('--foo', NS(foo=True, spam=None, badger='X')),
- ('--spam S', NS(foo=False, spam='S', badger='X')),
- ('X', NS(foo=False, spam=None, badger=['X'])),
- ('X Y Z', NS(foo=False, spam=None, badger=['X', 'Y', 'Z'])),
- ]
- successes_when_not_required = [
- ('', NS(foo=False, spam=None, badger='X')),
- ]
- usage_when_not_required = '''\
- usage: PROG [-h] [--foo | --spam SPAM | badger ...]
- '''
- usage_when_required = '''\
- usage: PROG [-h] (--foo | --spam SPAM | badger ...)
- '''
- help = '''\
- positional arguments:
- badger BADGER
- options:
- -h, --help show this help message and exit
- --foo FOO
- --spam SPAM SPAM
- '''
- class TestMutuallyExclusiveOptionalsMixed(MEMixin, TestCase):
- def get_parser(self, required):
- parser = ErrorRaisingArgumentParser(prog='PROG')
- parser.add_argument('-x', action='store_true', help='x help')
- group = parser.add_mutually_exclusive_group(required=required)
- group.add_argument('-a', action='store_true', help='a help')
- group.add_argument('-b', action='store_true', help='b help')
- parser.add_argument('-y', action='store_true', help='y help')
- group.add_argument('-c', action='store_true', help='c help')
- return parser
- failures = ['-a -b', '-b -c', '-a -c', '-a -b -c']
- successes = [
- ('-a', NS(a=True, b=False, c=False, x=False, y=False)),
- ('-b', NS(a=False, b=True, c=False, x=False, y=False)),
- ('-c', NS(a=False, b=False, c=True, x=False, y=False)),
- ('-a -x', NS(a=True, b=False, c=False, x=True, y=False)),
- ('-y -b', NS(a=False, b=True, c=False, x=False, y=True)),
- ('-x -y -c', NS(a=False, b=False, c=True, x=True, y=True)),
- ]
- successes_when_not_required = [
- ('', NS(a=False, b=False, c=False, x=False, y=False)),
- ('-x', NS(a=False, b=False, c=False, x=True, y=False)),
- ('-y', NS(a=False, b=False, c=False, x=False, y=True)),
- ]
- usage_when_required = usage_when_not_required = '''\
- usage: PROG [-h] [-x] [-a] [-b] [-y] [-c]
- '''
- help = '''\
- options:
- -h, --help show this help message and exit
- -x x help
- -a a help
- -b b help
- -y y help
- -c c help
- '''
- class TestMutuallyExclusiveInGroup(MEMixin, TestCase):
- def get_parser(self, required=None):
- parser = ErrorRaisingArgumentParser(prog='PROG')
- titled_group = parser.add_argument_group(
- title='Titled group', description='Group description')
- mutex_group = \
- titled_group.add_mutually_exclusive_group(required=required)
- mutex_group.add_argument('--bar', help='bar help')
- mutex_group.add_argument('--baz', help='baz help')
- return parser
- failures = ['--bar X --baz Y', '--baz X --bar Y']
- successes = [
- ('--bar X', NS(bar='X', baz=None)),
- ('--baz Y', NS(bar=None, baz='Y')),
- ]
- successes_when_not_required = [
- ('', NS(bar=None, baz=None)),
- ]
- usage_when_not_required = '''\
- usage: PROG [-h] [--bar BAR | --baz BAZ]
- '''
- usage_when_required = '''\
- usage: PROG [-h] (--bar BAR | --baz BAZ)
- '''
- help = '''\
- options:
- -h, --help show this help message and exit
- Titled group:
- Group description
- --bar BAR bar help
- --baz BAZ baz help
- '''
- class TestMutuallyExclusiveOptionalsAndPositionalsMixed(MEMixin, TestCase):
- def get_parser(self, required):
- parser = ErrorRaisingArgumentParser(prog='PROG')
- parser.add_argument('x', help='x help')
- parser.add_argument('-y', action='store_true', help='y help')
- group = parser.add_mutually_exclusive_group(required=required)
- group.add_argument('a', nargs='?', help='a help')
- group.add_argument('-b', action='store_true', help='b help')
- group.add_argument('-c', action='store_true', help='c help')
- return parser
- failures = ['X A -b', '-b -c', '-c X A']
- successes = [
- ('X A', NS(a='A', b=False, c=False, x='X', y=False)),
- ('X -b', NS(a=None, b=True, c=False, x='X', y=False)),
- ('X -c', NS(a=None, b=False, c=True, x='X', y=False)),
- ('X A -y', NS(a='A', b=False, c=False, x='X', y=True)),
- ('X -y -b', NS(a=None, b=True, c=False, x='X', y=True)),
- ]
- successes_when_not_required = [
- ('X', NS(a=None, b=False, c=False, x='X', y=False)),
- ('X -y', NS(a=None, b=False, c=False, x='X', y=True)),
- ]
- usage_when_required = usage_when_not_required = '''\
- usage: PROG [-h] [-y] [-b] [-c] x [a]
- '''
- help = '''\
- positional arguments:
- x x help
- a a help
- options:
- -h, --help show this help message and exit
- -y y help
- -b b help
- -c c help
- '''
- class TestMutuallyExclusiveNested(MEMixin, TestCase):
- # Nesting mutually exclusive groups is an undocumented feature
- # that came about by accident through inheritance and has been
- # the source of many bugs. It is deprecated and this test should
- # eventually be removed along with it.
- def get_parser(self, required):
- parser = ErrorRaisingArgumentParser(prog='PROG')
- group = parser.add_mutually_exclusive_group(required=required)
- group.add_argument('-a')
- group.add_argument('-b')
- with warnings.catch_warnings():
- warnings.simplefilter('ignore', DeprecationWarning)
- group2 = group.add_mutually_exclusive_group(required=required)
- group2.add_argument('-c')
- group2.add_argument('-d')
- with warnings.catch_warnings():
- warnings.simplefilter('ignore', DeprecationWarning)
- group3 = group2.add_mutually_exclusive_group(required=required)
- group3.add_argument('-e')
- group3.add_argument('-f')
- return parser
- usage_when_not_required = '''\
- usage: PROG [-h] [-a A | -b B | [-c C | -d D | [-e E | -f F]]]
- '''
- usage_when_required = '''\
- usage: PROG [-h] (-a A | -b B | (-c C | -d D | (-e E | -f F)))
- '''
- help = '''\
- options:
- -h, --help show this help message and exit
- -a A
- -b B
- -c C
- -d D
- -e E
- -f F
- '''
- # We are only interested in testing the behavior of format_usage().
- test_failures_when_not_required = None
- test_failures_when_required = None
- test_successes_when_not_required = None
- test_successes_when_required = None
- # =================================================
- # Mutually exclusive group in parent parser tests
- # =================================================
- class MEPBase(object):
- def get_parser(self, required=None):
- parent = super(MEPBase, self).get_parser(required=required)
- parser = ErrorRaisingArgumentParser(
- prog=parent.prog, add_help=False, parents=[parent])
- return parser
- class TestMutuallyExclusiveGroupErrorsParent(
- MEPBase, TestMutuallyExclusiveGroupErrors):
- pass
- class TestMutuallyExclusiveSimpleParent(
- MEPBase, TestMutuallyExclusiveSimple):
- pass
- class TestMutuallyExclusiveLongParent(
- MEPBase, TestMutuallyExclusiveLong):
- pass
- class TestMutuallyExclusiveFirstSuppressedParent(
- MEPBase, TestMutuallyExclusiveFirstSuppressed):
- pass
- class TestMutuallyExclusiveManySuppressedParent(
- MEPBase, TestMutuallyExclusiveManySuppressed):
- pass
- class TestMutuallyExclusiveOptionalAndPositionalParent(
- MEPBase, TestMutuallyExclusiveOptionalAndPositional):
- pass
- class TestMutuallyExclusiveOptionalsMixedParent(
- MEPBase, TestMutuallyExclusiveOptionalsMixed):
- pass
- class TestMutuallyExclusiveOptionalsAndPositionalsMixedParent(
- MEPBase, TestMutuallyExclusiveOptionalsAndPositionalsMixed):
- pass
- # =================
- # Set default tests
- # =================
- class TestSetDefaults(TestCase):
- def test_set_defaults_no_args(self):
- parser = ErrorRaisingArgumentParser()
- parser.set_defaults(x='foo')
- parser.set_defaults(y='bar', z=1)
- self.assertEqual(NS(x='foo', y='bar', z=1),
- parser.parse_args([]))
- self.assertEqual(NS(x='foo', y='bar', z=1),
- parser.parse_args([], NS()))
- self.assertEqual(NS(x='baz', y='bar', z=1),
- parser.parse_args([], NS(x='baz')))
- self.assertEqual(NS(x='baz', y='bar', z=2),
- parser.parse_args([], NS(x='baz', z=2)))
- def test_set_defaults_with_args(self):
- parser = ErrorRaisingArgumentParser()
- parser.set_defaults(x='foo', y='bar')
- parser.add_argument('-x', default='xfoox')
- self.assertEqual(NS(x='xfoox', y='bar'),
- parser.parse_args([]))
- self.assertEqual(NS(x='xfoox', y='bar'),
- parser.parse_args([], NS()))
- self.assertEqual(NS(x='baz', y='bar'),
- parser.parse_args([], NS(x='baz')))
- self.assertEqual(NS(x='1', y='bar'),
- parser.parse_args('-x 1'.split()))
- self.assertEqual(NS(x='1', y='bar'),
- parser.parse_args('-x 1'.split(), NS()))
- self.assertEqual(NS(x='1', y='bar'),
- parser.parse_args('-x 1'.split(), NS(x='baz')))
- def test_set_defaults_subparsers(self):
- parser = ErrorRaisingArgumentParser()
- parser.set_defaults(x='foo')
- subparsers = parser.add_subparsers()
- parser_a = subparsers.add_parser('a')
- parser_a.set_defaults(y='bar')
- self.assertEqual(NS(x='foo', y='bar'),
- parser.parse_args('a'.split()))
- def test_set_defaults_parents(self):
- parent = ErrorRaisingArgumentParser(add_help=False)
- parent.set_defaults(x='foo')
- parser = ErrorRaisingArgumentParser(parents=[parent])
- self.assertEqual(NS(x='foo'), parser.parse_args([]))
- def test_set_defaults_on_parent_and_subparser(self):
- parser = argparse.ArgumentParser()
- xparser = parser.add_subparsers().add_parser('X')
- parser.set_defaults(foo=1)
- xparser.set_defaults(foo=2)
- self.assertEqual(NS(foo=2), parser.parse_args(['X']))
- def test_set_defaults_same_as_add_argument(self):
- parser = ErrorRaisingArgumentParser()
- parser.set_defaults(w='W', x='X', y='Y', z='Z')
- parser.add_argument('-w')
- parser.add_argument('-x', default='XX')
- parser.add_argument('y', nargs='?')
- parser.add_argument('z', nargs='?', default='ZZ')
- # defaults set previously
- self.assertEqual(NS(w='W', x='XX', y='Y', z='ZZ'),
- parser.parse_args([]))
- # reset defaults
- parser.set_defaults(w='WW', x='X', y='YY', z='Z')
- self.assertEqual(NS(w='WW', x='X', y='YY', z='Z'),
- parser.parse_args([]))
- def test_set_defaults_same_as_add_argument_group(self):
- parser = ErrorRaisingArgumentParser()
- parser.set_defaults(w='W', x='X', y='Y', z='Z')
- group = parser.add_argument_group('foo')
- group.add_argument('-w')
- group.add_argument('-x', default='XX')
- group.add_argument('y', nargs='?')
- group.add_argument('z', nargs='?', default='ZZ')
- # defaults set previously
- self.assertEqual(NS(w='W', x='XX', y='Y', z='ZZ'),
- parser.parse_args([]))
- # reset defaults
- parser.set_defaults(w='WW', x='X', y='YY', z='Z')
- self.assertEqual(NS(w='WW', x='X', y='YY', z='Z'),
- parser.parse_args([]))
- # =================
- # Get default tests
- # =================
- class TestGetDefault(TestCase):
- def test_get_default(self):
- parser = ErrorRaisingArgumentParser()
- self.assertIsNone(parser.get_default("foo"))
- self.assertIsNone(parser.get_default("bar"))
- parser.add_argument("--foo")
- self.assertIsNone(parser.get_default("foo"))
- self.assertIsNone(parser.get_default("bar"))
- parser.add_argument("--bar", type=int, default=42)
- self.assertIsNone(parser.get_default("foo"))
- self.assertEqual(42, parser.get_default("bar"))
- parser.set_defaults(foo="badger")
- self.assertEqual("badger", parser.get_default("foo"))
- self.assertEqual(42, parser.get_default("bar"))
- # ==========================
- # Namespace 'contains' tests
- # ==========================
- class TestNamespaceContainsSimple(TestCase):
- def test_empty(self):
- ns = argparse.Namespace()
- self.assertNotIn('', ns)
- self.assertNotIn('x', ns)
- def test_non_empty(self):
- ns = argparse.Namespace(x=1, y=2)
- self.assertNotIn('', ns)
- self.assertIn('x', ns)
- self.assertIn('y', ns)
- self.assertNotIn('xx', ns)
- self.assertNotIn('z', ns)
- # =====================
- # Help formatting tests
- # =====================
- class TestHelpFormattingMetaclass(type):
- def __init__(cls, name, bases, bodydict):
- if name == 'HelpTestCase':
- return
- class AddTests(object):
- def __init__(self, test_class, func_suffix, std_name):
- self.func_suffix = func_suffix
- self.std_name = std_name
- for test_func in [self.test_format,
- self.test_print,
- self.test_print_file]:
- test_name = '%s_%s' % (test_func.__name__, func_suffix)
- def test_wrapper(self, test_func=test_func):
- test_func(self)
- try:
- test_wrapper.__name__ = test_name
- except TypeError:
- pass
- setattr(test_class, test_name, test_wrapper)
- def _get_parser(self, tester):
- parser = argparse.ArgumentParser(
- *tester.parser_signature.args,
- **tester.parser_signature.kwargs)
- for argument_sig in getattr(tester, 'argument_signatures', []):
- parser.add_argument(*argument_sig.args,
- **argument_sig.kwargs)
- group_sigs = getattr(tester, 'argument_group_signatures', [])
- for group_sig, argument_sigs in group_sigs:
- group = parser.add_argument_group(*group_sig.args,
- **group_sig.kwargs)
- for argument_sig in argument_sigs:
- group.add_argument(*argument_sig.args,
- **argument_sig.kwargs)
- subparsers_sigs = getattr(tester, 'subparsers_signatures', [])
- if subparsers_sigs:
- subparsers = parser.add_subparsers()
- for subparser_sig in subparsers_sigs:
- subparsers.add_parser(*subparser_sig.args,
- **subparser_sig.kwargs)
- return parser
- def _test(self, tester, parser_text):
- expected_text = getattr(tester, self.func_suffix)
- expected_text = textwrap.dedent(expected_text)
- tester.maxDiff = None
- tester.assertEqual(expected_text, parser_text)
- def test_format(self, tester):
- parser = self._get_parser(tester)
- format = getattr(parser, 'format_%s' % self.func_suffix)
- self._test(tester, format())
- def test_print(self, tester):
- parser = self._get_parser(tester)
- print_ = getattr(parser, 'print_%s' % self.func_suffix)
- old_stream = getattr(sys, self.std_name)
- setattr(sys, self.std_name, StdIOBuffer())
- try:
- print_()
- parser_text = getattr(sys, self.std_name).getvalue()
- finally:
- setattr(sys, self.std_name, old_stream)
- self._test(tester, parser_text)
- def test_print_file(self, tester):
- parser = self._get_parser(tester)
- print_ = getattr(parser, 'print_%s' % self.func_suffix)
- sfile = StdIOBuffer()
- print_(sfile)
- parser_text = sfile.getvalue()
- self._test(tester, parser_text)
- # add tests for {format,print}_{usage,help}
- for func_suffix, std_name in [('usage', 'stdout'),
- ('help', 'stdout')]:
- AddTests(cls, func_suffix, std_name)
- bases = TestCase,
- HelpTestCase = TestHelpFormattingMetaclass('HelpTestCase', bases, {})
- class TestHelpBiggerOptionals(HelpTestCase):
- """Make sure that argument help aligns when options are longer"""
- parser_signature = Sig(prog='PROG', description='DESCRIPTION',
- epilog='EPILOG')
- argument_signatures = [
- Sig('-v', '--version', action='version', version='0.1'),
- Sig('-x', action='store_true', help='X HELP'),
- Sig('--y', help='Y HELP'),
- Sig('foo', help='FOO HELP'),
- Sig('bar', help='BAR HELP'),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] [-v] [-x] [--y Y] foo bar
- '''
- help = usage + '''\
- DESCRIPTION
- positional arguments:
- foo FOO HELP
- bar BAR HELP
- options:
- -h, --help show this help message and exit
- -v, --version show program's version number and exit
- -x X HELP
- --y Y Y HELP
- EPILOG
- '''
- version = '''\
- 0.1
- '''
- class TestShortColumns(HelpTestCase):
- '''Test extremely small number of columns.
- TestCase prevents "COLUMNS" from being too small in the tests themselves,
- but we don't want any exceptions thrown in such cases. Only ugly representation.
- '''
- def setUp(self):
- env = self.enterContext(os_helper.EnvironmentVarGuard())
- env.set("COLUMNS", '15')
- parser_signature = TestHelpBiggerOptionals.parser_signature
- argument_signatures = TestHelpBiggerOptionals.argument_signatures
- argument_group_signatures = TestHelpBiggerOptionals.argument_group_signatures
- usage = '''\
- usage: PROG
- [-h]
- [-v]
- [-x]
- [--y Y]
- foo
- bar
- '''
- help = usage + '''\
- DESCRIPTION
- positional arguments:
- foo
- FOO HELP
- bar
- BAR HELP
- options:
- -h, --help
- show this
- help
- message and
- exit
- -v, --version
- show
- program's
- version
- number and
- exit
- -x
- X HELP
- --y Y
- Y HELP
- EPILOG
- '''
- version = TestHelpBiggerOptionals.version
- class TestHelpBiggerOptionalGroups(HelpTestCase):
- """Make sure that argument help aligns when options are longer"""
- parser_signature = Sig(prog='PROG', description='DESCRIPTION',
- epilog='EPILOG')
- argument_signatures = [
- Sig('-v', '--version', action='version', version='0.1'),
- Sig('-x', action='store_true', help='X HELP'),
- Sig('--y', help='Y HELP'),
- Sig('foo', help='FOO HELP'),
- Sig('bar', help='BAR HELP'),
- ]
- argument_group_signatures = [
- (Sig('GROUP TITLE', description='GROUP DESCRIPTION'), [
- Sig('baz', help='BAZ HELP'),
- Sig('-z', nargs='+', help='Z HELP')]),
- ]
- usage = '''\
- usage: PROG [-h] [-v] [-x] [--y Y] [-z Z [Z ...]] foo bar baz
- '''
- help = usage + '''\
- DESCRIPTION
- positional arguments:
- foo FOO HELP
- bar BAR HELP
- options:
- -h, --help show this help message and exit
- -v, --version show program's version number and exit
- -x X HELP
- --y Y Y HELP
- GROUP TITLE:
- GROUP DESCRIPTION
- baz BAZ HELP
- -z Z [Z ...] Z HELP
- EPILOG
- '''
- version = '''\
- 0.1
- '''
- class TestHelpBiggerPositionals(HelpTestCase):
- """Make sure that help aligns when arguments are longer"""
- parser_signature = Sig(usage='USAGE', description='DESCRIPTION')
- argument_signatures = [
- Sig('-x', action='store_true', help='X HELP'),
- Sig('--y', help='Y HELP'),
- Sig('ekiekiekifekang', help='EKI HELP'),
- Sig('bar', help='BAR HELP'),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: USAGE
- '''
- help = usage + '''\
- DESCRIPTION
- positional arguments:
- ekiekiekifekang EKI HELP
- bar BAR HELP
- options:
- -h, --help show this help message and exit
- -x X HELP
- --y Y Y HELP
- '''
- version = ''
- class TestHelpReformatting(HelpTestCase):
- """Make sure that text after short names starts on the first line"""
- parser_signature = Sig(
- prog='PROG',
- description=' oddly formatted\n'
- 'description\n'
- '\n'
- 'that is so long that it should go onto multiple '
- 'lines when wrapped')
- argument_signatures = [
- Sig('-x', metavar='XX', help='oddly\n'
- ' formatted -x help'),
- Sig('y', metavar='yyy', help='normal y help'),
- ]
- argument_group_signatures = [
- (Sig('title', description='\n'
- ' oddly formatted group\n'
- '\n'
- 'description'),
- [Sig('-a', action='store_true',
- help=' oddly \n'
- 'formatted -a help \n'
- ' again, so long that it should be wrapped over '
- 'multiple lines')]),
- ]
- usage = '''\
- usage: PROG [-h] [-x XX] [-a] yyy
- '''
- help = usage + '''\
- oddly formatted description that is so long that it should go onto \
- multiple
- lines when wrapped
- positional arguments:
- yyy normal y help
- options:
- -h, --help show this help message and exit
- -x XX oddly formatted -x help
- title:
- oddly formatted group description
- -a oddly formatted -a help again, so long that it should \
- be wrapped
- over multiple lines
- '''
- version = ''
- class TestHelpWrappingShortNames(HelpTestCase):
- """Make sure that text after short names starts on the first line"""
- parser_signature = Sig(prog='PROG', description= 'D\nD' * 30)
- argument_signatures = [
- Sig('-x', metavar='XX', help='XHH HX' * 20),
- Sig('y', metavar='yyy', help='YH YH' * 20),
- ]
- argument_group_signatures = [
- (Sig('ALPHAS'), [
- Sig('-a', action='store_true', help='AHHH HHA' * 10)]),
- ]
- usage = '''\
- usage: PROG [-h] [-x XX] [-a] yyy
- '''
- help = usage + '''\
- D DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD \
- DD DD DD
- DD DD DD DD D
- positional arguments:
- yyy YH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH \
- YHYH YHYH
- YHYH YHYH YHYH YHYH YHYH YHYH YHYH YH
- options:
- -h, --help show this help message and exit
- -x XX XHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH \
- HXXHH HXXHH
- HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HXXHH HX
- ALPHAS:
- -a AHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH HHAAHHH \
- HHAAHHH
- HHAAHHH HHAAHHH HHA
- '''
- version = ''
- class TestHelpWrappingLongNames(HelpTestCase):
- """Make sure that text after long names starts on the next line"""
- parser_signature = Sig(usage='USAGE', description= 'D D' * 30)
- argument_signatures = [
- Sig('-v', '--version', action='version', version='V V' * 30),
- Sig('-x', metavar='X' * 25, help='XH XH' * 20),
- Sig('y', metavar='y' * 25, help='YH YH' * 20),
- ]
- argument_group_signatures = [
- (Sig('ALPHAS'), [
- Sig('-a', metavar='A' * 25, help='AH AH' * 20),
- Sig('z', metavar='z' * 25, help='ZH ZH' * 20)]),
- ]
- usage = '''\
- usage: USAGE
- '''
- help = usage + '''\
- D DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD DD \
- DD DD DD
- DD DD DD DD D
- positional arguments:
- yyyyyyyyyyyyyyyyyyyyyyyyy
- YH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH \
- YHYH YHYH
- YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YHYH YH
- options:
- -h, --help show this help message and exit
- -v, --version show program's version number and exit
- -x XXXXXXXXXXXXXXXXXXXXXXXXX
- XH XHXH XHXH XHXH XHXH XHXH XHXH XHXH XHXH \
- XHXH XHXH
- XHXH XHXH XHXH XHXH XHXH XHXH XHXH XHXH XHXH XH
- ALPHAS:
- -a AAAAAAAAAAAAAAAAAAAAAAAAA
- AH AHAH AHAH AHAH AHAH AHAH AHAH AHAH AHAH \
- AHAH AHAH
- AHAH AHAH AHAH AHAH AHAH AHAH AHAH AHAH AHAH AH
- zzzzzzzzzzzzzzzzzzzzzzzzz
- ZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH \
- ZHZH ZHZH
- ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZHZH ZH
- '''
- version = '''\
- V VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV VV \
- VV VV VV
- VV VV VV VV V
- '''
- class TestHelpUsage(HelpTestCase):
- """Test basic usage messages"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('-w', nargs='+', help='w'),
- Sig('-x', nargs='*', help='x'),
- Sig('a', help='a'),
- Sig('b', help='b', nargs=2),
- Sig('c', help='c', nargs='?'),
- Sig('--foo', help='Whether to foo', action=argparse.BooleanOptionalAction),
- Sig('--bar', help='Whether to bar', default=True,
- action=argparse.BooleanOptionalAction),
- Sig('-f', '--foobar', '--barfoo', action=argparse.BooleanOptionalAction),
- Sig('--bazz', action=argparse.BooleanOptionalAction,
- default=argparse.SUPPRESS, help='Bazz!'),
- ]
- argument_group_signatures = [
- (Sig('group'), [
- Sig('-y', nargs='?', help='y'),
- Sig('-z', nargs=3, help='z'),
- Sig('d', help='d', nargs='*'),
- Sig('e', help='e', nargs='+'),
- ])
- ]
- usage = '''\
- usage: PROG [-h] [-w W [W ...]] [-x [X ...]] [--foo | --no-foo]
- [--bar | --no-bar]
- [-f | --foobar | --no-foobar | --barfoo | --no-barfoo]
- [--bazz | --no-bazz] [-y [Y]] [-z Z Z Z]
- a b b [c] [d ...] e [e ...]
- '''
- help = usage + '''\
- positional arguments:
- a a
- b b
- c c
- options:
- -h, --help show this help message and exit
- -w W [W ...] w
- -x [X ...] x
- --foo, --no-foo Whether to foo
- --bar, --no-bar Whether to bar
- -f, --foobar, --no-foobar, --barfoo, --no-barfoo
- --bazz, --no-bazz Bazz!
- group:
- -y [Y] y
- -z Z Z Z z
- d d
- e e
- '''
- version = ''
- class TestHelpOnlyUserGroups(HelpTestCase):
- """Test basic usage messages"""
- parser_signature = Sig(prog='PROG', add_help=False)
- argument_signatures = []
- argument_group_signatures = [
- (Sig('xxxx'), [
- Sig('-x', help='x'),
- Sig('a', help='a'),
- ]),
- (Sig('yyyy'), [
- Sig('b', help='b'),
- Sig('-y', help='y'),
- ]),
- ]
- usage = '''\
- usage: PROG [-x X] [-y Y] a b
- '''
- help = usage + '''\
- xxxx:
- -x X x
- a a
- yyyy:
- b b
- -y Y y
- '''
- version = ''
- class TestHelpUsageLongProg(HelpTestCase):
- """Test usage messages where the prog is long"""
- parser_signature = Sig(prog='P' * 60)
- argument_signatures = [
- Sig('-w', metavar='W'),
- Sig('-x', metavar='X'),
- Sig('a'),
- Sig('b'),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPP
- [-h] [-w W] [-x X] a b
- '''
- help = usage + '''\
- positional arguments:
- a
- b
- options:
- -h, --help show this help message and exit
- -w W
- -x X
- '''
- version = ''
- class TestHelpUsageLongProgOptionsWrap(HelpTestCase):
- """Test usage messages where the prog is long and the optionals wrap"""
- parser_signature = Sig(prog='P' * 60)
- argument_signatures = [
- Sig('-w', metavar='W' * 25),
- Sig('-x', metavar='X' * 25),
- Sig('-y', metavar='Y' * 25),
- Sig('-z', metavar='Z' * 25),
- Sig('a'),
- Sig('b'),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPP
- [-h] [-w WWWWWWWWWWWWWWWWWWWWWWWWW] \
- [-x XXXXXXXXXXXXXXXXXXXXXXXXX]
- [-y YYYYYYYYYYYYYYYYYYYYYYYYY] [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
- a b
- '''
- help = usage + '''\
- positional arguments:
- a
- b
- options:
- -h, --help show this help message and exit
- -w WWWWWWWWWWWWWWWWWWWWWWWWW
- -x XXXXXXXXXXXXXXXXXXXXXXXXX
- -y YYYYYYYYYYYYYYYYYYYYYYYYY
- -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
- '''
- version = ''
- class TestHelpUsageLongProgPositionalsWrap(HelpTestCase):
- """Test usage messages where the prog is long and the positionals wrap"""
- parser_signature = Sig(prog='P' * 60, add_help=False)
- argument_signatures = [
- Sig('a' * 25),
- Sig('b' * 25),
- Sig('c' * 25),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPP
- aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
- ccccccccccccccccccccccccc
- '''
- help = usage + '''\
- positional arguments:
- aaaaaaaaaaaaaaaaaaaaaaaaa
- bbbbbbbbbbbbbbbbbbbbbbbbb
- ccccccccccccccccccccccccc
- '''
- version = ''
- class TestHelpUsageOptionalsWrap(HelpTestCase):
- """Test usage messages where the optionals wrap"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('-w', metavar='W' * 25),
- Sig('-x', metavar='X' * 25),
- Sig('-y', metavar='Y' * 25),
- Sig('-z', metavar='Z' * 25),
- Sig('a'),
- Sig('b'),
- Sig('c'),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] [-w WWWWWWWWWWWWWWWWWWWWWWWWW] \
- [-x XXXXXXXXXXXXXXXXXXXXXXXXX]
- [-y YYYYYYYYYYYYYYYYYYYYYYYYY] \
- [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
- a b c
- '''
- help = usage + '''\
- positional arguments:
- a
- b
- c
- options:
- -h, --help show this help message and exit
- -w WWWWWWWWWWWWWWWWWWWWWWWWW
- -x XXXXXXXXXXXXXXXXXXXXXXXXX
- -y YYYYYYYYYYYYYYYYYYYYYYYYY
- -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
- '''
- version = ''
- class TestHelpUsagePositionalsWrap(HelpTestCase):
- """Test usage messages where the positionals wrap"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('-x'),
- Sig('-y'),
- Sig('-z'),
- Sig('a' * 25),
- Sig('b' * 25),
- Sig('c' * 25),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] [-x X] [-y Y] [-z Z]
- aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
- ccccccccccccccccccccccccc
- '''
- help = usage + '''\
- positional arguments:
- aaaaaaaaaaaaaaaaaaaaaaaaa
- bbbbbbbbbbbbbbbbbbbbbbbbb
- ccccccccccccccccccccccccc
- options:
- -h, --help show this help message and exit
- -x X
- -y Y
- -z Z
- '''
- version = ''
- class TestHelpUsageOptionalsPositionalsWrap(HelpTestCase):
- """Test usage messages where the optionals and positionals wrap"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('-x', metavar='X' * 25),
- Sig('-y', metavar='Y' * 25),
- Sig('-z', metavar='Z' * 25),
- Sig('a' * 25),
- Sig('b' * 25),
- Sig('c' * 25),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] [-x XXXXXXXXXXXXXXXXXXXXXXXXX] \
- [-y YYYYYYYYYYYYYYYYYYYYYYYYY]
- [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
- aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
- ccccccccccccccccccccccccc
- '''
- help = usage + '''\
- positional arguments:
- aaaaaaaaaaaaaaaaaaaaaaaaa
- bbbbbbbbbbbbbbbbbbbbbbbbb
- ccccccccccccccccccccccccc
- options:
- -h, --help show this help message and exit
- -x XXXXXXXXXXXXXXXXXXXXXXXXX
- -y YYYYYYYYYYYYYYYYYYYYYYYYY
- -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
- '''
- version = ''
- class TestHelpUsageOptionalsOnlyWrap(HelpTestCase):
- """Test usage messages where there are only optionals and they wrap"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('-x', metavar='X' * 25),
- Sig('-y', metavar='Y' * 25),
- Sig('-z', metavar='Z' * 25),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] [-x XXXXXXXXXXXXXXXXXXXXXXXXX] \
- [-y YYYYYYYYYYYYYYYYYYYYYYYYY]
- [-z ZZZZZZZZZZZZZZZZZZZZZZZZZ]
- '''
- help = usage + '''\
- options:
- -h, --help show this help message and exit
- -x XXXXXXXXXXXXXXXXXXXXXXXXX
- -y YYYYYYYYYYYYYYYYYYYYYYYYY
- -z ZZZZZZZZZZZZZZZZZZZZZZZZZ
- '''
- version = ''
- class TestHelpUsagePositionalsOnlyWrap(HelpTestCase):
- """Test usage messages where there are only positionals and they wrap"""
- parser_signature = Sig(prog='PROG', add_help=False)
- argument_signatures = [
- Sig('a' * 25),
- Sig('b' * 25),
- Sig('c' * 25),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG aaaaaaaaaaaaaaaaaaaaaaaaa bbbbbbbbbbbbbbbbbbbbbbbbb
- ccccccccccccccccccccccccc
- '''
- help = usage + '''\
- positional arguments:
- aaaaaaaaaaaaaaaaaaaaaaaaa
- bbbbbbbbbbbbbbbbbbbbbbbbb
- ccccccccccccccccccccccccc
- '''
- version = ''
- class TestHelpVariableExpansion(HelpTestCase):
- """Test that variables are expanded properly in help messages"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('-x', type=int,
- help='x %(prog)s %(default)s %(type)s %%'),
- Sig('-y', action='store_const', default=42, const='XXX',
- help='y %(prog)s %(default)s %(const)s'),
- Sig('--foo', choices='abc',
- help='foo %(prog)s %(default)s %(choices)s'),
- Sig('--bar', default='baz', choices=[1, 2], metavar='BBB',
- help='bar %(prog)s %(default)s %(dest)s'),
- Sig('spam', help='spam %(prog)s %(default)s'),
- Sig('badger', default=0.5, help='badger %(prog)s %(default)s'),
- ]
- argument_group_signatures = [
- (Sig('group'), [
- Sig('-a', help='a %(prog)s %(default)s'),
- Sig('-b', default=-1, help='b %(prog)s %(default)s'),
- ])
- ]
- usage = ('''\
- usage: PROG [-h] [-x X] [-y] [--foo {a,b,c}] [--bar BBB] [-a A] [-b B]
- spam badger
- ''')
- help = usage + '''\
- positional arguments:
- spam spam PROG None
- badger badger PROG 0.5
- options:
- -h, --help show this help message and exit
- -x X x PROG None int %
- -y y PROG 42 XXX
- --foo {a,b,c} foo PROG None a, b, c
- --bar BBB bar PROG baz bar
- group:
- -a A a PROG None
- -b B b PROG -1
- '''
- version = ''
- class TestHelpVariableExpansionUsageSupplied(HelpTestCase):
- """Test that variables are expanded properly when usage= is present"""
- parser_signature = Sig(prog='PROG', usage='%(prog)s FOO')
- argument_signatures = []
- argument_group_signatures = []
- usage = ('''\
- usage: PROG FOO
- ''')
- help = usage + '''\
- options:
- -h, --help show this help message and exit
- '''
- version = ''
- class TestHelpVariableExpansionNoArguments(HelpTestCase):
- """Test that variables are expanded properly with no arguments"""
- parser_signature = Sig(prog='PROG', add_help=False)
- argument_signatures = []
- argument_group_signatures = []
- usage = ('''\
- usage: PROG
- ''')
- help = usage
- version = ''
- class TestHelpSuppressUsage(HelpTestCase):
- """Test that items can be suppressed in usage messages"""
- parser_signature = Sig(prog='PROG', usage=argparse.SUPPRESS)
- argument_signatures = [
- Sig('--foo', help='foo help'),
- Sig('spam', help='spam help'),
- ]
- argument_group_signatures = []
- help = '''\
- positional arguments:
- spam spam help
- options:
- -h, --help show this help message and exit
- --foo FOO foo help
- '''
- usage = ''
- version = ''
- class TestHelpSuppressOptional(HelpTestCase):
- """Test that optional arguments can be suppressed in help messages"""
- parser_signature = Sig(prog='PROG', add_help=False)
- argument_signatures = [
- Sig('--foo', help=argparse.SUPPRESS),
- Sig('spam', help='spam help'),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG spam
- '''
- help = usage + '''\
- positional arguments:
- spam spam help
- '''
- version = ''
- class TestHelpSuppressOptionalGroup(HelpTestCase):
- """Test that optional groups can be suppressed in help messages"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('--foo', help='foo help'),
- Sig('spam', help='spam help'),
- ]
- argument_group_signatures = [
- (Sig('group'), [Sig('--bar', help=argparse.SUPPRESS)]),
- ]
- usage = '''\
- usage: PROG [-h] [--foo FOO] spam
- '''
- help = usage + '''\
- positional arguments:
- spam spam help
- options:
- -h, --help show this help message and exit
- --foo FOO foo help
- '''
- version = ''
- class TestHelpSuppressPositional(HelpTestCase):
- """Test that positional arguments can be suppressed in help messages"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('--foo', help='foo help'),
- Sig('spam', help=argparse.SUPPRESS),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] [--foo FOO]
- '''
- help = usage + '''\
- options:
- -h, --help show this help message and exit
- --foo FOO foo help
- '''
- version = ''
- class TestHelpRequiredOptional(HelpTestCase):
- """Test that required options don't look optional"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('--foo', required=True, help='foo help'),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] --foo FOO
- '''
- help = usage + '''\
- options:
- -h, --help show this help message and exit
- --foo FOO foo help
- '''
- version = ''
- class TestHelpAlternatePrefixChars(HelpTestCase):
- """Test that options display with different prefix characters"""
- parser_signature = Sig(prog='PROG', prefix_chars='^;', add_help=False)
- argument_signatures = [
- Sig('^^foo', action='store_true', help='foo help'),
- Sig(';b', ';;bar', help='bar help'),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [^^foo] [;b BAR]
- '''
- help = usage + '''\
- options:
- ^^foo foo help
- ;b BAR, ;;bar BAR bar help
- '''
- version = ''
- class TestHelpNoHelpOptional(HelpTestCase):
- """Test that the --help argument can be suppressed help messages"""
- parser_signature = Sig(prog='PROG', add_help=False)
- argument_signatures = [
- Sig('--foo', help='foo help'),
- Sig('spam', help='spam help'),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [--foo FOO] spam
- '''
- help = usage + '''\
- positional arguments:
- spam spam help
- options:
- --foo FOO foo help
- '''
- version = ''
- class TestHelpNone(HelpTestCase):
- """Test that no errors occur if no help is specified"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('--foo'),
- Sig('spam'),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] [--foo FOO] spam
- '''
- help = usage + '''\
- positional arguments:
- spam
- options:
- -h, --help show this help message and exit
- --foo FOO
- '''
- version = ''
- class TestHelpTupleMetavar(HelpTestCase):
- """Test specifying metavar as a tuple"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('-w', help='w', nargs='+', metavar=('W1', 'W2')),
- Sig('-x', help='x', nargs='*', metavar=('X1', 'X2')),
- Sig('-y', help='y', nargs=3, metavar=('Y1', 'Y2', 'Y3')),
- Sig('-z', help='z', nargs='?', metavar=('Z1', )),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] [-w W1 [W2 ...]] [-x [X1 [X2 ...]]] [-y Y1 Y2 Y3] \
- [-z [Z1]]
- '''
- help = usage + '''\
- options:
- -h, --help show this help message and exit
- -w W1 [W2 ...] w
- -x [X1 [X2 ...]] x
- -y Y1 Y2 Y3 y
- -z [Z1] z
- '''
- version = ''
- class TestHelpRawText(HelpTestCase):
- """Test the RawTextHelpFormatter"""
- parser_signature = Sig(
- prog='PROG', formatter_class=argparse.RawTextHelpFormatter,
- description='Keep the formatting\n'
- ' exactly as it is written\n'
- '\n'
- 'here\n')
- argument_signatures = [
- Sig('--foo', help=' foo help should also\n'
- 'appear as given here'),
- Sig('spam', help='spam help'),
- ]
- argument_group_signatures = [
- (Sig('title', description=' This text\n'
- ' should be indented\n'
- ' exactly like it is here\n'),
- [Sig('--bar', help='bar help')]),
- ]
- usage = '''\
- usage: PROG [-h] [--foo FOO] [--bar BAR] spam
- '''
- help = usage + '''\
- Keep the formatting
- exactly as it is written
- here
- positional arguments:
- spam spam help
- options:
- -h, --help show this help message and exit
- --foo FOO foo help should also
- appear as given here
- title:
- This text
- should be indented
- exactly like it is here
- --bar BAR bar help
- '''
- version = ''
- class TestHelpRawDescription(HelpTestCase):
- """Test the RawTextHelpFormatter"""
- parser_signature = Sig(
- prog='PROG', formatter_class=argparse.RawDescriptionHelpFormatter,
- description='Keep the formatting\n'
- ' exactly as it is written\n'
- '\n'
- 'here\n')
- argument_signatures = [
- Sig('--foo', help=' foo help should not\n'
- ' retain this odd formatting'),
- Sig('spam', help='spam help'),
- ]
- argument_group_signatures = [
- (Sig('title', description=' This text\n'
- ' should be indented\n'
- ' exactly like it is here\n'),
- [Sig('--bar', help='bar help')]),
- ]
- usage = '''\
- usage: PROG [-h] [--foo FOO] [--bar BAR] spam
- '''
- help = usage + '''\
- Keep the formatting
- exactly as it is written
- here
- positional arguments:
- spam spam help
- options:
- -h, --help show this help message and exit
- --foo FOO foo help should not retain this odd formatting
- title:
- This text
- should be indented
- exactly like it is here
- --bar BAR bar help
- '''
- version = ''
- class TestHelpArgumentDefaults(HelpTestCase):
- """Test the ArgumentDefaultsHelpFormatter"""
- parser_signature = Sig(
- prog='PROG', formatter_class=argparse.ArgumentDefaultsHelpFormatter,
- description='description')
- argument_signatures = [
- Sig('--foo', help='foo help - oh and by the way, %(default)s'),
- Sig('--bar', action='store_true', help='bar help'),
- Sig('--taz', action=argparse.BooleanOptionalAction,
- help='Whether to taz it', default=True),
- Sig('--corge', action=argparse.BooleanOptionalAction,
- help='Whether to corge it', default=argparse.SUPPRESS),
- Sig('--quux', help="Set the quux", default=42),
- Sig('spam', help='spam help'),
- Sig('badger', nargs='?', default='wooden', help='badger help'),
- ]
- argument_group_signatures = [
- (Sig('title', description='description'),
- [Sig('--baz', type=int, default=42, help='baz help')]),
- ]
- usage = '''\
- usage: PROG [-h] [--foo FOO] [--bar] [--taz | --no-taz] [--corge | --no-corge]
- [--quux QUUX] [--baz BAZ]
- spam [badger]
- '''
- help = usage + '''\
- description
- positional arguments:
- spam spam help
- badger badger help (default: wooden)
- options:
- -h, --help show this help message and exit
- --foo FOO foo help - oh and by the way, None
- --bar bar help (default: False)
- --taz, --no-taz Whether to taz it (default: True)
- --corge, --no-corge Whether to corge it
- --quux QUUX Set the quux (default: 42)
- title:
- description
- --baz BAZ baz help (default: 42)
- '''
- version = ''
- class TestHelpVersionAction(HelpTestCase):
- """Test the default help for the version action"""
- parser_signature = Sig(prog='PROG', description='description')
- argument_signatures = [Sig('-V', '--version', action='version', version='3.6')]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] [-V]
- '''
- help = usage + '''\
- description
- options:
- -h, --help show this help message and exit
- -V, --version show program's version number and exit
- '''
- version = ''
- class TestHelpVersionActionSuppress(HelpTestCase):
- """Test that the --version argument can be suppressed in help messages"""
- parser_signature = Sig(prog='PROG')
- argument_signatures = [
- Sig('-v', '--version', action='version', version='1.0',
- help=argparse.SUPPRESS),
- Sig('--foo', help='foo help'),
- Sig('spam', help='spam help'),
- ]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] [--foo FOO] spam
- '''
- help = usage + '''\
- positional arguments:
- spam spam help
- options:
- -h, --help show this help message and exit
- --foo FOO foo help
- '''
- class TestHelpSubparsersOrdering(HelpTestCase):
- """Test ordering of subcommands in help matches the code"""
- parser_signature = Sig(prog='PROG',
- description='display some subcommands')
- argument_signatures = [Sig('-v', '--version', action='version', version='0.1')]
- subparsers_signatures = [Sig(name=name)
- for name in ('a', 'b', 'c', 'd', 'e')]
- usage = '''\
- usage: PROG [-h] [-v] {a,b,c,d,e} ...
- '''
- help = usage + '''\
- display some subcommands
- positional arguments:
- {a,b,c,d,e}
- options:
- -h, --help show this help message and exit
- -v, --version show program's version number and exit
- '''
- version = '''\
- 0.1
- '''
- class TestHelpSubparsersWithHelpOrdering(HelpTestCase):
- """Test ordering of subcommands in help matches the code"""
- parser_signature = Sig(prog='PROG',
- description='display some subcommands')
- argument_signatures = [Sig('-v', '--version', action='version', version='0.1')]
- subcommand_data = (('a', 'a subcommand help'),
- ('b', 'b subcommand help'),
- ('c', 'c subcommand help'),
- ('d', 'd subcommand help'),
- ('e', 'e subcommand help'),
- )
- subparsers_signatures = [Sig(name=name, help=help)
- for name, help in subcommand_data]
- usage = '''\
- usage: PROG [-h] [-v] {a,b,c,d,e} ...
- '''
- help = usage + '''\
- display some subcommands
- positional arguments:
- {a,b,c,d,e}
- a a subcommand help
- b b subcommand help
- c c subcommand help
- d d subcommand help
- e e subcommand help
- options:
- -h, --help show this help message and exit
- -v, --version show program's version number and exit
- '''
- version = '''\
- 0.1
- '''
- class TestHelpMetavarTypeFormatter(HelpTestCase):
- def custom_type(string):
- return string
- parser_signature = Sig(prog='PROG', description='description',
- formatter_class=argparse.MetavarTypeHelpFormatter)
- argument_signatures = [Sig('a', type=int),
- Sig('-b', type=custom_type),
- Sig('-c', type=float, metavar='SOME FLOAT')]
- argument_group_signatures = []
- usage = '''\
- usage: PROG [-h] [-b custom_type] [-c SOME FLOAT] int
- '''
- help = usage + '''\
- description
- positional arguments:
- int
- options:
- -h, --help show this help message and exit
- -b custom_type
- -c SOME FLOAT
- '''
- version = ''
- # =====================================
- # Optional/Positional constructor tests
- # =====================================
- class TestInvalidArgumentConstructors(TestCase):
- """Test a bunch of invalid Argument constructors"""
- def assertTypeError(self, *args, **kwargs):
- parser = argparse.ArgumentParser()
- self.assertRaises(TypeError, parser.add_argument,
- *args, **kwargs)
- def assertValueError(self, *args, **kwargs):
- parser = argparse.ArgumentParser()
- self.assertRaises(ValueError, parser.add_argument,
- *args, **kwargs)
- def test_invalid_keyword_arguments(self):
- self.assertTypeError('-x', bar=None)
- self.assertTypeError('-y', callback='foo')
- self.assertTypeError('-y', callback_args=())
- self.assertTypeError('-y', callback_kwargs={})
- def test_missing_destination(self):
- self.assertTypeError()
- for action in ['append', 'store']:
- self.assertTypeError(action=action)
- def test_invalid_option_strings(self):
- self.assertValueError('--')
- self.assertValueError('---')
- def test_invalid_type(self):
- self.assertValueError('--foo', type='int')
- self.assertValueError('--foo', type=(int, float))
- def test_invalid_action(self):
- self.assertValueError('-x', action='foo')
- self.assertValueError('foo', action='baz')
- self.assertValueError('--foo', action=('store', 'append'))
- parser = argparse.ArgumentParser()
- with self.assertRaises(ValueError) as cm:
- parser.add_argument("--foo", action="store-true")
- self.assertIn('unknown action', str(cm.exception))
- def test_multiple_dest(self):
- parser = argparse.ArgumentParser()
- parser.add_argument(dest='foo')
- with self.assertRaises(ValueError) as cm:
- parser.add_argument('bar', dest='baz')
- self.assertIn('dest supplied twice for positional argument',
- str(cm.exception))
- def test_no_argument_actions(self):
- for action in ['store_const', 'store_true', 'store_false',
- 'append_const', 'count']:
- for attrs in [dict(type=int), dict(nargs='+'),
- dict(choices='ab')]:
- self.assertTypeError('-x', action=action, **attrs)
- def test_no_argument_no_const_actions(self):
- # options with zero arguments
- for action in ['store_true', 'store_false', 'count']:
- # const is always disallowed
- self.assertTypeError('-x', const='foo', action=action)
- # nargs is always disallowed
- self.assertTypeError('-x', nargs='*', action=action)
- def test_more_than_one_argument_actions(self):
- for action in ['store', 'append']:
- # nargs=0 is disallowed
- self.assertValueError('-x', nargs=0, action=action)
- self.assertValueError('spam', nargs=0, action=action)
- # const is disallowed with non-optional arguments
- for nargs in [1, '*', '+']:
- self.assertValueError('-x', const='foo',
- nargs=nargs, action=action)
- self.assertValueError('spam', const='foo',
- nargs=nargs, action=action)
- def test_required_const_actions(self):
- for action in ['store_const', 'append_const']:
- # nargs is always disallowed
- self.assertTypeError('-x', nargs='+', action=action)
- def test_parsers_action_missing_params(self):
- self.assertTypeError('command', action='parsers')
- self.assertTypeError('command', action='parsers', prog='PROG')
- self.assertTypeError('command', action='parsers',
- parser_class=argparse.ArgumentParser)
- def test_required_positional(self):
- self.assertTypeError('foo', required=True)
- def test_user_defined_action(self):
- class Success(Exception):
- pass
- class Action(object):
- def __init__(self,
- option_strings,
- dest,
- const,
- default,
- required=False):
- if dest == 'spam':
- if const is Success:
- if default is Success:
- raise Success()
- def __call__(self, *args, **kwargs):
- pass
- parser = argparse.ArgumentParser()
- self.assertRaises(Success, parser.add_argument, '--spam',
- action=Action, default=Success, const=Success)
- self.assertRaises(Success, parser.add_argument, 'spam',
- action=Action, default=Success, const=Success)
- # ================================
- # Actions returned by add_argument
- # ================================
- class TestActionsReturned(TestCase):
- def test_dest(self):
- parser = argparse.ArgumentParser()
- action = parser.add_argument('--foo')
- self.assertEqual(action.dest, 'foo')
- action = parser.add_argument('-b', '--bar')
- self.assertEqual(action.dest, 'bar')
- action = parser.add_argument('-x', '-y')
- self.assertEqual(action.dest, 'x')
- def test_misc(self):
- parser = argparse.ArgumentParser()
- action = parser.add_argument('--foo', nargs='?', const=42,
- default=84, type=int, choices=[1, 2],
- help='FOO', metavar='BAR', dest='baz')
- self.assertEqual(action.nargs, '?')
- self.assertEqual(action.const, 42)
- self.assertEqual(action.default, 84)
- self.assertEqual(action.type, int)
- self.assertEqual(action.choices, [1, 2])
- self.assertEqual(action.help, 'FOO')
- self.assertEqual(action.metavar, 'BAR')
- self.assertEqual(action.dest, 'baz')
- # ================================
- # Argument conflict handling tests
- # ================================
- class TestConflictHandling(TestCase):
- def test_bad_type(self):
- self.assertRaises(ValueError, argparse.ArgumentParser,
- conflict_handler='foo')
- def test_conflict_error(self):
- parser = argparse.ArgumentParser()
- parser.add_argument('-x')
- self.assertRaises(argparse.ArgumentError,
- parser.add_argument, '-x')
- parser.add_argument('--spam')
- self.assertRaises(argparse.ArgumentError,
- parser.add_argument, '--spam')
- def test_resolve_error(self):
- get_parser = argparse.ArgumentParser
- parser = get_parser(prog='PROG', conflict_handler='resolve')
- parser.add_argument('-x', help='OLD X')
- parser.add_argument('-x', help='NEW X')
- self.assertEqual(parser.format_help(), textwrap.dedent('''\
- usage: PROG [-h] [-x X]
- options:
- -h, --help show this help message and exit
- -x X NEW X
- '''))
- parser.add_argument('--spam', metavar='OLD_SPAM')
- parser.add_argument('--spam', metavar='NEW_SPAM')
- self.assertEqual(parser.format_help(), textwrap.dedent('''\
- usage: PROG [-h] [-x X] [--spam NEW_SPAM]
- options:
- -h, --help show this help message and exit
- -x X NEW X
- --spam NEW_SPAM
- '''))
- def test_subparser_conflict(self):
- parser = argparse.ArgumentParser()
- sp = parser.add_subparsers()
- sp.add_parser('fullname', aliases=['alias'])
- self.assertRaises(argparse.ArgumentError,
- sp.add_parser, 'fullname')
- self.assertRaises(argparse.ArgumentError,
- sp.add_parser, 'alias')
- self.assertRaises(argparse.ArgumentError,
- sp.add_parser, 'other', aliases=['fullname'])
- self.assertRaises(argparse.ArgumentError,
- sp.add_parser, 'other', aliases=['alias'])
- # =============================
- # Help and Version option tests
- # =============================
- class TestOptionalsHelpVersionActions(TestCase):
- """Test the help and version actions"""
- def assertPrintHelpExit(self, parser, args_str):
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_args(args_str.split())
- self.assertEqual(parser.format_help(), cm.exception.stdout)
- def assertArgumentParserError(self, parser, *args):
- self.assertRaises(ArgumentParserError, parser.parse_args, args)
- def test_version(self):
- parser = ErrorRaisingArgumentParser()
- parser.add_argument('-v', '--version', action='version', version='1.0')
- self.assertPrintHelpExit(parser, '-h')
- self.assertPrintHelpExit(parser, '--help')
- self.assertRaises(AttributeError, getattr, parser, 'format_version')
- def test_version_format(self):
- parser = ErrorRaisingArgumentParser(prog='PPP')
- parser.add_argument('-v', '--version', action='version', version='%(prog)s 3.5')
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_args(['-v'])
- self.assertEqual('PPP 3.5\n', cm.exception.stdout)
- def test_version_no_help(self):
- parser = ErrorRaisingArgumentParser(add_help=False)
- parser.add_argument('-v', '--version', action='version', version='1.0')
- self.assertArgumentParserError(parser, '-h')
- self.assertArgumentParserError(parser, '--help')
- self.assertRaises(AttributeError, getattr, parser, 'format_version')
- def test_version_action(self):
- parser = ErrorRaisingArgumentParser(prog='XXX')
- parser.add_argument('-V', action='version', version='%(prog)s 3.7')
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_args(['-V'])
- self.assertEqual('XXX 3.7\n', cm.exception.stdout)
- def test_no_help(self):
- parser = ErrorRaisingArgumentParser(add_help=False)
- self.assertArgumentParserError(parser, '-h')
- self.assertArgumentParserError(parser, '--help')
- self.assertArgumentParserError(parser, '-v')
- self.assertArgumentParserError(parser, '--version')
- def test_alternate_help_version(self):
- parser = ErrorRaisingArgumentParser()
- parser.add_argument('-x', action='help')
- parser.add_argument('-y', action='version')
- self.assertPrintHelpExit(parser, '-x')
- self.assertArgumentParserError(parser, '-v')
- self.assertArgumentParserError(parser, '--version')
- self.assertRaises(AttributeError, getattr, parser, 'format_version')
- def test_help_version_extra_arguments(self):
- parser = ErrorRaisingArgumentParser()
- parser.add_argument('--version', action='version', version='1.0')
- parser.add_argument('-x', action='store_true')
- parser.add_argument('y')
- # try all combinations of valid prefixes and suffixes
- valid_prefixes = ['', '-x', 'foo', '-x bar', 'baz -x']
- valid_suffixes = valid_prefixes + ['--bad-option', 'foo bar baz']
- for prefix in valid_prefixes:
- for suffix in valid_suffixes:
- format = '%s %%s %s' % (prefix, suffix)
- self.assertPrintHelpExit(parser, format % '-h')
- self.assertPrintHelpExit(parser, format % '--help')
- self.assertRaises(AttributeError, getattr, parser, 'format_version')
- # ======================
- # str() and repr() tests
- # ======================
- class TestStrings(TestCase):
- """Test str() and repr() on Optionals and Positionals"""
- def assertStringEqual(self, obj, result_string):
- for func in [str, repr]:
- self.assertEqual(func(obj), result_string)
- def test_optional(self):
- option = argparse.Action(
- option_strings=['--foo', '-a', '-b'],
- dest='b',
- type='int',
- nargs='+',
- default=42,
- choices=[1, 2, 3],
- required=False,
- help='HELP',
- metavar='METAVAR')
- string = (
- "Action(option_strings=['--foo', '-a', '-b'], dest='b', "
- "nargs='+', const=None, default=42, type='int', "
- "choices=[1, 2, 3], required=False, help='HELP', metavar='METAVAR')")
- self.assertStringEqual(option, string)
- def test_argument(self):
- argument = argparse.Action(
- option_strings=[],
- dest='x',
- type=float,
- nargs='?',
- default=2.5,
- choices=[0.5, 1.5, 2.5],
- required=True,
- help='H HH H',
- metavar='MV MV MV')
- string = (
- "Action(option_strings=[], dest='x', nargs='?', "
- "const=None, default=2.5, type=%r, choices=[0.5, 1.5, 2.5], "
- "required=True, help='H HH H', metavar='MV MV MV')" % float)
- self.assertStringEqual(argument, string)
- def test_namespace(self):
- ns = argparse.Namespace(foo=42, bar='spam')
- string = "Namespace(foo=42, bar='spam')"
- self.assertStringEqual(ns, string)
- def test_namespace_starkwargs_notidentifier(self):
- ns = argparse.Namespace(**{'"': 'quote'})
- string = """Namespace(**{'"': 'quote'})"""
- self.assertStringEqual(ns, string)
- def test_namespace_kwargs_and_starkwargs_notidentifier(self):
- ns = argparse.Namespace(a=1, **{'"': 'quote'})
- string = """Namespace(a=1, **{'"': 'quote'})"""
- self.assertStringEqual(ns, string)
- def test_namespace_starkwargs_identifier(self):
- ns = argparse.Namespace(**{'valid': True})
- string = "Namespace(valid=True)"
- self.assertStringEqual(ns, string)
- def test_parser(self):
- parser = argparse.ArgumentParser(prog='PROG')
- string = (
- "ArgumentParser(prog='PROG', usage=None, description=None, "
- "formatter_class=%r, conflict_handler='error', "
- "add_help=True)" % argparse.HelpFormatter)
- self.assertStringEqual(parser, string)
- # ===============
- # Namespace tests
- # ===============
- class TestNamespace(TestCase):
- def test_constructor(self):
- ns = argparse.Namespace()
- self.assertRaises(AttributeError, getattr, ns, 'x')
- ns = argparse.Namespace(a=42, b='spam')
- self.assertEqual(ns.a, 42)
- self.assertEqual(ns.b, 'spam')
- def test_equality(self):
- ns1 = argparse.Namespace(a=1, b=2)
- ns2 = argparse.Namespace(b=2, a=1)
- ns3 = argparse.Namespace(a=1)
- ns4 = argparse.Namespace(b=2)
- self.assertEqual(ns1, ns2)
- self.assertNotEqual(ns1, ns3)
- self.assertNotEqual(ns1, ns4)
- self.assertNotEqual(ns2, ns3)
- self.assertNotEqual(ns2, ns4)
- self.assertTrue(ns1 != ns3)
- self.assertTrue(ns1 != ns4)
- self.assertTrue(ns2 != ns3)
- self.assertTrue(ns2 != ns4)
- def test_equality_returns_notimplemented(self):
- # See issue 21481
- ns = argparse.Namespace(a=1, b=2)
- self.assertIs(ns.__eq__(None), NotImplemented)
- self.assertIs(ns.__ne__(None), NotImplemented)
- # ===================
- # File encoding tests
- # ===================
- class TestEncoding(TestCase):
- def _test_module_encoding(self, path):
- path, _ = os.path.splitext(path)
- path += ".py"
- with open(path, 'r', encoding='utf-8') as f:
- f.read()
- def test_argparse_module_encoding(self):
- self._test_module_encoding(argparse.__file__)
- def test_test_argparse_module_encoding(self):
- self._test_module_encoding(__file__)
- # ===================
- # ArgumentError tests
- # ===================
- class TestArgumentError(TestCase):
- def test_argument_error(self):
- msg = "my error here"
- error = argparse.ArgumentError(None, msg)
- self.assertEqual(str(error), msg)
- # =======================
- # ArgumentTypeError tests
- # =======================
- class TestArgumentTypeError(TestCase):
- def test_argument_type_error(self):
- def spam(string):
- raise argparse.ArgumentTypeError('spam!')
- parser = ErrorRaisingArgumentParser(prog='PROG', add_help=False)
- parser.add_argument('x', type=spam)
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_args(['XXX'])
- self.assertEqual('usage: PROG x\nPROG: error: argument x: spam!\n',
- cm.exception.stderr)
- # =========================
- # MessageContentError tests
- # =========================
- class TestMessageContentError(TestCase):
- def test_missing_argument_name_in_message(self):
- parser = ErrorRaisingArgumentParser(prog='PROG', usage='')
- parser.add_argument('req_pos', type=str)
- parser.add_argument('-req_opt', type=int, required=True)
- parser.add_argument('need_one', type=str, nargs='+')
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_args([])
- msg = str(cm.exception)
- self.assertRegex(msg, 'req_pos')
- self.assertRegex(msg, 'req_opt')
- self.assertRegex(msg, 'need_one')
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_args(['myXargument'])
- msg = str(cm.exception)
- self.assertNotIn(msg, 'req_pos')
- self.assertRegex(msg, 'req_opt')
- self.assertRegex(msg, 'need_one')
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_args(['myXargument', '-req_opt=1'])
- msg = str(cm.exception)
- self.assertNotIn(msg, 'req_pos')
- self.assertNotIn(msg, 'req_opt')
- self.assertRegex(msg, 'need_one')
- def test_optional_optional_not_in_message(self):
- parser = ErrorRaisingArgumentParser(prog='PROG', usage='')
- parser.add_argument('req_pos', type=str)
- parser.add_argument('--req_opt', type=int, required=True)
- parser.add_argument('--opt_opt', type=bool, nargs='?',
- default=True)
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_args([])
- msg = str(cm.exception)
- self.assertRegex(msg, 'req_pos')
- self.assertRegex(msg, 'req_opt')
- self.assertNotIn(msg, 'opt_opt')
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_args(['--req_opt=1'])
- msg = str(cm.exception)
- self.assertRegex(msg, 'req_pos')
- self.assertNotIn(msg, 'req_opt')
- self.assertNotIn(msg, 'opt_opt')
- def test_optional_positional_not_in_message(self):
- parser = ErrorRaisingArgumentParser(prog='PROG', usage='')
- parser.add_argument('req_pos')
- parser.add_argument('optional_positional', nargs='?', default='eggs')
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_args([])
- msg = str(cm.exception)
- self.assertRegex(msg, 'req_pos')
- self.assertNotIn(msg, 'optional_positional')
- # ================================================
- # Check that the type function is called only once
- # ================================================
- class TestTypeFunctionCallOnlyOnce(TestCase):
- def test_type_function_call_only_once(self):
- def spam(string_to_convert):
- self.assertEqual(string_to_convert, 'spam!')
- return 'foo_converted'
- parser = argparse.ArgumentParser()
- parser.add_argument('--foo', type=spam, default='bar')
- args = parser.parse_args('--foo spam!'.split())
- self.assertEqual(NS(foo='foo_converted'), args)
- # ==================================================================
- # Check semantics regarding the default argument and type conversion
- # ==================================================================
- class TestTypeFunctionCalledOnDefault(TestCase):
- def test_type_function_call_with_non_string_default(self):
- def spam(int_to_convert):
- self.assertEqual(int_to_convert, 0)
- return 'foo_converted'
- parser = argparse.ArgumentParser()
- parser.add_argument('--foo', type=spam, default=0)
- args = parser.parse_args([])
- # foo should *not* be converted because its default is not a string.
- self.assertEqual(NS(foo=0), args)
- def test_type_function_call_with_string_default(self):
- def spam(int_to_convert):
- return 'foo_converted'
- parser = argparse.ArgumentParser()
- parser.add_argument('--foo', type=spam, default='0')
- args = parser.parse_args([])
- # foo is converted because its default is a string.
- self.assertEqual(NS(foo='foo_converted'), args)
- def test_no_double_type_conversion_of_default(self):
- def extend(str_to_convert):
- return str_to_convert + '*'
- parser = argparse.ArgumentParser()
- parser.add_argument('--test', type=extend, default='*')
- args = parser.parse_args([])
- # The test argument will be two stars, one coming from the default
- # value and one coming from the type conversion being called exactly
- # once.
- self.assertEqual(NS(test='**'), args)
- def test_issue_15906(self):
- # Issue #15906: When action='append', type=str, default=[] are
- # providing, the dest value was the string representation "[]" when it
- # should have been an empty list.
- parser = argparse.ArgumentParser()
- parser.add_argument('--test', dest='test', type=str,
- default=[], action='append')
- args = parser.parse_args([])
- self.assertEqual(args.test, [])
- # ======================
- # parse_known_args tests
- # ======================
- class TestParseKnownArgs(TestCase):
- def test_arguments_tuple(self):
- parser = argparse.ArgumentParser()
- parser.parse_args(())
- def test_arguments_list(self):
- parser = argparse.ArgumentParser()
- parser.parse_args([])
- def test_arguments_tuple_positional(self):
- parser = argparse.ArgumentParser()
- parser.add_argument('x')
- parser.parse_args(('x',))
- def test_arguments_list_positional(self):
- parser = argparse.ArgumentParser()
- parser.add_argument('x')
- parser.parse_args(['x'])
- def test_optionals(self):
- parser = argparse.ArgumentParser()
- parser.add_argument('--foo')
- args, extras = parser.parse_known_args('--foo F --bar --baz'.split())
- self.assertEqual(NS(foo='F'), args)
- self.assertEqual(['--bar', '--baz'], extras)
- def test_mixed(self):
- parser = argparse.ArgumentParser()
- parser.add_argument('-v', nargs='?', const=1, type=int)
- parser.add_argument('--spam', action='store_false')
- parser.add_argument('badger')
- argv = ["B", "C", "--foo", "-v", "3", "4"]
- args, extras = parser.parse_known_args(argv)
- self.assertEqual(NS(v=3, spam=True, badger="B"), args)
- self.assertEqual(["C", "--foo", "4"], extras)
- # ===========================
- # parse_intermixed_args tests
- # ===========================
- class TestIntermixedArgs(TestCase):
- def test_basic(self):
- # test parsing intermixed optionals and positionals
- parser = argparse.ArgumentParser(prog='PROG')
- parser.add_argument('--foo', dest='foo')
- bar = parser.add_argument('--bar', dest='bar', required=True)
- parser.add_argument('cmd')
- parser.add_argument('rest', nargs='*', type=int)
- argv = 'cmd --foo x 1 --bar y 2 3'.split()
- args = parser.parse_intermixed_args(argv)
- # rest gets [1,2,3] despite the foo and bar strings
- self.assertEqual(NS(bar='y', cmd='cmd', foo='x', rest=[1, 2, 3]), args)
- args, extras = parser.parse_known_args(argv)
- # cannot parse the '1,2,3'
- self.assertEqual(NS(bar='y', cmd='cmd', foo='x', rest=[]), args)
- self.assertEqual(["1", "2", "3"], extras)
- argv = 'cmd --foo x 1 --error 2 --bar y 3'.split()
- args, extras = parser.parse_known_intermixed_args(argv)
- # unknown optionals go into extras
- self.assertEqual(NS(bar='y', cmd='cmd', foo='x', rest=[1]), args)
- self.assertEqual(['--error', '2', '3'], extras)
- # restores attributes that were temporarily changed
- self.assertIsNone(parser.usage)
- self.assertEqual(bar.required, True)
- def test_remainder(self):
- # Intermixed and remainder are incompatible
- parser = ErrorRaisingArgumentParser(prog='PROG')
- parser.add_argument('-z')
- parser.add_argument('x')
- parser.add_argument('y', nargs='...')
- argv = 'X A B -z Z'.split()
- # intermixed fails with '...' (also 'A...')
- # self.assertRaises(TypeError, parser.parse_intermixed_args, argv)
- with self.assertRaises(TypeError) as cm:
- parser.parse_intermixed_args(argv)
- self.assertRegex(str(cm.exception), r'\.\.\.')
- def test_exclusive(self):
- # mutually exclusive group; intermixed works fine
- parser = ErrorRaisingArgumentParser(prog='PROG')
- group = parser.add_mutually_exclusive_group(required=True)
- group.add_argument('--foo', action='store_true', help='FOO')
- group.add_argument('--spam', help='SPAM')
- parser.add_argument('badger', nargs='*', default='X', help='BADGER')
- args = parser.parse_intermixed_args('1 --foo 2'.split())
- self.assertEqual(NS(badger=['1', '2'], foo=True, spam=None), args)
- self.assertRaises(ArgumentParserError, parser.parse_intermixed_args, '1 2'.split())
- self.assertEqual(group.required, True)
- def test_exclusive_incompatible(self):
- # mutually exclusive group including positional - fail
- parser = ErrorRaisingArgumentParser(prog='PROG')
- group = parser.add_mutually_exclusive_group(required=True)
- group.add_argument('--foo', action='store_true', help='FOO')
- group.add_argument('--spam', help='SPAM')
- group.add_argument('badger', nargs='*', default='X', help='BADGER')
- self.assertRaises(TypeError, parser.parse_intermixed_args, [])
- self.assertEqual(group.required, True)
- class TestIntermixedMessageContentError(TestCase):
- # case where Intermixed gives different error message
- # error is raised by 1st parsing step
- def test_missing_argument_name_in_message(self):
- parser = ErrorRaisingArgumentParser(prog='PROG', usage='')
- parser.add_argument('req_pos', type=str)
- parser.add_argument('-req_opt', type=int, required=True)
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_args([])
- msg = str(cm.exception)
- self.assertRegex(msg, 'req_pos')
- self.assertRegex(msg, 'req_opt')
- with self.assertRaises(ArgumentParserError) as cm:
- parser.parse_intermixed_args([])
- msg = str(cm.exception)
- self.assertNotRegex(msg, 'req_pos')
- self.assertRegex(msg, 'req_opt')
- # ==========================
- # add_argument metavar tests
- # ==========================
- class TestAddArgumentMetavar(TestCase):
- EXPECTED_MESSAGE = "length of metavar tuple does not match nargs"
- def do_test_no_exception(self, nargs, metavar):
- parser = argparse.ArgumentParser()
- parser.add_argument("--foo", nargs=nargs, metavar=metavar)
- def do_test_exception(self, nargs, metavar):
- parser = argparse.ArgumentParser()
- with self.assertRaises(ValueError) as cm:
- parser.add_argument("--foo", nargs=nargs, metavar=metavar)
- self.assertEqual(cm.exception.args[0], self.EXPECTED_MESSAGE)
- # Unit tests for different values of metavar when nargs=None
- def test_nargs_None_metavar_string(self):
- self.do_test_no_exception(nargs=None, metavar="1")
- def test_nargs_None_metavar_length0(self):
- self.do_test_exception(nargs=None, metavar=tuple())
- def test_nargs_None_metavar_length1(self):
- self.do_test_no_exception(nargs=None, metavar=("1",))
- def test_nargs_None_metavar_length2(self):
- self.do_test_exception(nargs=None, metavar=("1", "2"))
- def test_nargs_None_metavar_length3(self):
- self.do_test_exception(nargs=None, metavar=("1", "2", "3"))
- # Unit tests for different values of metavar when nargs=?
- def test_nargs_optional_metavar_string(self):
- self.do_test_no_exception(nargs="?", metavar="1")
- def test_nargs_optional_metavar_length0(self):
- self.do_test_exception(nargs="?", metavar=tuple())
- def test_nargs_optional_metavar_length1(self):
- self.do_test_no_exception(nargs="?", metavar=("1",))
- def test_nargs_optional_metavar_length2(self):
- self.do_test_exception(nargs="?", metavar=("1", "2"))
- def test_nargs_optional_metavar_length3(self):
- self.do_test_exception(nargs="?", metavar=("1", "2", "3"))
- # Unit tests for different values of metavar when nargs=*
- def test_nargs_zeroormore_metavar_string(self):
- self.do_test_no_exception(nargs="*", metavar="1")
- def test_nargs_zeroormore_metavar_length0(self):
- self.do_test_exception(nargs="*", metavar=tuple())
- def test_nargs_zeroormore_metavar_length1(self):
- self.do_test_no_exception(nargs="*", metavar=("1",))
- def test_nargs_zeroormore_metavar_length2(self):
- self.do_test_no_exception(nargs="*", metavar=("1", "2"))
- def test_nargs_zeroormore_metavar_length3(self):
- self.do_test_exception(nargs="*", metavar=("1", "2", "3"))
- # Unit tests for different values of metavar when nargs=+
- def test_nargs_oneormore_metavar_string(self):
- self.do_test_no_exception(nargs="+", metavar="1")
- def test_nargs_oneormore_metavar_length0(self):
- self.do_test_exception(nargs="+", metavar=tuple())
- def test_nargs_oneormore_metavar_length1(self):
- self.do_test_exception(nargs="+", metavar=("1",))
- def test_nargs_oneormore_metavar_length2(self):
- self.do_test_no_exception(nargs="+", metavar=("1", "2"))
- def test_nargs_oneormore_metavar_length3(self):
- self.do_test_exception(nargs="+", metavar=("1", "2", "3"))
- # Unit tests for different values of metavar when nargs=...
- def test_nargs_remainder_metavar_string(self):
- self.do_test_no_exception(nargs="...", metavar="1")
- def test_nargs_remainder_metavar_length0(self):
- self.do_test_no_exception(nargs="...", metavar=tuple())
- def test_nargs_remainder_metavar_length1(self):
- self.do_test_no_exception(nargs="...", metavar=("1",))
- def test_nargs_remainder_metavar_length2(self):
- self.do_test_no_exception(nargs="...", metavar=("1", "2"))
- def test_nargs_remainder_metavar_length3(self):
- self.do_test_no_exception(nargs="...", metavar=("1", "2", "3"))
- # Unit tests for different values of metavar when nargs=A...
- def test_nargs_parser_metavar_string(self):
- self.do_test_no_exception(nargs="A...", metavar="1")
- def test_nargs_parser_metavar_length0(self):
- self.do_test_exception(nargs="A...", metavar=tuple())
- def test_nargs_parser_metavar_length1(self):
- self.do_test_no_exception(nargs="A...", metavar=("1",))
- def test_nargs_parser_metavar_length2(self):
- self.do_test_exception(nargs="A...", metavar=("1", "2"))
- def test_nargs_parser_metavar_length3(self):
- self.do_test_exception(nargs="A...", metavar=("1", "2", "3"))
- # Unit tests for different values of metavar when nargs=1
- def test_nargs_1_metavar_string(self):
- self.do_test_no_exception(nargs=1, metavar="1")
- def test_nargs_1_metavar_length0(self):
- self.do_test_exception(nargs=1, metavar=tuple())
- def test_nargs_1_metavar_length1(self):
- self.do_test_no_exception(nargs=1, metavar=("1",))
- def test_nargs_1_metavar_length2(self):
- self.do_test_exception(nargs=1, metavar=("1", "2"))
- def test_nargs_1_metavar_length3(self):
- self.do_test_exception(nargs=1, metavar=("1", "2", "3"))
- # Unit tests for different values of metavar when nargs=2
- def test_nargs_2_metavar_string(self):
- self.do_test_no_exception(nargs=2, metavar="1")
- def test_nargs_2_metavar_length0(self):
- self.do_test_exception(nargs=2, metavar=tuple())
- def test_nargs_2_metavar_length1(self):
- self.do_test_exception(nargs=2, metavar=("1",))
- def test_nargs_2_metavar_length2(self):
- self.do_test_no_exception(nargs=2, metavar=("1", "2"))
- def test_nargs_2_metavar_length3(self):
- self.do_test_exception(nargs=2, metavar=("1", "2", "3"))
- # Unit tests for different values of metavar when nargs=3
- def test_nargs_3_metavar_string(self):
- self.do_test_no_exception(nargs=3, metavar="1")
- def test_nargs_3_metavar_length0(self):
- self.do_test_exception(nargs=3, metavar=tuple())
- def test_nargs_3_metavar_length1(self):
- self.do_test_exception(nargs=3, metavar=("1",))
- def test_nargs_3_metavar_length2(self):
- self.do_test_exception(nargs=3, metavar=("1", "2"))
- def test_nargs_3_metavar_length3(self):
- self.do_test_no_exception(nargs=3, metavar=("1", "2", "3"))
- class TestInvalidNargs(TestCase):
- EXPECTED_INVALID_MESSAGE = "invalid nargs value"
- EXPECTED_RANGE_MESSAGE = ("nargs for store actions must be != 0; if you "
- "have nothing to store, actions such as store "
- "true or store const may be more appropriate")
- def do_test_range_exception(self, nargs):
- parser = argparse.ArgumentParser()
- with self.assertRaises(ValueError) as cm:
- parser.add_argument("--foo", nargs=nargs)
- self.assertEqual(cm.exception.args[0], self.EXPECTED_RANGE_MESSAGE)
- def do_test_invalid_exception(self, nargs):
- parser = argparse.ArgumentParser()
- with self.assertRaises(ValueError) as cm:
- parser.add_argument("--foo", nargs=nargs)
- self.assertEqual(cm.exception.args[0], self.EXPECTED_INVALID_MESSAGE)
- # Unit tests for different values of nargs
- def test_nargs_alphabetic(self):
- self.do_test_invalid_exception(nargs='a')
- self.do_test_invalid_exception(nargs="abcd")
- def test_nargs_zero(self):
- self.do_test_range_exception(nargs=0)
- # ============================
- # from argparse import * tests
- # ============================
- class TestImportStar(TestCase):
- def test(self):
- for name in argparse.__all__:
- self.assertTrue(hasattr(argparse, name))
- def test_all_exports_everything_but_modules(self):
- items = [
- name
- for name, value in vars(argparse).items()
- if not (name.startswith("_") or name == 'ngettext')
- if not inspect.ismodule(value)
- ]
- self.assertEqual(sorted(items), sorted(argparse.__all__))
- class TestWrappingMetavar(TestCase):
- def setUp(self):
- super().setUp()
- self.parser = ErrorRaisingArgumentParser(
- 'this_is_spammy_prog_with_a_long_name_sorry_about_the_name'
- )
- # this metavar was triggering library assertion errors due to usage
- # message formatting incorrectly splitting on the ] chars within
- metavar = '<http[s]://example:1234>'
- self.parser.add_argument('--proxy', metavar=metavar)
- def test_help_with_metavar(self):
- help_text = self.parser.format_help()
- self.assertEqual(help_text, textwrap.dedent('''\
- usage: this_is_spammy_prog_with_a_long_name_sorry_about_the_name
- [-h] [--proxy <http[s]://example:1234>]
- options:
- -h, --help show this help message and exit
- --proxy <http[s]://example:1234>
- '''))
- class TestExitOnError(TestCase):
- def setUp(self):
- self.parser = argparse.ArgumentParser(exit_on_error=False)
- self.parser.add_argument('--integers', metavar='N', type=int)
- def test_exit_on_error_with_good_args(self):
- ns = self.parser.parse_args('--integers 4'.split())
- self.assertEqual(ns, argparse.Namespace(integers=4))
- def test_exit_on_error_with_bad_args(self):
- with self.assertRaises(argparse.ArgumentError):
- self.parser.parse_args('--integers a'.split())
- def tearDownModule():
- # Remove global references to avoid looking like we have refleaks.
- RFile.seen = {}
- WFile.seen = set()
- if __name__ == '__main__':
- unittest.main()
|