Skip to content

Commit 9d9c1bc

Browse files
committed
lint
1 parent 18beca6 commit 9d9c1bc

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/utils/test_general.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -626,24 +626,25 @@ def test_strip_string(input, max_length, result):
626626

627627
def test_normalize_data_with_pydantic_class():
628628
"""Test that _normalize_data handles Pydantic model classes"""
629+
629630
class TestClass:
630631
name: str = None
631632

632633
def __init__(self, name: str):
633634
self.name = name
634-
635+
635636
def model_dump(self):
636637
return {"name": self.name}
637-
638+
638639
# Test with class (should NOT call model_dump())
639640
result = _normalize_data(TestClass)
640641
assert result == "<ClassType: TestClass>"
641-
642+
642643
# Test with instance (should call model_dump())
643644
instance = TestClass(name="test")
644645
result = _normalize_data(instance)
645646
assert result == {"name": "test"}
646-
647+
647648
# Test with dict containing class
648649
result = _normalize_data({"schema": TestClass, "count": 5})
649650
assert result == {"schema": "<ClassType: TestClass>", "count": 5}

0 commit comments

Comments
 (0)