@@ -85,12 +85,8 @@ def test_clean_fn(self):
8585 self .assertEqual (cleaned ["fn" ], None )
8686
8787 def test_remove_slash (self ):
88- self .assertEqual (
89- "http://posthog.io" , utils .remove_trailing_slash ("http://posthog.io/" )
90- )
91- self .assertEqual (
92- "http://posthog.io" , utils .remove_trailing_slash ("http://posthog.io" )
93- )
88+ self .assertEqual ("http://posthog.io" , utils .remove_trailing_slash ("http://posthog.io/" ))
89+ self .assertEqual ("http://posthog.io" , utils .remove_trailing_slash ("http://posthog.io" ))
9490
9591 def test_clean_pydantic (self ):
9692 class ModelV2 (BaseModel ):
@@ -105,9 +101,7 @@ class ModelV1(BaseModelV1):
105101 class NestedModel (BaseModel ):
106102 foo : ModelV2
107103
108- self .assertEqual (
109- utils .clean (ModelV2 (foo = "1" , bar = 2 )), {"foo" : "1" , "bar" : 2 , "baz" : None }
110- )
104+ self .assertEqual (utils .clean (ModelV2 (foo = "1" , bar = 2 )), {"foo" : "1" , "bar" : 2 , "baz" : None })
111105 self .assertEqual (utils .clean (ModelV1 (foo = 1 , bar = "2" )), {"foo" : 1 , "bar" : "2" })
112106 self .assertEqual (
113107 utils .clean (NestedModel (foo = ModelV2 (foo = "1" , bar = 2 , baz = "3" ))),
@@ -128,6 +122,7 @@ class InnerDataClass:
128122 inner_bar : int
129123 inner_uuid : UUID
130124 inner_date : datetime
125+ inner_optional : str | None = None
131126
132127 @dataclass
133128 class TestDataClass :
@@ -156,6 +151,7 @@ class TestDataClass:
156151 "inner_bar" : 4 ,
157152 "inner_uuid" : "12345678-1234-5678-1234-567812345678" ,
158153 "inner_date" : datetime (2025 , 1 , 1 ),
154+ "inner_optional" : None ,
159155 },
160156 },
161157 )
0 commit comments