Coverage for install/scipp/serialization.py: 86%

1# SPDX-License-Identifier: BSD-3-Clause 

2# Copyright (c) 2023 Scipp contributors (https://github.com/scipp) 

3# @author Simon Heybrock 

4 

5from typing import Any 

6 

7from .core import DataArray, DataGroup, Dataset, Variable 

8 

9 

10def serialize( 

11 var: Variable | DataArray | Dataset | DataGroup[Any], 

12) -> tuple[dict[str, Any], list[bytes]]: 

13 """Serialize Scipp object.""" 

14 from io import BytesIO 

15 

16 from .io.hdf5 import save_hdf5 

17 

18 header: dict[str, Any] = {} 

19 buf = BytesIO() 

20 save_hdf5(var, buf) 

21 frames = [buf.getvalue()] 

22 return header, frames 

23 

24 

25def deserialize( 

26 header: dict[str, Any], frames: list[bytes] 

27) -> Variable | DataArray | Dataset | DataGroup[Any]: 

28 """Deserialize Scipp object.""" 

29 from io import BytesIO 

30 

31 from .io.hdf5 import load_hdf5 

32 

33 return load_hdf5(BytesIO(frames[0])) 

34 

35 

36try: 

37 from distributed.protocol import register_serialization 

38 

39 register_serialization(Variable, serialize, deserialize) 

40 register_serialization(DataArray, serialize, deserialize) 

41 register_serialization(Dataset, serialize, deserialize) 

42except ImportError: 

43 pass 

44__all__ = ['serialize', 'deserialize']