o iq@sdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl mZmZdd Zd d Zd d ZddZddZddZddZddZe ejfddZe ddZddZddZdS) z Utilities for testing code. N)Path)contextmanager) __version__) check_version get_loggercCsftj|sJt|dd }|}Wdn1swYdddg}||ks1JdS)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)fnameZtinydatacontent true_contentr`/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/pooch/tests/utils.pycheck_tiny_datas rcCsttj|sJt|dd }|}Wdn1swYdg}|dgd|d|ks8JdS)zO Load the large-data.txt file and check that the contents are correct. rr Nz*# A larer data file for test purposes onlyr irr )r rrrrextendrr)rdatarrrrrcheck_large_data#s rcCsttdd}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)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/rr!rrrpooch_test_figshare_urlB 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/rr$rrrpooch_test_zenodo_urlSr&r'cCr#)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/rr$rrr pooch_test_zenodo_with_slash_urldsr(cCr#)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/rr$rrrpooch_test_dataverse_urlvs 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_registrys  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_logs   r5c 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 rrZ anon_rootshutilcopyfilerremove)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)rootmodeitemrrrrDs  rDcCs4t|}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. )rF)rnamer9copytreerDrBS_IWUSR)source destinationZmirrorrrrmirror_directorys  rM)__doc__r r,r.r9rBpathlibr contextlibrrrutilsrrrrr"r%r'r(r)r+DEBUGr5r>rDrMrrrrs0