diff --git a/tests/test_serializer.py b/tests/test_serializer.py index 3e33bcb3..f88ae75e 100644 --- a/tests/test_serializer.py +++ b/tests/test_serializer.py @@ -1,4 +1,9 @@ -from datetime import UTC, datetime +try: + from datetime import UTC, datetime +except ImportError: + from datetime import datetime, timezone + + UTC = timezone.utc import logistro import numpy as np @@ -25,10 +30,12 @@ async def test_de_serialize(): message = wire.serialize(data) assert message == expected_message obj = wire.deserialize(message) - for o, t in zip(obj, converted_type, strict=False): + assert len(obj) == len(converted_type) + for o, t in zip(obj, converted_type): assert isinstance(o, t) message_np = wire.serialize(np.array(data)) assert message_np == expected_message obj_np = wire.deserialize(message_np) - for o, t in zip(obj_np, converted_type, strict=False): + assert len(obj_np) == len(converted_type) + for o, t in zip(obj_np, converted_type): assert isinstance(o, t)