def test_hdf_is_open(hdf_dataset): """ Test hdf_is_open function """ _, fid = hdf_dataset assert hdf_is_open(fid) fid.close() assert not hdf_is_open(fid)
def hdf_dataset(self): """ Setups and tears down a sample HDF5 file """ filename = 'temp_test_ui.h5' fid = h5py.File(filename, 'w') data_m, data_n, data_p = [20, 22, 24] data = np.random.randn(data_m, data_n, data_p) fid.create_dataset('base', data=data) grp1 = fid.create_group('Group1') grp3 = fid.create_group('Group2/Group3') grp6 = fid.create_group('Group4/Group5/Group6') grp1.create_dataset('ingroup1_1', data=data) grp1.create_dataset('ingroup1_2', data=data) fid.create_dataset('Group2/ingroup2', data=data) grp3.create_dataset('ingroup3', data=data) grp6.create_dataset('ingroup6', data=data) fid['base'].attrs['Attribute_str'] = 'Test' fid['base'].attrs['Attribute_bytes'] = b'Test' fid['base'].attrs['Attribute_np_bytes'] = np.bytes_('Test') # pylint: disable=no-member fid['base'].attrs.create('Attribute_int', 1) fid['base'].attrs.create('Attribute_float', 1.1) fid['base'].attrs.create('Attribute_np_1d', np.array([1, 2, 3])) fid['base'].attrs.create('Attribute_np_2d', np.array([[1, 2, 3], [4, 5, 6]])) app = QApplication(sys.argv) # pylint: disable=C0103, W0612 yield filename # Tear-down if hdf_is_open(fid): fid.close() os.remove(filename)
def hdf_dataset(): """ Setups and tears down a sample HDF5 file """ filename = 'temp_test_utils.h5' fid = h5py.File(filename, 'w') data_m, data_n, data_p = [20, 22, 24] data = np.random.randn(data_m, data_n, data_p) fid.create_dataset('base', data=data) yield filename, fid # Tear-down if hdf_is_open(fid): fid.close() os.remove(filename)
def hdf_dataset(): """ Setups and tears down a sample HDF5 file """ filename = 'temp_test.h5' fid = h5py.File(filename, 'w') data_m, data_n, data_p = [20, 22, 24] data = np.random.randn(data_m, data_n, data_p) fid.create_dataset('base', data=data) grp1 = fid.create_group('Group1') grp3 = fid.create_group('Group2/Group3') grp6 = fid.create_group('Group4/Group5/Group6') grp1.create_dataset('ingroup1_1', data=data) grp1.create_dataset('ingroup1_2', data=data) fid.create_dataset('Group2/ingroup2', data=data) grp3.create_dataset('ingroup3', data=data) grp6.create_dataset('ingroup6', data=data) fid['base'].attrs['Attribute_str'] = 'Test' fid['base'].attrs['Attribute_bytes'] = b'Test' fid['base'].attrs['Attribute_np_bytes'] = np.bytes_('Test') # pylint: disable=no-member fid['base'].attrs['Attribute_np_bytes_inarray'] = np.array(b'Test') # pylint: disable=no-member fid['base'].attrs['Attribute_np_bytes_inarray2'] = np.array([b'Test']) # pylint: disable=no-member fid['base'].attrs.create('Attribute_int', 1) fid['base'].attrs.create('Attribute_float', 1.1) fid['base'].attrs.create('Attribute_np_1d', np.array([1, 2, 3])) fid['base'].attrs.create('Attribute_np_2d', np.array([[1, 2, 3], [4, 5, 6]])) fid['base'].attrs.create('Attribute_np_sgl_int', np.array([1])) fid['base'].attrs.create('Attribute_np_sgl_float', np.array([1.0])) fid['base'].attrs.create('Attribute_np_array_float', np.array([1.0, 2.0])) fid['base'].attrs.create('Attribute_np_sgl_complex', np.array([1.0 + 1j])) yield filename, fid # Tear-down if hdf_is_open(fid): fid.close() time.sleep(1) try: os.remove(filename) except: print('Could not delete {}'.format(filename))