iqdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl mZmZdZd Zd Zd Zd Zd ZdZdZe ej0fdZe dZdZdZy)z Utilities for testing code. N)Path)contextmanager) __version__) check_version get_loggerctjj|sJt|d5}|j }ddddj ddg}j |k(sJy#1swY2xYw)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 f/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pooch/tests/utils.pycheck_tiny_datarsl 77>>%  eg &(--/ '99 46HIL ==?l ** * ' &s A11A:c"tjj|sJt|d5}|j }ddddg}|j dgdzj dj|k(sJy#1swYFxYw)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{ 77>>%  eg &$))+ '@AL+,t34 ==?dii 5 55 5 ' &s BBc4ttd}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 27);M 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_urlr/ds (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_urlr1vs &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=s1\\ b[W^Z_^^ H Or'c#Ktj}tj|}|j |t j ||t j|yw)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_logrIsV{{}H##H-G ULG$ NLw'sA2A4c#Ktttjdz |z }tj j |j|} tj||d|}|tj j|rt j|yy#tj j|rt j|wwxYww)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_ftprUs&tH~,,v5=>L'',,v//7K# k2 ( 77>>+ & IIk " '277>>+ & IIk " 'sAC%B+56C%+7C""C%c|jD]K}|js|j|jj|zt ||My)zg Recursively change the permissions on the child directories using a bitwise OR operation. N)iterdiris_dirchmodstatst_mode_recursive_chmod_directories)rootmodeitems rr\r\sD   ;;= JJtyy{**T1 2 (t 4r'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^)rnamerOcopytreer\rZS_IWUSR)source destinationmirrors rmirror_directoryrgsA$&\F +  ,F OOFF# dll; Mr')__doc__rr?rArOrZpathlibr contextlibrrr#utilsrrrrr&r+r-r/r1r=DEBUGrIrUr\rgr)r'rrns  %*- + 6&""$ 2mm(("##:5r'