U +c[/@sXdZddlmZddlmZddlmZddlZddlZddlZddlZej rZddl m Z d{d d Z eed e _d|d dZd}ddZGdddeZGdddZGdddeZGdddeZGdddeZGdddeZGdddeZGd d!d!ZGd"d#d#eZGd$d%d%eZGd&d'd'eZGd(d)d)eZGd*d+d+eZGd,d-d-eZ Gd.d/d/eZ!Gd0d1d1eZ"Gd2d3d3eZ#Gd4d5d5eZ$Gd6d7d7eZ%Gd8d9d9eZ&Gd:d;d;eZ'Gdd?d?eZ)Gd@dAdAeZ*GdBdCdCeZ+GdDdEdEeZ,GdFdGdGeZ-e.dHdIdJgZ/GdKdLdLe/Z0GdMdNdNeZ1e1Z2GdOdPdPeZ3GdQdRdReZ4dSdTZ5dUdVZ6GdWdXdXe7Z8e8d dYdZZ9Gd[d\d\ej:Z;Gd]d^d^eZdedfZ?GdgdhdheZ@GdidjdjeZAGdkdldleZBdmdnZCeCddpdqZDdrdsZEddtduZFddydzZGdS)zAThis module has components that are used for testing Python Fire.)absolute_import)division)print_functionN)test_components_py3 cOs||||||fSN)arg1arg2arg3arg4Zarg5Zarg6r r 8/tmp/pip-unpacked-wheel-1xt1w7un/fire/test_components.pyidentitysr)r r cCs||S)zMultiplies num by rate. Args: num (int): the num you want to multiply rate (int): the rate for multiplication Returns: Multiplication of num by rate r )numZrater r rmultiplier_with_docstring%s rTcCs|Srr )helpr r rfunction_with_help1src@s eZdZdS)EmptyN__name__ __module__ __qualname__r r r rr5src@s eZdZdS) OldStyleEmptyNrr r r rr9src@seZdZddZdS)WithInitcCsdSrr selfr r r__init__?szWithInit.__init__Nrrrrr r r rr=src@seZdZdddZdS)ErrorInConstructorvaluecCs||_tddS)NzError in constructor)r! ValueErrorrr!r r rrEszErrorInConstructor.__init__N)r!rr r r rr Csr c@seZdZdZdddZdS) WithHelpArgz2Test class for testing when class has a help= arg.TcCs||_ddi|_dS)NZ__helpzhelp in a dict)Zhas_help dictionary)rrr r rrMszWithHelpArg.__init__N)Trrr__doc__rr r r rr$Jsr$c@seZdZddZddZdS) NoDefaultscCsd|SNrr rcountr r rdoubleTszNoDefaults.doublecCsd|SNr r*r r rtripleWszNoDefaults.tripleNrrrr,r/r r r rr(Rsr(c@s.eZdZdZd ddZd ddZd dd Zd S) WithDefaultsz1Class with functions that have default arguments.rcCsd|S)zReturns the input multiplied by 2. Args: count: Input number that you want to double. Returns: A number that is the double of count. rr r*r r rr,^s zWithDefaults.doublecCsd|Sr-r r*r r rr/iszWithDefaults.tripleP00010203040506070809101112131415161718192021222324252627282930313233343536373839cCs|Srr )rstringr r rtextlszWithDefaults.textN)r)r)r2)rrrr'r,r/r4r r r rr1[s  r1c@s eZdZdddZdddZdS) OldStyleWithDefaultsrcCsd|Sr)r r*r r rr,vszOldStyleWithDefaults.doublecCsd|Sr-r r*r r rr/yszOldStyleWithDefaults.tripleN)r)rr0r r r rr5ts r5c@s(eZdZddZd ddZd ddZd S) MixedDefaultscCsdS)Nrr rr r rtenszMixedDefaults.tenrcCs |d|Sr)r ralphabetar r rsumszMixedDefaults.sum0cCs||fSrr r8r r rrszMixedDefaults.identityN)rr)r<)rrrr7r;rr r r rr6}s r6c@s eZdZdddZdddZdS) SimilarArgNamesFcCs||fSrr )rZbool_oneZbool_twor r rrszSimilarArgNames.identityNcCs||fSrr )rar9r r r identity2szSimilarArgNames.identity2)FF)NN)rrrrr?r r r rr=s r=c@seZdZdddZdS)CapitalizedArgNames?@cCs||Srr )rDeltaGammar r rr;szCapitalizedArgNames.sumN)rArB)rrrr;r r r rr@sr@c@s4eZdZdddZd ddZdeie_deie_dS) AnnotationsrcCsd|Sr)r r*r r rr,szAnnotations.doublecCsd|Sr-r r*r r rr/szAnnotations.tripler+N)r)r)rrrr,r/float__annotations__r r r rrEs   rEc@seZdZdZddZdS)TypedPropertieszDTest class for testing Python Fire with properties of various types.cCs@d|_d|_t|_ddddd|_dd g|_d |_d |_dS) NT)rr.Er>b)rrI)echoZnestZalexZbethany)ZcarrydivideZmyexcitingstring)r9r:r1ZcharliedeltarLZfoxgammarr r rrs zTypedProperties.__init__Nr&r r r rrHsrHc@s"eZdZdZddZdddZdS) VarArgsz@Test class for testing Python Fire with a property with varargs.cGs6d}g}|D]$}|dkr|}n||7}||q |Sr)append)ritemstotalZsumsitemr r rcumsumss zVarArgs.cumsumsrcGs||d|fS)Njoin)rr9r:charsr r rvarcharsszVarArgs.varcharsN)rr)rrrr'rUrZr r r rrPs rPc@seZdZddZddZdS) UnderscorescCs d|_dS)Nz fish fingers)Zunderscore_examplerr r rrszUnderscores.__init__cCs|Srr )rZunderscore_argr r runderscore_functionszUnderscores.underscore_functionN)rrrrr\r r r rr[sr[c@seZdZdddZdS) BoolConverterFcCst|Sr)bool)rargr r ras_boolszBoolConverter.as_boolN)F)rrrr`r r r rr]sr]c@seZdZddZdS) ReturnsObjcGs~tSr)r])rrRr r rget_objszReturnsObj.get_objN)rrrrbr r r rrasrac@s eZdZdddZd ddZdS) NumberDefaults$@cCsd|SNrAr rZdivisorr r r reciprocalszNumberDefaults.reciprocalrcCsd|Srer rfr r rinteger_reciprocalsz!NumberDefaults.integer_reciprocalN)rd)r)rrrrgrhr r r rrcs rcc@seZdZddZddZdS) InstanceVarscCs||_||_dSrr r rr r r r rrszInstanceVars.__init__cCs|j|j||fSrrjrkr r rrunszInstanceVars.runN)rrrrrlr r r rrisric@s&eZdZddZddZdddZdS) KwargscKs|Srr rkwargsr r rpropssz Kwargs.propscKsdt|S)N )rXsortedkeysupperrnr r rrtsz Kwargs.upperNcKs |||fSrr )r positionalnamedror r rrlsz Kwargs.run)N)rrrrprtrlr r r rrmsrmc@seZdZddZdS) ErrorRaisercCs tddS)NzThis error is part of a test.r"rr r rfailszErrorRaiser.failN)rrrryr r r rrwsrwc@seZdZddZddZdS) NonComparablecCs tddSNz+Instances of this class cannot be compared.rxrotherr r r__eq__szNonComparable.__eq__cCs tddSr{rxr|r r r__ne__ szNonComparable.__ne__N)rrrr~rr r r rrzsrzc@seZdZddZddZdS)EmptyDictOutputcCsiSrr rr r r totally_emptyszEmptyDictOutput.totally_emptycCsddiS)NZ__do_not_print_merIr rr r rnothing_printablesz!EmptyDictOutput.nothing_printableN)rrrrrr r r rr src@seZdZddZdS)CircularReferencecCsi}||d<|S)Nyr rxr r rcreateszCircularReference.createN)rrrrr r r rrsrc@seZdZddZddZdS)OrderedDictionarycCstSr collections OrderedDictrr r rempty szOrderedDictionary.emptycCst}d|d<d|d<|S)NArr)rZ ordered_dictr r r non_empty#szOrderedDictionary.non_emptyN)rrrrrr r r rrsrc@s eZdZdZddZddZdS) NamedTuplez2Functions returning named tuples used for testing.cCstdddg}|dddS)z(Point example straight from Python docs.Pointrr )rr namedtuplerrr r rpoint-szNamedTuple.pointcCstdddg}|dddS)zField name equals value.rrrrrrrr r rmatching_names3szNamedTuple.matching_namesN)rrrr'rrr r r rr*src@s$eZdZdZdZddZddZdS)CallableWithPositionalArgs#Test class for supporting callable.rIcCs||Srr )rrrr r r__call__?sz#CallableWithPositionalArgs.__call__cCs|dSNrIr rr r rfnBszCallableWithPositionalArgs.fnN)rrrr'ZTESTrrr r r rr:srNamedTuplePointrrc@seZdZdZddZdS)SubPointz@Used for verifying subclasses of namedtuples behave as intended.cCs |j|jSrrrr r rcoordinate_sumLszSubPoint.coordinate_sumN)rrrr'rr r r rrIsrc@s eZdZdZddZddZdS)CallableWithKeywordArgumentrcKs&|D]\}}td||fqdS)Nz%s: %s)rRprint)rrokeyr!r r rrSsz$CallableWithKeywordArgument.__call__cCs t|dSrrrmsgr r r print_msgWsz%CallableWithKeywordArgument.print_msgN)rrrr'rrr r r rrPsrc@s$eZdZdZdddZd ddZdS) ClassWithDocstringzbTest class for testing help text output. This is some detail description of this test class. Hello!cCs ||_dS)zConstructor of the test class. Constructs a new ClassWithDocstring object. Args: message: The default message to print. N)message)rrr r rrdszClassWithDocstring.__init__NcCs|dkr|j}t|dS)zPrints a message.N)rrrr r rrnszClassWithDocstring.print_msg)r)N)rrrr'rrr r r rr^s rc@seZdZdZeddZdS)ClassWithMultilineDocstringzTest class for testing help text output with multiline docstring. This is a test class that has a long docstring description that spans across multiple lines for testing line breaking in help text. ccst|D] }|VqdS)aGenerators have a ``Yields`` section instead of a ``Returns`` section. Args: n (int): The upper limit of the range to generate, from 0 to `n` - 1. Yields: int: The next number in the range of 0 to `n` - 1. Examples: Examples should be written in doctest format, and should illustrate how to use the function. >>> print([i for i in example_generator(4)]) [0, 1, 2, 3] N)range)nir r rexample_generator|s z-ClassWithMultilineDocstring.example_generatorN)rrrr' staticmethodrr r r rrusrcCs dddhSNrIrZthreer r r r r simple_setsrcCstdddhSr) frozensetr r r rsimple_frozensetsrc@seZdZdZdS)Subdictz)A subclass of dict, for testing purposes.N)rrrr'r r r rrsrZblue)rIZredc@seZdZdZdZdZdS)ColorrIrr.N)rrrREDGREENBLUEr r r rrsrc@s4eZdZdZdZddZeddZeddZ d S) HasStaticAndClassMethodsz0A class with a static method and a class method.rIcCs ||_dSr)instance_state)rrr r rrsz!HasStaticAndClassMethods.__init__cCs|Srr )argsr r r static_fnsz"HasStaticAndClassMethods.static_fncCs ||jSr) CLASS_STATE)clsrr r rclass_fnsz!HasStaticAndClassMethods.class_fnN) rrrr'rrrr classmethodrr r r rrs rrIcGs ~~~|S)zFunction with varargs. Args: arg1: Position arg docstring. arg2: Position arg docstring. arg3: Flags docstring. *varargs: Accepts unlimited positional args. Returns: The unlimited positional args. r )r r r varargsr r rfunction_with_varargss rr.cKs ~||fSrr r r ror r rfunction_with_keyword_argumentssrcCsdS)zThis has code in the docstring. Example: x = fn_with_code_in_docstring() indentation_matters = True Returns: True. Tr r r r rfn_with_code_in_docstringsrc@sJeZdZdZdddZddZddZd d Zd d Zd dZ ddZ dS) BinaryCanvasz:A canvas with which to make binary art, one bit at a time.rcs.fddtD|_|_d|_d|_dS)Ncsg|]}dgqS)rr ).0_sizer r sz)BinaryCanvas.__init__..r)rpixels_size_row_col)rrr rrrszBinaryCanvas.__init__cCsddd|jDS)N css"|]}ddd|DVqdS)rqcss|]}t|VqdSr)str)rZpixelr r r sz1BinaryCanvas.__str__...NrW)rrowr r rrsz'BinaryCanvas.__str__..)rXrrr r r__str__s zBinaryCanvas.__str__cCs t||Srrrr r rshowszBinaryCanvas.showcCs||j|_||j|_|Sr)rrr)rrcolr r rmoves  zBinaryCanvas.movecCs |dSrsetrr r ronszBinaryCanvas.oncCs |dS)Nrrrr r roffszBinaryCanvas.offcCs||j|j|j<|Sr)rrrr#r r rrszBinaryCanvas.setN)r) rrrr'rrrrrrrr r r rrs rc@seZdZddZddZdS) DefaultMethodcCsd|Sr)r rnumberr r rr,szDefaultMethod.doublecCs dd}|S)NcSsdS)NzUndefined functionr r r r r_missingsz+DefaultMethod.__getattr__.._missingr )rnamerr r r __getattr__szDefaultMethod.__getattr__N)rrrr,rr r r rrsrc@s eZdZddZeddZdS)InvalidPropertycCsd|Sr)r rr r rr, szInvalidProperty.doublecCs tddS)Ntestrxrr r rpropszInvalidProperty.propN)rrrr,propertyrr r r rr srcstfdd}|S)Ncs ||Srr )rrofr rwrappersz!simple_decorator..wrapper) functoolswraps)rrr rrsimple_decoratorsrWorldcCsd|S)NzHello %sr )rr r rdecorated_methodsrcKs~~|dS)zFunction with kwarg. :param arg1: Description of arg1. :param arg2: Description of arg2. :key arg3: Description of arg3. r getrr r r fn_with_kwarg!srcKs~~~|dS)zFunction with kwarg and defaults. :param arg1: Description of arg1. :param arg2: Description of arg2. :key arg3: Description of arg3. r r)r r optror r rfn_with_kwarg_and_defaults,srfirstlastlatecCs~~|S)zFunction with kwarg and defaults. :key first: Description of first. :key last: Description of last. :key late: Description of late. r )rrrr r rfn_with_multiple_defaults7sr)rr)r)T)rI)r.)r)T)rrr)Hr' __future__rrrrenumrsixPY3ZfirerZpy3rintrGrrobjectrrrr r$r(r1r5r6r=r@rErHrPr[r]rarcrirmrwrzrrrrrrrrrZCALLABLE_WITH_KEYWORD_ARGUMENTrrrrdictrZSUBDICTEnumrrrrrrrrrrrrrr r r rs                       !