a v=icÁã@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úm/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/h5py/tests/test_dataset_swmr.pyrs zTestSwmrNotAvailable.setUpcCsR|jj}|j ¡| t¡"tj|ddd|_Wdƒn1sD0YdS)NÚrT©Zswmr)r ÚfilenameÚcloseÚ assertRaisesÚ ValueErrorÚh5pyÚFile©rÚfnamerrrÚtest_open_swmr_raisess  z*TestSwmrNotAvailable.test_open_swmr_raisescCs8| t¡|j ¡Wdƒn1s*0YdS)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_raisescCs8| t¡|j ¡Wdƒn1s*0YdS)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 J‚dS)z. The SWMR getter should just be False N)r Ú swmr_moderrrrÚtest_swmr_mode_false'sz)TestSwmrNotAvailable.test_swmr_mode_falsecCsB| t¡d|j_Wdƒn1s(0Y|jjr>J‚dS)zl If the SWMR feature is not available, setting swmr_mode = True should raise a RuntimeError TN)rÚ RuntimeErrorr 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_refreshcCsD| t¡d|j_Wdƒn1s(0Y| |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*r4rrrrÚtest_force_swmr_mode_on_raisesNs &z2TestDatasetSwmrRead.test_force_swmr_mode_on_raisescCsD| t¡d|j_Wdƒn1s(0Y| |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*r4rrrrÚtest_force_swmr_mode_off_raisesUs &z3TestDatasetSwmrRead.test_force_swmr_mode_off_raisesN) r.r/r0r1rr5r7r8r:r<rrrrr24s r2c@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)ÚTestDatasetSwmrWriter3cCsHtj| ¡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 rrBrrrrrrcszTestDatasetSwmrWrite.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*r4rrrrÚtest_switch_swmr_mode_ontsz-TestDatasetSwmrWrite.test_switch_swmr_mode_oncCsZd|j_| |jj¡| t¡d|j_Wdƒn1s>0Y| |jj¡dS)r;TFN)r r*r4rrrrrrÚ 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*r4rÚresizer rAr(r&r6rrrrÚ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?rI) r r*r4rrGr r(r&r6rrrrÚtest_extend_dset_multiple‘s      z.TestDatasetSwmrWrite.test_extend_dset_multipleN) r.r/r0r1rrDrErFrHrJrrrrr=]s  r=) Únumpyrr ÚcommonrrZ skipUnlessÚversionZhdf5_version_tuplerr2r=rrrrÚs,(