bec-lib 4.4.0__tar.gz → 4.4.1__tar.gz

{bec_lib-4.4.0 → bec_lib-4.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: bec_lib
-Version: 4.4.0
+Version: 4.4.1
 Summary: BEC library
 Project-URL: Bug Tracker, https://github.com/bec-project/bec/issues
 Project-URL: Homepage, https://github.com/bec-project/bec

{bec_lib-4.4.0 → bec_lib-4.4.1}/bec_lib/codecs.py

@@ -10,7 +10,7 @@ from pydantic import BaseModel
 from bec_lib import messages as messages_module
 from bec_lib import numpy_encoder
 from bec_lib.device import DeviceBase
-from bec_lib.endpoints import EndpointInfo
+from bec_lib.endpoints import EndpointInfo, MessageOp
 from bec_lib.messages import BECMessage, BECStatus
 
 
@@ -126,7 +126,7 @@ class EndpointInfoEncoder(BECCodec):
         return EndpointInfo(
             endpoint=data["endpoint"],
             message_type=getattr(messages_module, data["message_type"]),
-            message_op=data["message_op"],
+            message_op=MessageOp(data["message_op"]),
         )
 
 

{bec_lib-4.4.0 → bec_lib-4.4.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "bec_lib"
-version = "4.4.0"
+version = "4.4.1"
 description = "BEC library"
 requires-python = ">=3.11"
 classifiers = [

{bec_lib-4.4.0 → bec_lib-4.4.1}/tests/test_serializer.py

@@ -12,7 +12,7 @@ from bec_lib import messages
 from bec_lib.codecs import BECCodec
 from bec_lib.device import DeviceBase
 from bec_lib.devicemanager import DeviceManagerBase
-from bec_lib.endpoints import MessageEndpoints
+from bec_lib.endpoints import MessageEndpoints, MessageOp
 from bec_lib.serialization import MsgpackSerialization, json_ext, msgpack
 
 
@@ -214,6 +214,13 @@ def test_enum_serializer(serializer):
     assert serializer.loads(serializer.dumps(CustomEnum.VALUE1)) == "value1"
 
 
+def test_endpoint_info_decodes_to_enum(serializer):
+    endpoint = MessageEndpoints.scan_interlock_enabled()
+    restored = serializer.loads(serializer.dumps(endpoint))
+    assert restored == endpoint
+    assert restored.message_op is MessageOp.STREAM
+
+
 def test_serializer_encoding_on_failure():
     """
     Test that an exception raised during serialization is caught and the original object is returned.