:Mc`dZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZmZddlZddlmZGdd ejZd d ZdS) zBase test class for nbconvertN)PIPEPopen)TemporaryDirectory)v4write)_contextlib_chdirceZdZdZ ddZdZejddZe dZ d Z dd Z d Z dd ZdS) TestsBasezPBase tests class. Contains useful fuzzy comparison and nbconvert functions.TFc | s(|}|}|r,|dd}|dd}|r,|dd}|dd}|r,|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_replacelower assertEqual) selfabnewlines_are_spacestabs_are_spaces fuzzy_spacing ignore_spacesignore_newlinescase_sensitive leave_paddings 4lib/python3.11/site-packages/nbconvert/tests/base.py fuzzy_comparezTestsBase.fuzzy_comparesR&  A A  $ $##A $##A  % $$$A $$$A  % $$$A $$$A  # #r""A #r""A  5&&q$44A&&q$44A  A A AcB||vr|||}||v|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)rtextsearch replacements r rzTestsBase.recursive_replaceLs6(n 5<< 44Dn 5 r"Nc#Kt5}tj|5|||||Vdddn #1swxYwYddddS#1swxYwYdS)N)dest)rr chdir copy_files_to)rcopy_filenamestds r create_temp_cwdzTestsBase.create_temp_cwdds  ! ! R):)@)D)D   <"">";;;HHH                                 s4A(A A(A A(A A((A,/A,c>i}|D]}|||SN)update)cls dict_argsoutcomeds r merge_dictszTestsBase.merge_dictsks2  A NN1    r"ctj}t|dd5}t||dddddS#1swxYwYdS)Nwzutf-8)encoding)r new_notebookopenr)rpathnbfs r create_empty_notebookzTestsBase.create_empty_notebookss _   $g . . . ! "aOOO                  sAA  A .c tj|stj||}|D]}t jtj||}|sJ|D]R}tj|tj|tj |SdS)z.Copy test files into the destination directoryN) osr<isdirmakedirs_get_files_pathglobjoinshutilcopyfilebasename)rr+r( files_pathpatternfilesmatchs r r*zTestsBase.copy_files_toxsw}}T""  K   ))++ % T TGIbgll:w??@@ELLL T Trw||D"':J:J5:Q:Q'R'RSSSS T T Tr"c|jddd}|dtjt j}tjj|g|RS)Nr@rM) __module__splitappendrBr<dirname nbconvert__file__rG)rnamesr<s r rEzTestsBase._get_files_pathsf%%c**1R40 Wwy122w|D)5))))r"cbtjddg}tjdkrKt|tfrd|dz|z}nHd||z}n/t|tfrt j|}||z }t|ttt}| |\}}|j dks%|s#t| dd | dd | dd fS) 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-mrVwin32r)stdoutstderrstdin)inputrutf8r)sys executableplatform isinstancestrrGshlexrSrr communicate returncodeOSErrordecode)r parametersignore_return_coder]cmdpr[r\s r rVzTestsBase.nbconverts~t[1 <7 " *sf-- 1hhsmmc)J6hhsZ/00*sf-- 5"[44 : C #d4t < < <U33 ! <%7 <&-- ::;; ;}}VY//vy1Q1QQQr")TTTFFFFr/)r@)FN)__name__rR __qualname____doc__r!r contextlibcontextmanagerr- classmethodr5r?r*rErVr"r r r s!////b0 [ T T T T * * *RRRRRRr"r Pc 8tdt||D]2}||||z}||||z}||ksJd|||fz3t|t|krDtdt|t||t|dfzt|t|krDtdt|t||t|dfzdS)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)rangelenAssertionError)rr chunk_sizeichunk_achunk_bs r assert_big_text_equalr~s8 1c!ffj ) )VVAJ&'AJ&''!UU#=GW@U#UUUUU 1vvA  =QQQRSVWXSYSYS[S[Q\@] ]    Q#a&&  =QQQRSVWXSYSYS[S[Q\@] ]     r")ru)rprqrFrBrerHr`unittest subprocessrrtempfilernbformatrrrVutilsr TestCaser r~rtr"r rs ##  """"""""''''''%%%%%%RRRRRRRRRR!RRRRRRj      r"