Skip to content

Commit b188923

Browse files
committed
test: Fix Name of the test model
1 parent 22f741d commit b188923

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

tests/test_serialize_columns.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,13 @@
44
def test_serialize_columns_class_level(get_instance):
55
"""Test class-level serialize_columns attribute"""
66

7-
class CustomFlatModel(FlatModel):
7+
class ClassLevelCustomFlatModel(FlatModel):
88
serialize_columns = {
99
"id": lambda v: str(v),
1010
"string": lambda v: v.upper() if v else None,
1111
}
1212

13-
i = get_instance(CustomFlatModel)
13+
i = get_instance(ClassLevelCustomFlatModel)
1414
data = i.to_dict()
1515

1616
# Custom serializer should be applied
@@ -118,12 +118,12 @@ def test_serialize_columns_backward_compatibility(get_instance):
118118
def test_serialize_columns_parameter_overrides_class(get_instance):
119119
"""Test that parameter-level serialize_columns overrides class-level"""
120120

121-
class CustomFlatModel(FlatModel):
121+
class ParameterOverrideCustomFlatModel(FlatModel):
122122
serialize_columns = {
123123
"id": lambda v: f"CLASS_{v}",
124124
}
125125

126-
i = get_instance(CustomFlatModel)
126+
i = get_instance(ParameterOverrideCustomFlatModel)
127127

128128
# Parameter should override class-level
129129
data = i.to_dict(

0 commit comments

Comments
 (0)