HR-efddlZddlZddlZddlZddlZddlZddlZddlmZej gdgddZ ej gdgddZ Gdd Z dS) N)fits)Fstrpathlib) home_is_datazhome_is_data, pathlib)paramsidsc|jrJ||j_|j|j|jdkdSdS)a Pytest fixture to run a test case both with and without tilde paths. In the tilde-path case, calls like self.data('filename.fits') will produce '~/filename.fits', and environment variables will be temporarily modified so that '~' resolves to the data directory. rN)paraminstance monkeypatchset_home_as_dataset_paths_via_pathlibrequestr s >lib/python3.11/site-packages/astropy/io/fits/tests/conftest.pyrrs_}K'2$))+++..w} /IJJJJJ KKc|jrJ||j_|j|j|jdkdSdS)a Pytest fixture to run a test case both with and without tilde paths. In the tilde-path case, calls like self.temp('filename.fits') will produce '~/filename.fits', and environment variables will be temporarily modified so that '~' resolves to the temp directory. These files will also be tracked so that, after the test case, we can verify no files were written to a literal tilde path. rN)r r r set_home_as_temprrs r home_is_tempr#s_}K'2$))+++..w} /IJJJJJ KKrc>eZdZdZdZdZdZdZdZdZ dZ d S) FitsTestCasectjtjtd|_t jd|_d|_ d|_ t|_ d|_ dtj_dtj_dtj_dtj_dS)Ndataz fits-test-)prefixFT)ospathjoindirname__file__data_dirtempfilemkdtemptemp_dirrrsettemp_files_used use_pathlibrconf"enable_record_valued_keyword_cardsextension_name_case_sensitivestrip_header_whitespace use_memmapselfs r setup_methodzFitsTestCase.setup_method:s RW__X%>%>GG  ( === !!"uu  8< 427 /,0 )# rc|jrO|jD]G\}}tj|rJtj|sJHt |drntj|jrJd}|rF tj|jn+#t$rtj d|dz}YnwxYw|Ftj dtj dtj dtj ddS) Nr$g?r)r*r+r,)rr&rrexistshasattrr$shutilrmtreeOSErrortimesleeprr(reset)r. temp_filetemp_file_no_tildetriess rteardown_methodzFitsTestCase.teardown_methodKsI   :151E : :- -7>>)44444w~~&8999999 4 $ $  )F)F E M$-000JsOOOQJEEE    <=== 7888 1222  %%%%%sB**#CCctjtj||tj||tjtj||tj tj zdS)zdCopies a backup of a test data file to the temp dir and sets its mode to writeable. N) r5copyrr expanduserrtempchmodstatS_IREADS_IWRITE)r.filenames r copy_filezFitsTestCase.copy_fileds  G  tyy22 3 3 G  tyy22 3 3    ##DIIh$7$788$,:VWWWWWrc|jrd}n|j}|jrtj||St j||S)z%Returns the path to a test data file.~)rr!r'rPathrrr)r.rGrs rrzFitsTestCase.datansN   #FF]F   2<11 1w||FH---rcZtj|j|}|jr=d}|jtj|||fn|j}|jrtj ||Stj||S)z5Returns the full path to a file in the test temp dir.rJ) rrrr$rr&addr'rrK)r.rG real_targetrs rrBzFitsTestCase.tempysgll4=(;;   #F  $ $bgll68&D&Dk%R S S S S]F   2<11 1w||FH---rcd|_|jd|j|jd|jdS)z This overrides the HOME environment variable, so that paths beginning with '~/' expand to the data directory. Used by the `home_is_data` fixture. THOME USERPROFILEN)rr setenvr!r-s rrzFitsTestCase.set_home_as_datasH !  666  t}=====rcd|_|jd|j|jd|jdS)az This overrides the HOME environment variable, so that paths beginning with '~/' expand to the temp directory. In conjunction with self.temp(), temporary files are tracked as they are created, so we can verify they end up in the temporary directory and not unexpected places in the filesystem. Used by the `home_is_temp` fixture. TrPrQN)rr rRr$r-s rrzFitsTestCase.set_home_as_tempsH!  666  t}=====rc||_dS)N)r')r.r's rrz"FitsTestCase.set_paths_via_pathlibs&rN) __name__ __module__ __qualname__r/r>rHrrBrrrrrrr9s$$$"&&&2XXX . . . . . . > > > > > >'''''rr) rrr5rDr"r8pytest astropy.iorfixturerrrrXrrr\s     $ $ $*W*W*WKKK" $ $ $*W*W*WKKK&j'j'j'j'j'j'j'j'j'j'r