Skip to content

Commit 07a6d50

Browse files
committed
Change expected output types from list to tuples as per immutability rule for QueryResult
1 parent 9c389a4 commit 07a6d50

1 file changed

Lines changed: 12 additions & 12 deletions

File tree

tests/unit/core/test_types.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ def test_query_result_with_dict_response_output_only(self):
236236
decoded = msgspec.msgpack.decode(encoded, type=QueryResult)
237237

238238
assert isinstance(decoded.response_output, dict)
239-
assert decoded.response_output["output"] == ["First chunk", "rest of output"]
239+
assert decoded.response_output["output"] == ("First chunk", "rest of output")
240240
assert "reasoning" not in decoded.response_output
241241

242242
def test_query_result_with_dict_response_output_and_reasoning(self):
@@ -254,10 +254,10 @@ def test_query_result_with_dict_response_output_and_reasoning(self):
254254

255255
assert isinstance(decoded.response_output, dict)
256256
assert decoded.response_output["output"] == "Final output text"
257-
assert decoded.response_output["reasoning"] == [
257+
assert decoded.response_output["reasoning"] == (
258258
"First reasoning chunk",
259259
"rest of reasoning",
260-
]
260+
)
261261

262262
def test_query_result_with_dict_response_empty_output(self):
263263
"""Test QueryResult with dict response output (empty output)."""
@@ -267,7 +267,7 @@ def test_query_result_with_dict_response_empty_output(self):
267267
decoded = msgspec.msgpack.decode(encoded, type=QueryResult)
268268

269269
assert isinstance(decoded.response_output, dict)
270-
assert decoded.response_output["output"] == []
270+
assert decoded.response_output["output"] == ()
271271

272272
def test_query_result_with_metadata(self):
273273
"""Test QueryResult with comprehensive metadata."""
@@ -573,10 +573,10 @@ def test_query_result_reasoning_chunks_pattern(self):
573573
decoded = msgspec.msgpack.decode(encoded, type=QueryResult)
574574

575575
assert decoded.response_output["output"] == "The answer is 42"
576-
assert decoded.response_output["reasoning"] == [
576+
assert decoded.response_output["reasoning"] == (
577577
"Let me think...",
578578
" step by step to solve this",
579-
]
579+
)
580580
assert len(decoded.response_output["reasoning"]) == 2
581581

582582
def test_query_result_output_only_pattern(self):
@@ -602,7 +602,7 @@ def test_query_result_output_only_pattern(self):
602602
encoded = msgspec.msgpack.encode(result)
603603
decoded = msgspec.msgpack.decode(encoded, type=QueryResult)
604604

605-
assert decoded.response_output["output"] == ["Hello", " world!"]
605+
assert decoded.response_output["output"] == ("Hello", " world!")
606606
assert len(decoded.response_output["output"]) == 2
607607
assert "reasoning" not in decoded.response_output
608608

@@ -620,7 +620,7 @@ def test_query_result_no_chunks_pattern(self):
620620
encoded = msgspec.msgpack.encode(result)
621621
decoded = msgspec.msgpack.decode(encoded, type=QueryResult)
622622

623-
assert decoded.response_output["output"] == []
623+
assert decoded.response_output["output"] == ()
624624
assert "reasoning" not in decoded.response_output
625625

626626
def test_query_result_single_reasoning_chunk(self):
@@ -643,7 +643,7 @@ def test_query_result_single_reasoning_chunk(self):
643643
encoded = msgspec.msgpack.encode(result)
644644
decoded = msgspec.msgpack.decode(encoded, type=QueryResult)
645645

646-
assert decoded.response_output["reasoning"] == ["Quick thought"]
646+
assert decoded.response_output["reasoning"] == ("Quick thought",)
647647
assert len(decoded.response_output["reasoning"]) == 1
648648

649649
def test_query_result_single_output_chunk(self):
@@ -662,7 +662,7 @@ def test_query_result_single_output_chunk(self):
662662
encoded = msgspec.msgpack.encode(result)
663663
decoded = msgspec.msgpack.decode(encoded, type=QueryResult)
664664

665-
assert decoded.response_output["output"] == ["SingleResponse"]
665+
assert decoded.response_output["output"] == ("SingleResponse",)
666666
assert len(decoded.response_output["output"]) == 1
667667

668668

@@ -748,8 +748,8 @@ def test_query_result_with_nested_metadata(self):
748748
assert decoded.metadata["model_info"]["parameters"]["temperature"] == 0.7
749749
assert decoded.metadata["usage"]["breakdown"] == [5, 5, 10, 10]
750750
assert "production" in decoded.metadata["tags"]
751-
assert decoded.response_output["output"] == ["First chunk", "remaining output"]
752-
assert decoded.response_output["reasoning"] == ["Reasoning process"]
751+
assert decoded.response_output["output"] == ("First chunk", "remaining output")
752+
assert decoded.response_output["reasoning"] == ("Reasoning process",)
753753

754754
def test_query_with_none_values_in_data(self):
755755
"""Test Query with None values in data dict."""

0 commit comments

Comments
 (0)