o i]+ @sdZddlmZddlmZddlZddlZddlmZddl m Z m Z m Z dd l mZmZmZmZeZeZejjejjd gd gd d ddZejjddZejjejjdddgddgd ejjdddgfdddgfddgfddgfdddgfggdd ejjde d fe d!fgd"d#gd d$d%Zejjejde d fe d!fgd&d'Zejjejde d fe d!fgd(d)Zd*d+Zd,d-Zejjejde d fe d!fgd.d/Zejjejde d fe d!fgd0d1ZdS)2z Test the processor hooks )Path)TemporaryDirectoryN)Pooch)UnzipUntar Decompress)pooch_test_urlpooch_test_registrycheck_tiny_data capture_logzmethod,ext,name))autoxzN)lzmarN)rrN)bzip2bz2N)gzipgzN)rrzdifferent-name.txt)rrrrrname)Zidsc Cs~t||d}t}t|}|durt|dd|dg}nt||}t|ttd}t>}|j d||d} | } | } t | d ksIJ| d  d d ksUJ| d d s^J|| d vsfJWdn1spwY| |ks{Jt| t}|j d||d} | dksJWdn1swY| |ksJt| WddS1swYdS)z=Check that decompression after download works for all formats)methodrN. tiny-data.txtdecomppathbase_urlregistryztiny-data.txt. processorrr DownloadingZ Decompressing)rrrstrjoinrBASEURLREGISTRYr fetchgetvalue splitlineslensplit startswithr ) rextrr local_storer true_pathpuplog_filefnameZlogslinesr4j/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/pooch/tests/test_processors.pytest_decompresss2     "r6c Cst8}t|}t|ttd}tt&}t |j dt dddWdn1s/wYWdn1s>wY|j j ddsNJd |j j dvsXJtt&}t |j dt d ddWdn1swwYWdn1swY|j j dd sJd |j j dvsJtt&}t |j dt d ddWdn1swYWdn1swY|j j dd sJd |j j dvsJtt'}t |j dt dddWdn 1swYWdn 1swY|j j dds)Jd |j j dvs4JWddS1s@wYdS)zGShould fail if method='auto' and no extension is given in the file namerrr)rrNrz"Unrecognized file extension '.txt'zpooch.Unzip/UntarZblaz Invalid compression method 'bla'zipz Invalid compression method 'zip'z store.zipz"Unrecognized file extension '.zip')rrrr%r&pytestZraises ValueErrorwarningscatch_warningsr'rvalueargsr,)r.rr0 exceptionr4r4r5test_decompress_failsEsH         $r? target_pathZsome_custom_pathZ default_pathZ custom_pathzarchive,members tiny-datar)storeNrBstore/tiny-data.txtstore/subdir/tiny-data.txtz store/subdir)Z single_fileZ archive_allZ archive_fileZarchive_subdir_fileZarchive_subdirZarchive_multiplezprocessor_class,extensionz.zipz.tar.gzrrc CsD|||d}|dur|||j}t}t|}t|||||j\}}t|ttd} t} | j |||d} t | |ksAJt | |Wdn1sPwY| D]} t | qWt} | j |||d} t | |kssJt | gWdn1swY| D]} t | qWddS1swYdS)zGTests the behaviour of processors for unpacking archives (Untar, Unzip))members extract_dirNrr) suffixrr"_unpacking_expected_paths_and_logs__name__rr%r&r r'set _check_logsr ) processor_class extensionr@archiverErr true_pathsZ expected_logr0r1fnamesr2r4r4r5test_unpackingds2     "rQc Cstj}tt|ttd}|dgd}|jd||d}t|dks$Jt|d|ddgd}|jd||d}t|d ksBJt|dt|d|jd||d}t|dks_Jt|dWd d S1spwYd S) zHTest that multiple subsequent calls to a processor yield correct resultsrrCrErBrr rrDrNrrrr%r&r'r*r ) rLrMr.r0 processor1 filenames1 processor2 filenames2Z filenames3r4r4r5test_multiple_unpackings"    "rXcCsxt/}tt|ttd}|dgd}|jd||d}t|dks$Jt|dWddS1s5wYdS) zDTest that unpack members can also be specifed both with a leading ./rz./store/tiny-data.txtrRrBrr rNrS)rLrMr.r0rTrUr4r4r5$test_unpack_members_with_leading_dots "rYcCsF|}t|t|ksJt||D] \}}||s JqdS)zC Assert that the lines in the log match the expected ones. N)r(r)r*r7r,)r1Zexpected_linesr3lineZ expected_liner4r4r5rKs rKcCsdg}|dkrt|dh}|d||fS|dkrA|durAt|ddt|dddh}|||dd ||fS|dkrx|durxg}|D]&}|t|d }t|dsc|d}|t||d |d qMt|}||fS) zo Generate the appropriate expected paths and log message depending on the parameters for the test. r rArzExtracting 'tiny-data.txt'rBNsubdirr!z ing contents/z Extracting '')r#appendrr+endswithrJ)rNrErr log_linesrOmemberr/r4r4r5rHs*  rHcCt=}tt|ttd}|dgd}|jd||d}t|dks$J|}|jd||d}t|dks8JWddS1sCwYdS)zx Test that calling with valid members then without them works. https://github.com/fatiando/pooch/issues/364 rrCrRrBrr Nrrrr%r&r'r*rLrMr.r0rTrUrVrWr4r4r5&test_unpacking_members_then_no_members  "recCrb)zz Test that calling with invalid members then without them works. https://github.com/fatiando/pooch/issues/364 rznot-a-valid-file.csvrRrBrrNrcrdr4r4r5,test_unpacking_wrong_members_then_no_members rfrg) __doc__pathlibrtempfilerr:r8r"rZ processorsrrrutilsr r r r r&r%marknetworkZ parametrizer6r?rQrXrYrKrHrergr4r4r4r5s