a j-I@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z m Z edZedZe e fZdZeedsdd e_d ed d d dfddZddZddZdDddZddZddZddZddZe Zdd Zd!d"Z dEd#d$Z!d%d&Z"d'd(Z#dFd)d*Z$d+d,Z%d-e%_&d.d/Z'd-e'_&d0d1Z(d2d3Z)d4d5Z*d6d7Z+Gd8d9d9e,Z-d:d;Z.drz|-- z| z`-- z c Csdt||||||S)N )join_ls_tree_lines)dir_path skip_patternindent branch_indent last_indentlast_branch_indentr r r ls_treesrc#s dkrtt}|gg}}|D]>} t| rBq0tjtj| rd| | q0| | q0t t dd|Ddd|D} fdd} | ddD]"\} } | | | D] } | Vqq| r| d\} } | | | D] } | VqdS)NrcSsg|] }|dfqS)Fr .0namer r r 5rz"_ls_tree_lines..cSsg|] }|dfqS)Tr rr r r r5rc3s^|s||VnJtj|}tj|sZ||Vt|}|D]}||VqJdSN)ospathrislinkr)ris_dirindZ branch_indr!Zsubtreexrrrrrrr r ls_entry8s   z _ls_tree_lines..ls_entry) r getcwdlistdirsortrematchr!isdirrappendlist itertoolschain)rrrrrrnamesdirsnondirsrentriesr'r#liner r&r r%s,        rcCsJtj|s.tjtjtjt|}|dusBtj|sFdS|Sr)r r!isabsnormpathabspathrr)r.)r!r r r absdirMs r;cCs|}|durt}t|ttfrJ|D]}t||}|dur&|Sq&dStj|sttjtj tj ||}|dustj |s|tkrtjtj tj t|}|dustj |sdStj |rtj |d}tj |r|Sntj |r |SdS)N __init__.py)r r) isinstancer0tupleabsfiler!r8r9r:rexistsr.isfile)r!whereorigZ maybe_pathZ maybe_absinitr r r r?Ws2      r?cstfdd|DS)Nc3s|]}|VqdSrr )ritem predicater r trzanyp..)any)rGiterabler rFr anypssrKcCs:tj|p8tj|p8|dp8ttj|d S)zA name is file-like if it is a path that exists, or it has a directory part, or it ends in .py, or it isn't a legal python identifier..pyr)r r!r@dirnameendswithident_rer-splitext)rr r r file_likews  rQc CsFz|jWSty@z|jjWYSty:YYdS0Yn0dS)zmGet the line number of a function. First looks for compat_co_firstlineno, then func_code.co_first_lineno.r(N)Zcompat_co_firstlinenoAttributeError__code__co_firstlinenor r r r func_linenos  rUcCst|}|tvpt|tSr)type class_types issubclass)objobj_typer r r isclasssr[cCsrtj|rntj|}t|rndD] }tjtj||r&dSq&tj drntjtj|drndSdS)zIs this path a package directory? >>> ispackage('nose') True >>> ispackage('unit_tests') False >>> ispackage('nose/plugins') True >>> ispackage('nose/loader.py') False )r<z __init__.pycz __init__.pyoTjavaz__init__$py.classF) r r!r.basenamerOr-rArsysplatform startswith)r!endrDr r r ispackages    rbcCs t|tuS)zIs this a property? >>> class Foo: ... def got(self): ... return 2 ... def get(self): ... return 1 ... get = property(get) >>> isproperty(Foo.got) False >>> isproperty(Foo.get) True )rVpropertyrYr r r ispropertys recCs\|durt}tj|tj|d}tj|dr@|S|d}tj|rX|SdS)zFind the python source file for a package, relative to a particular directory (defaults to current working directory if not given).N.z /__init__.pyrL)r r)r!rsepsplitr@)packageZ relativeTor!filenamer r r getfilenames rkcCst|}tj|s|ds*t|s*dStjtj|\}}|dkrPg}n|g}tjtj|d\}}|rttj ||r| |nqtj|\}}qr| d |S)aFind the full dotted package name for a given python source file name. Returns None if the file is not a python source file. >>> getpackage('foo.py') 'foo' >>> getpackage('biff/baf.py') 'baf' >>> getpackage('nose/util.py') 'nose.util' Works for directories too. >>> getpackage('nose') 'nose' >>> getpackage('nose/plugins') 'nose.plugins' And __init__ files stuck onto directories >>> getpackage('nose/plugins/__init__.py') 'nose.plugins' Absolute paths also work. >>> path = os.path.abspath(os.path.join('nose', 'plugins')) >>> getpackage(path) 'nose.plugins' rLN__init__rrf) srcr r!r.rNrbrPr]rhrr/reverse)rjZsrc_filebaseextZ mod_partsr!partr r r getpackages(  rrcCsRt|d}d|d}dd||d|f}dt|}|dkrN|d|}|S)zDraw a 70-char-wide divider, with label in the middle. >>> ln('hello there') '---------------------------- hello there -----------------------------' Fz%s %s %s-r)len)labelZ label_lenchunkoutpadr r r lns    r{cCs|d}|dd}|durr|rfz"td|td|}WqfWqtyb|d=|s^Yq0q|dd}|}td|||||D]}t||}q|S)a3Resolve a dotted name to a module and its parts. This is stolen wholesale from unittest.TestLoader.loadTestByName. >>> resolve_name('nose.util') #doctest: +ELLIPSIS >>> resolve_name('nose.util.resolve_name') #doctest: +ELLIPSIS rfNz __import__ %sr(zresolve: %s, %s, %s, %s)rhlogdebug __import__r ImportErrorgetattr)rmodulepartsZ parts_copyrYrqr r r resolve_name s$       rcCs:tjj}|}d}d|vr8t|r.||ddfSd|dfStj|\}}|sz$|d\}}t|rn|d}}WnXty|d}t|ddkrd|dd|d}}ntd|fYn0n6|s|}n,d|vr|d\}}n|}tj||g}|r,t|r ||d|fSd||fSn dd|fSdS)a+Split a test name into a 3-tuple containing file, module, and callable names, any of which (but not all) may be blank. Test names are in the form: file_or_module:callable Either side of the : may be dotted. To change the splitting behavior, you can alter nose.util.split_test_re.N:rr|r(zaTest name '%s' could not be parsed. Please format test names as path:callable or module:callable.) r r!r9rQrh ValueErrorrvrrg)testZnormZ file_or_modfnheadtailrZ file_partr r r split_test_name&s>     rFcCst|dr|St|}d}}}|tjkrVt|dd}t|dd}t|||fS|tjksrt|tsr|tkrt|dd}|durt j |}t|dd}|durt j |}t|dd}t|||fS|tjkrt|jj}t|d|dd|d |jffSt|tjrt|d s,t|d rXz t|jWStyVt|jYS0t|j}z |j}Wnty|j}Yn0t|d|dd|d |ffSt|d r|jjd vrt|jStd||fdS)zaFind the test address for a test, which may be a module, filename, class, method or function.addressN__file____name__ __module__rr|z%s.%srs_FunctionTestCase__testFunc _testFunc __class__) __builtin__builtinszI don't know what %s is (%s))hasattrrrVtypes ModuleTyperrm FunctionTyperXr^modulesr r!r: MethodType test_address__self__rrr=unittestZTestCaserrRrZ_TestCase__testMethodNameZ_testMethodNamer TypeError)rtfilercallmZcls_adr method_namer r r rWsb                   rc Cs|D]}t||d}|durt|tjrt|tjrHt|\}}}}n\t|r`t|s`|j }z t|\}}}}| dWn"t yt d||fYn0t |rt d|||||St d|||SqdS)zGiven a list of possible method names, try to run them with the provided object. Keep going until something works. Used to run setup/teardown methods for module, package, and function tests.NrzaAttribute %s of %r is not a python function. Only functions or callables may be used as fixtures.zcall fixture %s.%s(%s)zcall fixture %s.%s)rr=rrrrrcallableismethod__call__poprrvr}r~)rYr3rr argsvarargsvarkwdefaultsr r r try_runs2        rcCsb|dur |Stjdr8|dr8d|dddfStj|\}}|dvr^d|dfS|S)zFind the python source file for a .pyc, .pyo or $py.class file on jython. Returns the filename provided if it is not a python source file.Nr\z $py.classrfipy)z.pycz.pyorL)r^r_r`rNrr r!rP)rjrorpr r r rmsrmcsfdd}|S)aSort key function factory that puts items that match a regular expression last. >>> from nose.config import Config >>> from nose.pyversion import sort_list >>> c = Config() >>> regex = c.testMatch >>> entries = ['.', '..', 'a_test', 'src', 'lib', 'test', 'foo.py'] >>> sort_list(entries, regex_last_key(regex)) >>> entries ['.', '..', 'foo.py', 'lib', 'src', 'a_test', 'test'] cs|rd|fSd|fS)Nr|r)searchrdregexr r ks zregex_last_key..kr )rrr rr regex_last_keys rcCs^|dur dSz|g|WSty.Yn0ztd|WStyXt|YS0dS)arConvert a value that may be a list or a (possibly comma-separated) string into a list. The exception: None is returned as None, not [None]. >>> tolist(["one", "two"]) ['one', 'two'] >>> tolist("hello") ['hello'] >>> tolist("separate,values, with, commas, spaces , are ,ok") ['separate', 'values', 'with', 'commas', 'spaces', 'are', 'ok'] Nz\s*,\s*)extendrRr,rhrr0)valr r r tolists    rcseZdZdZfddZfddZfddZdd Zfd d Zfd d Z ddZ ddZ dfdd Z fddZ ddZZS)odictzqSimple ordered dict implementation, based on: http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/107747 cs g|_tt|j|i|dSr)_keyssuperrrl)selfargkwrr r rlszodict.__init__cs tt|||j|dSr)rr __delitem__rremove)rkeyrr r rszodict.__delitem__cs,tt|||||jvr(|j|dSr)rr __setitem__rr/)rrrErr r rs zodict.__setitem__cCs ddddt|DS)Nz{%s}z, cSsg|]\}}d||fqS)z%r: %rr )rrvr r r rrz!odict.__str__..)rr0itemsrr r r __str__sz odict.__str__cstt|g|_dSr)rrclearrrrr r rsz odict.clearcs"tt|}|jdd|_|Sr)rrcopyr)rdrr r rsz odict.copycCstt|jt|Sr)r0ziprvaluesrr r r rsz odict.itemscCs|jddSr)rrr r r keys sz odict.keysNcs,tt|||}||jvr(|j||Sr)rr setdefaultrr/)rrfailobjrErr r r s  zodict.setdefaultcs<tt||t|D]}||jvr|j|qdSr)rrupdater0rrr/)rdictrrr r rs z odict.updatecCstt|j|jSr)r0mapgetrrr r r rsz odict.values)N)rr __qualname____doc__rlrrrrrrrrrr __classcell__r r rr rs      rcsDddlm}tr"fdd}n fdd}||}||_|S)aMake a function imported from module A appear as if it is located in module B. >>> from pprint import pprint >>> pprint.__module__ 'pprint' >>> pp = transplant_func(pprint, __name__) >>> pp.__module__ 'nose.util' The original function is not modified. >>> pprint.__module__ 'pprint' Calling the transplanted function calls the original. >>> pp([1, 2]) [1, 2] >>> pprint([1,2]) [1, 2] r)make_decoratorc?s|i|EdHdSrr rrr r r newfunc2sz transplant_func..newfunccs|i|Srr rr r r r5s)Z nose.toolsrrr)r rrrr r r transplant_funcs   rcCs"Gddd|}||_|j|_|S)a7Make a class appear to reside in `module`, rather than the module in which it is actually defined. >>> from nose.failure import Failure >>> Failure.__module__ 'nose.failure' >>> Nf = transplant_class(Failure, __name__) >>> Nf.__module__ 'nose.util' >>> Nf.__name__ 'Failure' c@s eZdZdS)ztransplant_class..CN)rrrr r r r CHsr)rr)clsrrr r r transplant_class=s rutf-8csVz t|WStyPt|tr>dfdd|DYSt|YS0dS)N csg|]}t|qSr )safe_str)rrencodingr r rUrzsafe_str..)strUnicodeEncodeErrorr= Exceptionrencode)rrr rr rPs    rcCs6tj|sdSt|}t|jtjtjBtjB@S)NF) r r!r@statboolst_modeS_IXUSRS_IXGRPS_IXOTH)rstr r r is_executableYs  r__main__)N)N)N)r)5rrr1loggingrr r,r^rrZnose.pyversionrrr getLoggerr}compilerOrWrrrrrr;r?rKrQrUr[Z is_generatorrbrerkrrr{rrZ__test__rrrmrrrrrrrrrdoctesttestmodr r r r sh     (    2 .4$4!