ó §‰]c@sádZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z mZddlmZddlmZmZdejfd„ƒYZd d „ZdS( sBase test class for nbconvertiÿÿÿÿN(tPopentPIPE(tv4twrite(tTemporaryWorkingDirectory(t string_typest bytes_to_strt TestsBasecBs}eZdZeeeeeeed„Zd„Zd d„Ze d„ƒZ d„Z dd„Z d„Z ed d „ZRS( sPBase tests class. Contains useful fuzzy comparison and nbconvert functions.c Cs=| s!|jƒ}|jƒ}n|rN|jddƒ}|jddƒ}n|r{|jddƒ}|jddƒ}n|r¨|jddƒ}|jddƒ}n|rÕ|jddƒ}|jddƒ}n|r|j|ddƒ}|j|ddƒ}n|s)|jƒ}|jƒ}n|j||ƒdS(s Performs a fuzzy comparison of two strings. A fuzzy comparison is a comparison that ignores insignificant differences in the two comparands. The significance of certain differences can be specified via the keyword parameters of this method. s tt s s N(tstriptreplacetrecursive_replacetlowert assertEqual( tselftatbtnewlines_are_spacesttabs_are_spacest fuzzy_spacingt ignore_spacestignore_newlinestcase_sensitivet leave_padding((s3lib/python2.7/site-packages/nbconvert/tests/base.pyt fuzzy_compares,   cCs)x"||kr$|j||ƒ}qW|S(sµ Performs a recursive replacement operation. Replaces all instances of a search string in a text string with a replacement string until the search string no longer exists. Recursion is needed because the replacement string may generate additional search strings. For example: Replace "ii" with "i" in the string "Hiiii" yields "Hii" Another replacement cds "Hi" (the desired output) Parameters ---------- text : string Text to replace in. search : string String to search for within "text" replacement : string String to replace "search" with (R (Rttexttsearcht replacement((s3lib/python2.7/site-packages/nbconvert/tests/base.pyR CscCs2tƒ}|dk r.|j|d|jƒn|S(Ntdest(RtNonet copy_files_totname(Rtcopy_filenamesttemp_dir((s3lib/python2.7/site-packages/nbconvert/tests/base.pytcreate_temp_cwd[s  cGs(i}x|D]}|j|ƒq W|S(N(tupdate(tclst dict_argstoutcometd((s3lib/python2.7/site-packages/nbconvert/tests/base.pyt merge_dictses cCsAtjƒ}tj|dddƒ}t||dƒWdQXdS(Ntwtencodingsutf-8i(Rt new_notebooktiotopenR(Rtpathtnbtf((s3lib/python2.7/site-packages/nbconvert/tests/base.pytcreate_empty_notebookms t.cCs©tjj|ƒs"tj|ƒn|jƒ}xt|D]l}tjtjj||ƒƒ}|set‚x9|D]1}tj |tjj|tjj |ƒƒƒqlWq5WdS(s.Copy test files into the destination directoryN( tosR/tisdirtmakedirst_get_files_pathtglobtjointAssertionErrortshutiltcopyfiletbasename(RR!Rt files_pathtpatterntfilestmatch((s3lib/python2.7/site-packages/nbconvert/tests/base.pyRrs    cCsN|jjdƒdd!}|jdƒtjjtjƒ}tjj||ŒS(NR3iiÿÿÿÿR@( t __module__tsplittappendR4R/tdirnamet nbconvertt__file__R9(RtnamesR/((s3lib/python2.7/site-packages/nbconvert/tests/base.pyR7}s cCs tjddg}tjdkr`t|tƒrJdj|ƒd|}q‹dj||ƒ}n+t|tƒrtj|ƒ}n||7}t|dt dt dt ƒ}|j d|ƒ\}}|j d kpÓ|sët t |ƒƒ‚n|jd d ƒ|jd d ƒfS( s± Run nbconvert as a shell command, listening for both Errors and non-zero return codes. Returns the tuple (stdout, stderr) of output produced during the nbconvert run. Parameters ---------- parameters : str, list(str) List of parameters to pass to IPython. ignore_return_code : optional bool (default False) Throw an OSError if the return code s-mRFtwin32R tstdouttstderrtstdintinputitutf8R (tsyst executabletplatformt isinstanceRR9tshlexRCRRt communicatet returncodetOSErrorRtdecode(Rt parameterstignore_return_codeRLtcmdtpRJRK((s3lib/python2.7/site-packages/nbconvert/tests/base.pyRFˆs  N(t__name__RBt__doc__tTruetFalseRR RR#t classmethodR)R2RR7RF(((s3lib/python2.7/site-packages/nbconvert/tests/base.pyRs&    iPcCsxgtdt|ƒ|ƒD]M}||||!}||||!}||kstd|||fƒ‚qWt|ƒt|ƒkr´tdt|ƒt|ƒ|t|ƒfƒ‚nJt|ƒt|ƒkrþtdt|ƒt|ƒ|t|ƒfƒ‚ndS(sšassert that large strings are equal Zooms in on first chunk that differs, to give better info than vanilla assertEqual for large text blobs. is[offset: %i] %r != %rs.Length doesn't match (%i > %i). Extra text: %rs.Length doesn't match (%i < %i). Extra text: %rN(trangetlenR:(RRt chunk_sizetitchunk_atchunk_b((s3lib/python2.7/site-packages/nbconvert/tests/base.pytassert_big_text_equal¦s),(R]R-R4R8RSR;ROtunittestRFt subprocessRRtnbformatRRttestpath.tempdirRtipython_genutils.py3compatRRtTestCaseRRg(((s3lib/python2.7/site-packages/nbconvert/tests/base.pyts