Skip to content

Commit a834d1c

Browse files
committed
test(checkpoint): fix dumps_helper usage in serializer tests
Update tests to unpack (type_str, bytes) tuple from dumps_helper instead of expecting just bytes.
1 parent d21047d commit a834d1c

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

tests/test_jsonplus_serializer_default_handler.py

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,12 @@ def test_serializer_uses_default_handler_for_messages():
4646
human_msg = HumanMessage(content="What is the weather?", id="msg-1")
4747

4848
# This should NOT raise TypeError
49-
serialized_bytes = dumps_helper(serializer, human_msg)
49+
typed_data = dumps_helper(serializer, human_msg)
50+
type_str, serialized_bytes = typed_data
5051
assert isinstance(serialized_bytes, bytes)
5152

5253
# Deserialize and verify
53-
deserialized = loads_helper(serializer, serialized_bytes)
54+
deserialized = loads_helper(serializer, typed_data)
5455
assert isinstance(deserialized, HumanMessage)
5556
assert deserialized.content == "What is the weather?"
5657
assert deserialized.id == "msg-1"
@@ -72,11 +73,12 @@ def test_serializer_handles_all_message_types():
7273

7374
for msg in messages:
7475
# Serialize
75-
serialized = dumps_helper(serializer, msg)
76+
typed_data = dumps_helper(serializer, msg)
77+
type_str, serialized = typed_data
7678
assert isinstance(serialized, bytes)
7779

7880
# Deserialize
79-
deserialized = loads_helper(serializer, serialized)
81+
deserialized = loads_helper(serializer, typed_data)
8082

8183
# Verify type is preserved
8284
assert type(deserialized) == type(msg)
@@ -98,11 +100,12 @@ def test_serializer_handles_message_lists():
98100
]
99101

100102
# Serialize the list
101-
serialized = dumps_helper(serializer, messages)
103+
typed_data = dumps_helper(serializer, messages)
104+
type_str, serialized = typed_data
102105
assert isinstance(serialized, bytes)
103106

104107
# Deserialize
105-
deserialized = loads_helper(serializer, serialized)
108+
deserialized = loads_helper(serializer, typed_data)
106109

107110
# Verify structure
108111
assert isinstance(deserialized, list)
@@ -131,11 +134,12 @@ def test_serializer_handles_nested_structures_with_messages():
131134
}
132135

133136
# Serialize
134-
serialized = dumps_helper(serializer, state)
137+
typed_data = dumps_helper(serializer, state)
138+
type_str, serialized = typed_data
135139
assert isinstance(serialized, bytes)
136140

137141
# Deserialize
138-
deserialized = loads_helper(serializer, serialized)
142+
deserialized = loads_helper(serializer, typed_data)
139143

140144
# Verify structure
141145
assert "messages" in deserialized

0 commit comments

Comments
 (0)