dycw-utilities 0.165.0__tar.gz → 0.165.1__tar.gz

{dycw_utilities-0.165.0 → dycw_utilities-0.165.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.165.0
+Version: 0.165.1
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

{dycw_utilities-0.165.0 → dycw_utilities-0.165.1}/pyproject.toml

@@ -233,7 +233,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.165.0"
+version = "0.165.1"
 
 [project.entry-points.pytest11]
 pytest-randomly = "utilities.pytest_plugins.pytest_randomly"
@@ -266,7 +266,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.165.0"
+current_version = "0.165.1"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.165.0 → dycw_utilities-0.165.1}/src/tests/test_contextlib.py

@@ -4,6 +4,7 @@ import asyncio
 import time
 from asyncio import run
 from dataclasses import dataclass
+from inspect import signature
 from multiprocessing import Process
 from pathlib import Path
 from typing import TYPE_CHECKING, override
@@ -103,6 +104,15 @@ class TestEnhancedContextManager:
             assert not cleared
         assert cleared
 
+    def test_sync_signature(self) -> None:
+        @enhanced_context_manager
+        def yield_marker(x: int, y: int, /) -> Iterator[int]:
+            yield x + y
+
+        sig = set(signature(yield_marker).parameters)
+        expected = {"x", "y"}
+        assert sig == expected
+
     @given(
         sigabrt=booleans(),
         sigfpe=booleans(),
@@ -142,6 +152,15 @@ class TestEnhancedContextManager:
             assert not cleared
         assert cleared
 
+    def test_async_signature(self) -> None:
+        @enhanced_async_context_manager
+        async def yield_marker(x: int, y: int, /) -> AsyncIterator[int]:
+            yield x + y
+
+        sig = set(signature(yield_marker).parameters)
+        expected = {"x", "y"}
+        assert sig == expected
+
     @mark.parametrize(
         "target",
         [

{dycw_utilities-0.165.0 → dycw_utilities-0.165.1}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.165.0"
+__version__ = "0.165.1"

{dycw_utilities-0.165.0 → dycw_utilities-0.165.1}/src/utilities/contextlib.py

@@ -8,7 +8,7 @@ from contextlib import (
     asynccontextmanager,
     contextmanager,
 )
-from functools import partial
+from functools import partial, wraps
 from signal import SIGABRT, SIGFPE, SIGILL, SIGINT, SIGSEGV, SIGTERM, getsignal, signal
 from typing import TYPE_CHECKING, Any, assert_never, cast, overload
 
@@ -77,6 +77,7 @@ def enhanced_context_manager[**P, T_co](
     make_gcm = contextmanager(func)
 
     @contextmanager
+    @wraps(func)
     def wrapped(*args: P.args, **kwargs: P.kwargs) -> Iterator[T_co]:
         gcm = make_gcm(*args, **kwargs)
         sigabrt0 = _swap_handler(SIGABRT, gcm) if sigabrt else None
@@ -159,6 +160,7 @@ def enhanced_async_context_manager[**P, T_co](
     make_agcm = asynccontextmanager(func)
 
     @asynccontextmanager
+    @wraps(func)
     async def wrapped(*args: P.args, **kwargs: P.kwargs) -> AsyncIterator[T_co]:
         agcm = make_agcm(*args, **kwargs)
         sigabrt0 = _swap_handler(SIGABRT, agcm) if sigabrt else None