.Ph dZddlZddlZ ddlZn #e$rdZYnwxYwddlZejfdZdZ e dkr"ddl Z e j dZ e dd d e d d d e dedgddejdejdejdejdejdddddgdejdejejgdejddgZe ee dSdS)zP This script is used to generate test data for joblib/test/test_numpy_pickle.py Nc~d|dD}dd|DS)zNormalize joblib version by removing suffix. >>> get_joblib_version('0.8.4') '0.8.4' >>> get_joblib_version('0.8.4b1') '0.8.4' >>> get_joblib_version('0.9.dev0') '0.9' c8g|]}tjd|S)z(\d+).*)rematch).0eachs d/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/joblib/test/data/create_numpy_pickle.py z&get_joblib_version..s$PPPdrx D))PPP.c<g|]}||dS)N)group)rms r r z&get_joblib_version..s#BBBAAMQWWQZZMMMr )splitjoin)joblib_versionmatchess r get_joblib_versionrsGQPn6J6J36O6OPPPG 88BBBBB C CCr ci}|j}|j}t}dtj}dtj ddd}|r|dkrdnd}|r<|dkrd|d<d } n|d f|d<d |} |j r d |d <|dz }nd} d||||| } tj || fi|td| dS#t$r/} td| || Yd} ~ dSd} ~ wwxYw)Nz {0[0]}{0[1]}r zlib _compressedTcompressz.gzz.pkl.{}r cache_size _cache_sizez.pklzjoblib_{}{}_pickle_py{}_np{}{}z!File '{}' generated successfully.zCError: cannot generate file '{}' with arguments '{}'. Error was: {})rmethodrformatsys version_infornp __version__rrjoblibdumpprint Exception) to_pickleargskwargsrrr py_version numpy_versionbody extensionpickle_filenamees r write_test_pickler2 s F}H [F'))N&&s'788JGGBN0055bqb9::M& C&F*:*:==D  V  !%F: II"(!F: !((00I ? "#$F< M !D 6==j-O K I99&999 188IIJJJJJ      "F?FA>>          sD E$E  E__main__zJoblib pickle data generator.) descriptionz --cache_size store_truez;Force creation of companion numpy files for pickled arrays.)actionhelpz --compresszGenerate compress pickles.z--methodr)rgzipbz2xzlzmalz4zSet compression method.)typedefaultchoicesr7zrWs%  BBB '-&8 D D D D%K%K%KP zOOO $X $1P Q Q QF  J  \0L  <<< &   !828E??+++ !828E??+++!U!!,,-S999 #RX&&&..00 )))828E??333 Ii!2!2!4!455555Es