U `{jq@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_loggerc CsRtj|stt|dd}|}W5QRXdddg}||ksNtdS)zN Load the tiny-data.txt file and check that the contents are correct. utf-8encoding z)# A tiny data file for test purposes only1 2 3 4 5 6N)ospathexistsAssertionErroropenreadjoinstrip)fnameZtinydatacontent true_contentrP/var/www/html/bwave/python/venv/lib/python3.8/site-packages/pooch/tests/utils.pycheck_tiny_datasrc Cs`tj|stt|dd}|}W5QRXdg}|dgd|d|ks\tdS)zO Load the large-data.txt file and check that the contents are correct. rr z*# A larer data file for test purposes onlyr irr N) r rrrrrextendrr)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#cCsd}|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_urlBs r%cCsd}|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/rr$rrrpooch_test_zenodo_urlSs r&cCsd}|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/rr$rrr pooch_test_zenodo_with_slash_urldsr'cCsd}|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/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    r4c csdtttjd|}tj|j|}z t ||d|}|VW5tj|r^t |XdS)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_rootrremoveshutilcopyfile)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)rootmodeitemrrrrCs rCcCs4t|}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. )rE)rnamer9copytreerCrAS_IWUSR)sourceZ destinationZmirrorrrrmirror_directorys  rK)__doc__r r+r-r9rApathlibr contextlibrrr utilsrrrrr#r%r&r'r(r*DEBUGr4r=rCrKrrrrs.