liniting

Author: commonism

pydantic_extra_types/epoch.py

@@ -21,15 +21,15 @@ def __get_pydantic_json_schema__(
     def __get_pydantic_core_schema__(
         cls, source: type[Any], handler: Callable[[Any], CoreSchema]
     ) -> core_schema.CoreSchema:
-        def f(value, serializer):
+        def f(value: Any, serializer: Callable[[datetime.datetime], float]) -> float:
             return serializer(value.timestamp())
+
         return core_schema.with_info_after_validator_function(
-                cls._validate, core_schema.float_schema(),
-                serialization=core_schema.wrap_serializer_function_ser_schema(f, return_schema=core_schema.float_schema())
+            cls._validate,
+            core_schema.float_schema(),
+            serialization=core_schema.wrap_serializer_function_ser_schema(f, return_schema=core_schema.float_schema()),
         )
 
     @classmethod
-    def _validate(cls, __input_value: Any, _: Any) -> "Epoch":
+    def _validate(cls, __input_value: Any, _: Any) -> datetime.datetime:
         return EPOCH + datetime.timedelta(seconds=__input_value)
-
-

tests/test_epoch.py

@@ -4,7 +4,8 @@
 
 from pydantic_extra_types.epoch import Epoch
 
-@pytest.mark.parametrize('type_',[(int,),(float,)], ids=["integer", "float"])
+
+@pytest.mark.parametrize('type_', [(int,), (float,)], ids=['integer', 'float'])
 def test_type(type_):
     from pydantic import BaseModel
 
@@ -13,25 +14,25 @@ class A(BaseModel):
 
     now = datetime.datetime.now(tz=datetime.timezone.utc)
     ts = type_(now.timestamp())
-    a = A.model_validate({"epoch":ts})
+    a = A.model_validate({'epoch': ts})
     v = a.model_dump()
-    assert v["epoch"] == ts
+    assert v['epoch'] == ts
 
     b = A.model_construct(epoch=now)
 
     v = b.model_dump()
-    assert v["epoch"] == ts
+    assert v['epoch'] == ts
 
     c = A.model_validate(dict(epoch=ts))
     v = c.model_dump()
-    assert v["epoch"] == ts
+    assert v['epoch'] == ts
 
 
 def test_schema():
     from pydantic import BaseModel
+
     class A(BaseModel):
         dt: Epoch
 
     v = A.model_json_schema()
-    assert (dt:=v["properties"]["dt"])["type"] == "number" and dt["format"] == "date-time"
-
+    assert (dt := v['properties']['dt'])['type'] == 'number' and dt['format'] == 'date-time'