U ½mœdÁã@sddlZddlZddlmZmZe ejjdkd¡Gdd„deƒƒZ e ejjdkd¡Gd d „d eƒƒZ e ejjdkd¡Gd d „d eƒƒZ dS) éNé)ÚutÚTestCase)ré é²z6SWMR is available. Skipping backwards compatible testsc@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)ÚTestSwmrNotAvailablezÝ Test backwards compatibility behaviour when using SWMR functions with an older version of HDF5 which does not have this feature available. Skip this test if SWMR features *are* available in the HDF5 library. cCs8t |¡t d¡ d¡|_|jjddd|jd|_dS)Né ÚfÚdata©r©N©ÚchunksÚmaxshaper ) rÚsetUpÚnpÚarangeÚastyper r Úcreate_datasetÚdset©Úself©rúU/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/h5py/tests/test_dataset_swmr.pyrs zTestSwmrNotAvailable.setUpc Cs>|jj}|j ¡| t¡tj|ddd|_W5QRXdS)NÚrT©Zswmr)r ÚfilenameÚcloseÚ assertRaisesÚ ValueErrorÚh5pyÚFile©rÚfnamerrrÚtest_open_swmr_raisess  z*TestSwmrNotAvailable.test_open_swmr_raisesc Cs$| t¡|j ¡W5QRXdS)zd If the SWMR feature is not available then Dataset.refresh() should throw an AttributeError N)rÚAttributeErrorrÚrefreshrrrrÚtest_refresh_raisess z(TestSwmrNotAvailable.test_refresh_raisesc Cs$| t¡|j ¡W5QRXdS)zi If the SWMR feature is not available the Dataset.flush() should throw an AttributeError N)rr%rÚflushrrrrÚtest_flush_raises s z&TestSwmrNotAvailable.test_flush_raisescCs|jjr t‚dS)z. The SWMR getter should just be False N)r Ú swmr_modeÚAssertionErrorrrrrÚtest_swmr_mode_false'sz)TestSwmrNotAvailable.test_swmr_mode_falsec Cs.| t¡d|j_W5QRX|jjr*t‚dS)zl If the SWMR feature is not available, setting swmr_mode = True should raise a RuntimeError TN)rÚ RuntimeErrorr r*r+rrrrÚtest_set_swmr_mode_raises,s z.TestSwmrNotAvailable.test_set_swmr_mode_raisesN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rr$r'r)r,r.rrrrrsrzSWMR requires HDF5 >= 1.9.178c@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)ÚTestDatasetSwmrReadú| Testing SWMR functions when reading a dataset. Skip this test if the HDF5 library does not have the SWMR features. cCsht |¡t d¡ d¡|_|jjddd|jd|_|jj }|j  ¡t j |ddd |_|jd|_dS) Nrr r r r r rTr) rrrrrr r rrrrr r!r"rrrr:s  zTestDatasetSwmrRead.setUpcCs| |jj¡dS)z/ Verify that the file is initially in SWMR modeN)Ú assertTruer r*rrrrÚtest_initial_swmr_mode_onDsz-TestDatasetSwmrRead.test_initial_swmr_mode_oncCs| |j|j¡dSr )ÚassertArrayEqualrr rrrrÚtest_read_dataHsz"TestDatasetSwmrRead.test_read_datacCs|j ¡dSr )rr&rrrrÚ test_refreshKsz TestDatasetSwmrRead.test_refreshc Cs0| t¡d|j_W5QRX| |jj¡dS)zŒ Verify when reading a file cannot be forcibly switched to swmr mode. When reading with SWMR the file must be opened with swmr=True.TN)rÚ Exceptionr r*r5rrrrÚtest_force_swmr_mode_on_raisesNs z2TestDatasetSwmrRead.test_force_swmr_mode_on_raisesc Cs0| t¡d|j_W5QRX| |jj¡dS)úž Switching SWMR write mode off is only possible by closing the file. Attempts to forcibly switch off the SWMR mode should raise a ValueError. FN)rrr r*r5rrrrÚtest_force_swmr_mode_off_raisesUs z3TestDatasetSwmrRead.test_force_swmr_mode_off_raisesN) r/r0r1r2rr6r8r9r;r=rrrrr34s r3c@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)ÚTestDatasetSwmrWriter4cCsHtj| ¡ddd|_t d¡ d¡|_|jjdd|jj dd d |_ d S) za First setup a file with a small chunked and empty dataset. No data written yet. ÚwZlatest)Zlibverér r )r)ér )ÚshapeÚdtyperrN) r r!Úmktempr rrrr rrCrrrrrrcszTestDatasetSwmrWrite.setUpcCs| |jj¡dS)z3 Verify that the file is not initially in SWMR modeN)Z assertFalser r*rrrrÚtest_initial_swmr_mode_offpsz/TestDatasetSwmrWrite.test_initial_swmr_mode_offcCsd|j_| |jj¡dS)z Switch to SWMR mode and verify TN)r r*r5rrrrÚtest_switch_swmr_mode_ontsz-TestDatasetSwmrWrite.test_switch_swmr_mode_onc CsFd|j_| |jj¡| t¡d|j_W5QRX| |jj¡dS)r<TFN)r r*r5rrrrrrÚ test_switch_swmr_mode_off_raisesys  z5TestDatasetSwmrWrite.test_switch_swmr_mode_off_raisescCs^d|j_| |jj¡|j |jj¡|j|jdd…<|j ¡|j ¡|  |j|j¡dS)z) Extend and flush a SWMR dataset TN) r r*r5rÚresizer rBr(r&r7rrrrÚtest_extend_dsetƒs  z%TestDatasetSwmrWrite.test_extend_dsetcCsºd|j_| |jj¡|j d¡|j|jdd…<|j ¡|j ¡| |j|j¡|j d¡|j|jdd…<|j ¡|j ¡| |jdd…|j¡| |jdd…|j¡dS)NT)r@r)ér@rJ) r r*r5rrHr r(r&r7rrrrÚtest_extend_dset_multiple‘s      z.TestDatasetSwmrWrite.test_extend_dset_multipleN) r/r0r1r2rrErFrGrIrKrrrrr>]s  r>) Únumpyrr ÚcommonrrZ skipUnlessÚversionZhdf5_version_tuplerr3r>rrrrÚs,(