u-toolkit 0.1.18__tar.gz → 0.1.19__tar.gz

{u_toolkit-0.1.18 → u_toolkit-0.1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: u-toolkit
-Version: 0.1.18
+Version: 0.1.19
 Summary: Add your description here
 Requires-Python: >=3.11
 Requires-Dist: pydantic>=2.11.3

{u_toolkit-0.1.18 → u_toolkit-0.1.19}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "u-toolkit"
-version = "0.1.18"
+version = "0.1.19"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.11"

{u_toolkit-0.1.18 → u_toolkit-0.1.19}/src/u_toolkit/enum.py

@@ -15,29 +15,29 @@ __all__ = [
 
 class NameEnum(StrEnum):
     @staticmethod
-    def _generate_next_value_(name, *_, **__) -> str:
+    def _generate_next_value_(name: str, *_, **__) -> str:
         return name
 
 
 class PascalEnum(StrEnum):
     @staticmethod
-    def _generate_next_value_(name, *_, **__) -> str:
+    def _generate_next_value_(name: str, *_, **__) -> str:
         return to_pascal(name)
 
 
 class CamelEnum(StrEnum):
     @staticmethod
-    def _generate_next_value_(name, *_, **__) -> str:
+    def _generate_next_value_(name: str, *_, **__) -> str:
         return to_camel(name)
 
 
 class SnakeEnum(StrEnum):
     @staticmethod
-    def _generate_next_value_(name, *_, **__) -> str:
+    def _generate_next_value_(name: str, *_, **__) -> str:
         return to_snake(name)
 
 
 class TitleEnum(StrEnum):
     @staticmethod
-    def _generate_next_value_(name, *_, **__) -> str:
-        return name.replace("_", " ").title()
+    def _generate_next_value_(name: str, *_, **__) -> str:
+        return name.replace("_", " ").strip().title()

{u_toolkit-0.1.18 → u_toolkit-0.1.19}/src/u_toolkit/function.py

@@ -1,7 +1,7 @@
 from collections.abc import Callable
 
 
-def get_name(fn: Callable, /):
+def get_name(fn: Callable, /) -> str:
     return fn.__name__
 
 

{u_toolkit-0.1.18 → u_toolkit-0.1.19}/src/u_toolkit/helpers.py

@@ -2,4 +2,6 @@ from typing import Annotated, get_origin
 
 
 def is_annotated(target):
+    """判断值是否是 `typing.Annotated`"""
+
     return get_origin(target) is Annotated

u_toolkit-0.1.19/tests/test_enum.py

@@ -0,0 +1,130 @@
+from u_toolkit import enum
+
+
+def test_name_enum():
+    class _Enum(enum.NameEnum):
+        A = enum.auto()
+        Ab = enum.auto()
+        AbC = enum.auto()
+        ABCD = enum.auto()
+        ABCD_E = enum.auto()
+        aa = enum.auto()
+        a_a = enum.auto()
+        a2 = enum.auto()
+        Bb = enum.auto()
+        c_ = enum.auto()
+
+    assert _Enum.A == "A"
+    assert _Enum.Ab == "Ab"
+    assert _Enum.AbC == "AbC"
+    assert _Enum.ABCD == "ABCD"
+    assert _Enum.ABCD_E == "ABCD_E"
+    assert _Enum.aa == "aa"
+    assert _Enum.a_a == "a_a"
+    assert _Enum.a2 == "a2"
+    assert _Enum.Bb == "Bb"
+    assert _Enum.c_ == "c_"
+
+
+def test_pascal_enum():
+    class _Enum(enum.PascalEnum):
+        A = enum.auto()
+        Ab = enum.auto()
+        AbC = enum.auto()
+        ABCD = enum.auto()
+        ABCD_E = enum.auto()
+        aa = enum.auto()
+        a_a = enum.auto()
+        a2 = enum.auto()
+        Bb = enum.auto()
+        c_ = enum.auto()
+
+    assert _Enum.A == "A"
+    assert _Enum.Ab == "Ab"
+    assert _Enum.AbC == "Abc"
+    assert _Enum.ABCD == "ABCD"
+    assert _Enum.ABCD_E == "ABCD_E"
+    assert _Enum.aa == "Aa"
+    assert _Enum.a_a == "AA"
+    assert _Enum.a2 == "A2"
+    assert _Enum.Bb == "Bb"
+    assert _Enum.c_ == "C_"
+
+
+def test_camel_enum():
+    class _Enum(enum.CamelEnum):
+        A = enum.auto()
+        Ab = enum.auto()
+        AbC = enum.auto()
+        ABCD = enum.auto()
+        ABCD_E = enum.auto()
+        aa = enum.auto()
+        a_a = enum.auto()
+        a2 = enum.auto()
+        Bb = enum.auto()
+        c_ = enum.auto()
+
+    assert _Enum.A == "A"
+    assert _Enum.Ab == "ab"
+    assert _Enum.AbC == "abc"
+    assert _Enum.ABCD == "ABCD"
+    assert _Enum.ABCD_E == "ABCD_E"
+    assert _Enum.aa == "aa"
+    assert _Enum.a_a == "aA"
+    assert _Enum.a2 == "a2"
+    assert _Enum.Bb == "bb"
+    assert _Enum.c_ == "c_"
+
+
+def test_snake_enum():
+    class _Enum(enum.SnakeEnum):
+        A = enum.auto()
+        Ab = enum.auto()
+        AbC = enum.auto()
+        ABCD = enum.auto()
+        ABCD_E = enum.auto()
+        aa = enum.auto()
+        a_a = enum.auto()
+        a2 = enum.auto()
+        aa3 = enum.auto()
+        Bb = enum.auto()
+        c_ = enum.auto()
+
+    assert _Enum.A == "A"
+    assert _Enum.Ab == "ab"
+    assert _Enum.AbC == "ab_c"
+    assert _Enum.ABCD == "ABCD"
+    assert _Enum.ABCD_E == "ABCD_E"
+    assert _Enum.aa == "aa"
+    assert _Enum.a_a == "a_a"
+    assert _Enum.a2 == "a_2"
+    assert _Enum.aa3 == "aa_3"
+    assert _Enum.Bb == "bb"
+    assert _Enum.c_ == "c_"
+
+
+def test_title_enum():
+    class _Enum(enum.TitleEnum):
+        A = enum.auto()
+        Ab = enum.auto()
+        AbC = enum.auto()
+        ABCD = enum.auto()
+        ABCD_E = enum.auto()
+        aa = enum.auto()
+        a_a = enum.auto()
+        a2 = enum.auto()
+        aa3 = enum.auto()
+        Bb = enum.auto()
+        c_ = enum.auto()
+
+    assert _Enum.A == "A"
+    assert _Enum.Ab == "Ab"
+    assert _Enum.AbC == "Abc"
+    assert _Enum.ABCD == "Abcd"
+    assert _Enum.ABCD_E == "Abcd E"
+    assert _Enum.aa == "Aa"
+    assert _Enum.a_a == "A A"
+    assert _Enum.a2 == "A2"
+    assert _Enum.aa3 == "Aa3"
+    assert _Enum.Bb == "Bb"
+    assert _Enum.c_ == "C"

u_toolkit-0.1.19/tests/test_function.py

@@ -0,0 +1,26 @@
+from u_toolkit import function
+
+
+def test_get_name():
+    def fn(): ...
+
+    assert function.get_name(fn) == "fn"
+    assert function.get_name(lambda: 1) == "<lambda>"
+
+
+def test_add_document():
+    def fn(): ...
+
+    assert fn.__doc__ is None
+
+    function.add_document(fn, "lalala")
+
+    assert fn.__doc__ == "lalala"
+
+    def fn2():
+        """2333"""
+
+    assert fn2.__doc__ == "2333"
+
+    function.add_document(fn2, "lalala")
+    assert fn2.__doc__ == "2333\n\nlalala"

u_toolkit-0.1.19/tests/test_helpers.py

@@ -0,0 +1,13 @@
+from typing import Annotated
+
+from u_toolkit import helpers
+
+
+def test_is_annotated():
+    assert helpers.is_annotated(type) is False
+    assert helpers.is_annotated(int) is False
+    assert helpers.is_annotated(object()) is False
+    assert helpers.is_annotated(Annotated) is False
+    assert helpers.is_annotated(Annotated[int, 1]) is True
+    assert helpers.is_annotated(Annotated[int, type]) is True
+    assert helpers.is_annotated(Annotated[int, object()]) is True