-PhqdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl mZmZdZd Zd Zd Zd Zd ZdZdZe ejfdZe dZdZdZdS)z Utilities for testing code. N)Path)contextmanager) __version__) check_version get_loggerc$tj|sJt|d5}|}dddn #1swxYwYdddg}||ksJdS)zN Load the tiny-data.txt file and check that the contents are correct. utf-8encodingN z)# A tiny data file for test purposes only1 2 3 4 5 6)ospathexistsopenreadjoinstrip)fnametinydatacontent true_contents Q/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pooch/tests/utils.pycheck_tiny_datars 7>>%    eg & & &"(--//"""""""""""""""99 46HIL ==??l * * * * * *AAAcTtj|sJt|d5}|}dddn #1swxYwYdg}|dgdz|d|ksJdS)zO Load the large-data.txt file and check that the contents are correct. r r Nz*# A larer data file for test purposes onlyrirr )rrrrrextendrr)rdatarrs rcheck_large_datar #s 7>>%    eg & & &$))++@AL+,t3444 ==??dii 55 5 5 5 5 5 5rc>ttd}d|d}|S)as Get the base URL for the test data used in Pooch itself. The URL is a GitHub raw link to the ``pooch/tests/data`` directory from the `GitHub repository `__. It matches the pooch version specified in ``pooch.version.full_version``. Returns ------- url The versioned URL for pooch's test data. main)fallbackz&https://github.com/fatiando/pooch/raw/z/pooch/tests/data/)r full_version)versionurls rpooch_test_urlr'/s+L6:::G N7 N N NC Jc d}|S)z Get the base URL for the test data stored in figshare. The URL contains the DOI for the figshare dataset using the appropriate version for this version of Pooch. Returns ------- url The URL for pooch's test data. z$doi:10.6084/m9.figshare.14763051.v1/r&s rpooch_test_figshare_urlr,Bs 1C Jr(c d}|S)z Get the base URL for the test data stored in Zenodo. The URL contains the DOI for the Zenodo dataset using the appropriate version for this version of Pooch. Returns ------- url The URL for pooch's test data. zdoi:10.5281/zenodo.4924875/r*r+s rpooch_test_zenodo_urlr.Ss (C Jr(c d}|S)aj Get base URL for test data in Zenodo, where the file name contains a slash The URL contains the DOI for the Zenodo dataset that has a slash in the filename (created with the GitHub-Zenodo integration service), using the appropriate version for this version of Pooch. Returns ------- url The URL for pooch's test data. zdoi:10.5281/zenodo.7632643/r*r+s r pooch_test_zenodo_with_slash_urlr0ds (C Jr(c d}|S)z Get the base URL for the test data stored on a DataVerse instance. Returns ------- url The URL for pooch's test data. zdoi:10.11588/data/TKCFEF/r*r+s rpooch_test_dataverse_urlr2vs &C Jr(c  dddddddddd d }|S) z Get a registry for the test data used in Pooch itself. Returns ------- registry Dictionary with pooch's test data files and their hashes. @baee0894dba14b12085eacb204284b97e362f4f3e5a5807693cc90ef415c1b2d@98de171fb320da82982e6bf0f3994189fff4b42b23328769afce12bdd340444a@0d49e94f07bc1866ec57e7fd1b93a351fba36842ec9b13dd50bf94e8dfa35cbb@0498d2a001e71051bbd2acd2346f38da7cbd345a633cb7bf0f8a20938714b51a@41503f083814f43a01a8e9a30c28d7a9fe96839a99727a7fdd0acf7cd5bab63b@088c7f4e0f1859b1c769bb6065de24376f366374817ede8691a6ac2e49f29511@753663687a4040c90c8578061867d1df623e6aa8011c870a5dbd88ee3c82e306@2e2da6161291657617c32192dba95635706af80c6e7335750812907b58fd4b52@99dcb5c32a6e916344bacb4badcbc2f2b6ee196977d1d8187610c21e7e607765) z tiny-data.txtzlarge-data.txtzsubdir/tiny-data.txtz tiny-data.zipz store.zipztiny-data.tar.gzz store.tar.gzztiny-data.txt.bz2ztiny-data.txt.gzztiny-data.txt.xzr*)registrys rpooch_test_registryr>s6\\ b[W^Z_^^  H Or(c#Ktj}tj|}||t ||Vt |dS)z Create a context manager for reading from the logs. Yields ------ log_file : StringIO a file-like object to which the logs were written N)ioStringIOlogging StreamHandlersetLevelr addHandler removeHandler)levellog_filehandlers r capture_logrJsv{}}H#H--G ULLG$$$ NNNLLw'''''r(c#Ktttjdz |z }tj|j|} tj ||d|}|Vtj |rt j |dSdS#tj |rt j |wwxYw)a Add a test data file to the test FTP server and clean it up afterwards. Parameters ---------- server The ``ftpserver`` fixture provided by pytest-localftpserver. fname : str The name of a file *relative* to the test data folder of the package (usually just the file name, not the full path). Yields ------ url : str The download URL of the data file from the test FTP server. rzftp://localhost/N) strr__file__parentrrr anon_rootshutilcopyfilerremove)serverr package_path server_pathr&s r data_over_ftprVs&tH~~,v5=>>L',,v/77K# k222((( 7>>+ & & # Ik " " " " " # #27>>+ & & # Ik " " " " #s B**6C c|D]U}|r?||j|zt ||VdS)zg Recursively change the permissions on the child directories using a bitwise OR operation. N)iterdiris_dirchmodstatst_mode_recursive_chmod_directories)rootmodeitems rr]r]sh  55 ;;== 5 JJtyy{{*T1 2 2 2 (t 4 4 455r(ct|}t||jz }tj||t |t j|S)a Copy contents of the source directory into destination and fix permissions. Parameters ---------- source : str, :class:`pathlib.Path` Source data directory. destination : str, :class:`pathlib.Path` Destination directory that will contain the copy of source. The actual source directory (not just it's contents) is copied. Returns ------- mirror : :class:`pathlib.Path` The path of the mirrored output directory. )r_)rnamerPcopytreer]r[S_IWUSR)source destinationmirrors rmirror_directoryrhsO$&\\F +   ,F OFF### dl;;;; Mr()__doc__rr@rBrPr[pathlibr contextlibrrr$utilsrrrr r'r,r.r0r2r>DEBUGrJrVr]rhr*r(rrosl  %%%%%%******-------- + + + 6 6 6&""$   2m(((("###:555r(