B Ž[x@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z mZddlmZddlmZmZGdddejZd d d ZdS) zBase test class for nbconvertN)PopenPIPE)v4write)TemporaryWorkingDirectory) string_types bytes_to_strc@sPeZdZdZdddZddZdd d Zd d ZdddZddZ dddZ dS) TestsBasezPBase tests class. Contains useful fuzzy comparison and nbconvert functions.TFc Cs| s|}|}|r0|dd}|dd}|rL|dd}|dd}|rh|dd}|dd}|r|dd}|dd}|r||dd}||dd}|s|}|}|||dS)a 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.    z N)stripreplacerecursive_replacelowerZ assertEqual) selfabZnewlines_are_spacesZtabs_are_spacesZ fuzzy_spacingZ ignore_spacesZignore_newlinesZcase_sensitiveZ leave_paddingr3lib/python3.7/site-packages/nbconvert/tests/base.py fuzzy_compares,         zTestsBase.fuzzy_comparecCsx||kr|||}qW|S)a 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)rtextsearchZ replacementrrrrCs zTestsBase.recursive_replaceNcCs"t}|dk r|j||jd|S)N)dest)r copy_files_toname)rcopy_filenamesZtemp_dirrrrcreate_temp_cwd[szTestsBase.create_temp_cwdc Cs4t}tj|ddd}t||dWdQRXdS)Nwzutf-8)encoding)rZ new_notebookioopenr)rpathZnbfrrrcreate_empty_notebookeszTestsBase.create_empty_notebook.c Csxtj|st||}xT|D]L}ttj||}|sDtx*|D]"}t |tj|tj |qJWq$WdS)z.Copy test files into the destination directoryN) osr$isdirmakedirs_get_files_pathglobjoinAssertionErrorshutilZcopyfilebasename)rrrZ files_pathpatternfilesmatchrrrrjs    zTestsBase.copy_files_tocCs>|jddd}|dtjtj}tjj|f|S)Nr'r2) __module__splitappendr(r$dirname nbconvert__file__r-)rnamesr$rrrr+us zTestsBase._get_files_pathcCsvt|trt|}tjddg|}t|tttd}|j|d\}}|j dks^|s^t t || dd| ddfS)a 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 z-mr:)stdoutstderrstdin)inputrutf8r) isinstancershlexr7sys executablerrZ communicate returncodeOSErrorrdecode)rZ parametersZignore_return_coder?cmdpr=r>rrrr:s   zTestsBase.nbconvert)TTTFFFF)N)r')FN) __name__r6 __qualname____doc__rrrr&rr+r:rrrrr s &   r PcCsxRtdt||D]>}||||}||||}||kstd|||fqWt|t|krtdt|t||t|dfn6t|t|krtdt|t||t|dfdS)zassert that large strings are equal Zooms in on first chunk that differs, to give better info than vanilla assertEqual for large text blobs. rz[offset: %i] %r != %rz.Length doesn't match (%i > %i). Extra text: %rNz.Length doesn't match (%i < %i). Extra text: %r)rangelenr.)rrZ chunk_sizeiZchunk_aZchunk_brrrassert_big_text_equals$rR)rN)rMr"r(r,rCr/rDZunittestr: subprocessrrZnbformatrrZtestpath.tempdirrZipython_genutils.py3compatrrZTestCaser rRrrrrs