File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 44def 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):
118118def 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 (
You can’t perform that action at this time.
0 commit comments