o a&i@sdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl mZmZdd Zd d Zd d ZddZddZddZe ejfddZe ddZddZddZdS)z Utilities for testing code. N)Path)contextmanager) __version__) check_version get_loggercCsbtj|sJt| }|}Wdn1swYdddg}||ks/JdS)zN Load the tiny-data.txt file and check that the contents are correct. N z)# A tiny data file for test purposes only1 2 3 4 5 6)ospathexistsopenreadjoinstrip)fnameZtinydatacontent true_contentr`__. It matches the pooch version specified in ``pooch.version.full_version``. Returns ------- url The versioned URL for pooch's test data. zAhttps://github.com/fatiando/pooch/raw/{version}/pooch/tests/data/main)fallback)version)formatr full_versionurlrrrpooch_test_url/s r!cCd}|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/rrrrrpooch_test_figshare_urlC r#cCr")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/rrrrrpooch_test_zenodo_urlTr$r%c Csdddddddddd 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. Z@baee0894dba14b12085eacb204284b97e362f4f3e5a5807693cc90ef415c1b2dZ@98de171fb320da82982e6bf0f3994189fff4b42b23328769afce12bdd340444aZ@0d49e94f07bc1866ec57e7fd1b93a351fba36842ec9b13dd50bf94e8dfa35cbbZ@0498d2a001e71051bbd2acd2346f38da7cbd345a633cb7bf0f8a20938714b51aZ@41503f083814f43a01a8e9a30c28d7a9fe96839a99727a7fdd0acf7cd5bab63bZ@088c7f4e0f1859b1c769bb6065de24376f366374817ede8691a6ac2e49f29511Z@753663687a4040c90c8578061867d1df623e6aa8011c870a5dbd88ee3c82e306Z@2e2da6161291657617c32192dba95635706af80c6e7335750812907b58fd4b52Z@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)registryrrrpooch_test_registryes  r'ccs@t}t|}||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)levelZlog_filehandlerrrr capture_log~s   r1c cstttjd|}tj|j|}zt ||d|}|VWtj |r1t |dSdStj |r?t |ww)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__parentr r rZ anon_rootshutilcopyfiler remove)serverr package_pathZ server_pathr rrr data_over_ftps    r:cCs8|D]}|r||j|Bt||qdS)zg Recursively change the permissions on the child directories using a bitwise OR operation. N)iterdiris_dirchmodstatst_mode_recursive_chmod_directories)rootmodeitemrrrr@s  r@cCs4t|}t||j}t||t|tjd|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. )rB)rnamer5copytreer@r>S_IWUSR)source destinationZmirrorrrrmirror_directorys  rI)__doc__r r(r*r5r>pathlibr contextlibrrrutilsrrrrr!r#r%r'DEBUGr1r:r@rIrrrrs,