dycw-utilities 0.156.0__py3-none-any.whl → 0.157.0__py3-none-any.whl

{dycw_utilities-0.156.0.dist-info → dycw_utilities-0.157.0.dist-info}/METADATA

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

{dycw_utilities-0.156.0.dist-info → dycw_utilities-0.157.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
-utilities/__init__.py,sha256=qnrT_UTHUY5w9TUoe2Wq56gzkzQRpi_K6c4hWJV6mLs,60
+utilities/__init__.py,sha256=T1D5frjma6r3zBtDh4rvvHMOorqu5R_6WaB8xxZsqNU,60
 utilities/altair.py,sha256=92E2lCdyHY4Zb-vCw6rEJIsWdKipuu-Tu2ab1ufUfAk,9079
-utilities/asyncio.py,sha256=QXkTtugXkqtYt7Do23zgYErqzdp6jwzPpV_SP9fJ1gI,16780
+utilities/asyncio.py,sha256=PUedzQ5deqlSECQ33sam9cRzI9TnygHz3FdOqWJWPTM,15288
 utilities/atomicwrites.py,sha256=tPo6r-Rypd9u99u66B9z86YBPpnLrlHtwox_8Z7T34Y,5790
 utilities/atools.py,sha256=6neeCcgXxK2dlsc0xp15Za7nSucbCgFtAJepGI_-WXU,2549
 utilities/cachetools.py,sha256=v1-9sXHLdOLiwmkq6NB0OUbxeKBuVVN6wmAWefWoaHI,2744
@@ -13,7 +13,7 @@ utilities/cvxpy.py,sha256=Rv1-fD-XYerosCavRF8Pohop2DBkU3AlFaGTfD8AEAA,13776
 utilities/dataclasses.py,sha256=MXrvIPSZHlpV4msRdVVDRZZo7MC3gX5C9jDUSoNOdpE,32478
 utilities/enum.py,sha256=5l6pwZD1cjSlVW4ss-zBPspWvrbrYrdtJWcg6f5_J5w,5781
 utilities/errors.py,sha256=mFlDGSM0LI1jZ1pbqwLAH3ttLZ2JVIxyZLojw8tGVZU,1479
-utilities/eventkit.py,sha256=ddoleSwW9zdc2tjX5Ge0pMKtYwV_JMxhHYOxnWX2AGM,12609
+utilities/eventkit.py,sha256=wjqrSlqHSe7chAcAROXt9h1IVt8grzX_8jp8PGoeRAs,12804
 utilities/fastapi.py,sha256=3wpd63Tw9paSyy7STpAD7GGe8fLkLaRC6TPCwIGm1BU,1361
 utilities/fpdf2.py,sha256=HgM8JSvoioDXrjC0UR3HVLjnMnnb_mML7nL2EmkTwGI,1854
 utilities/functions.py,sha256=RNVAoLeT_sl-gXaBv2VI_U_EB-d-nSVosYR4gTeeojE,28261
@@ -31,7 +31,7 @@ utilities/json.py,sha256=-WcGtSsCr9Y42wHZzAMnfvU6ihAfVftylFfRUORaDFo,2102
 utilities/jupyter.py,sha256=ft5JA7fBxXKzP-L9W8f2-wbF0QeYc_2uLQNFDVk4Z-M,2917
 utilities/libcst.py,sha256=TKgKN4bNmtBNEE-TUfhTyd1BrTncfsl_7tTuhpesGYY,5585
 utilities/lightweight_charts.py,sha256=YM3ojBvJxuCSUBu_KrhFBmaMCvRPvupKC3qkm-UVZq4,2751
-utilities/logging.py,sha256=ihbfQJgjc7t3Pds0oPvF_J1eigiqFKzxNOijzoee8U4,18064
+utilities/logging.py,sha256=NBxAafc9q0KwCqZRHbxI_OEnYADiEGOleLxDsapS7g4,18379
 utilities/math.py,sha256=7ve4RxX3g-FGGVnWV0K9bBeGnKUEjnTbH13VxdvFtGE,26847
 utilities/memory_profiler.py,sha256=XzN56jDCa5aqXS_DxEjb_K4L6aIWh_5zyKi6OhcIxw0,853
 utilities/modules.py,sha256=iuvLluJya-hvl1Q25-Jk3dLgx2Es3ck4SjJiEkAlVTs,3195
@@ -40,7 +40,7 @@ utilities/numpy.py,sha256=Xn23sA2ZbVNqwUYEgNJD3XBYH6IbCri_WkHSNhg3NkY,26122
 utilities/operator.py,sha256=nhxn5q6CFNzUm1wpTwWPCu9JGCqVHSlaJf0o1-efoII,3616
 utilities/optuna.py,sha256=C-fhWYiXHVPo1l8QctYkFJ4DyhbSrGorzP1dJb_qvd8,1933
 utilities/orjson.py,sha256=Ll0U172ITMqOJc3kjV90C0eI-EWzSXlMHSdUBaUSe80,41499
-utilities/os.py,sha256=mFvjydySvjtSXpk7tLStUJcndauAoujxUUmj_CO7LWY,3778
+utilities/os.py,sha256=Zwznb1Y0cHHIPG7t0UfEWeo0VeDzCnBcwI0mTV-xh2M,3877
 utilities/parse.py,sha256=JcJn5yXKhIWXBCwgBdPsyu7Hvcuw6kyEdqvaebCaI9k,17951
 utilities/pathlib.py,sha256=qGuU8XPmdgGpy8tOMUgelfXx3kxI8h9IaV3TI_06QGE,8428
 utilities/pickle.py,sha256=MBT2xZCsv0pH868IXLGKnlcqNx2IRVKYNpRcqiQQqxw,653
@@ -48,23 +48,23 @@ utilities/platform.py,sha256=pTn7gw6N4T6LdKrf0virwarof_mze9WtoQlrGMzhGVI,2798
 utilities/polars.py,sha256=JOZjSpj9jitDijX044mKc-N00C5N_On3TJYJKJRhdcE,78494
 utilities/polars_ols.py,sha256=Uc9V5kvlWZ5cU93lKZ-cfAKdVFFw81tqwLW9PxtUvMs,5618
 utilities/postgres.py,sha256=ynCTTaF-bVEOSW-KEAR-dlLh_hYjeVVjm__-4pEU8Zk,12269
-utilities/pottery.py,sha256=HJ96oLRarTP37Vhg0WTyB3yAu2hETeg6HgRmpDIqyUs,6581
+utilities/pottery.py,sha256=ggMN72Y7wx7Js8VN6eyNyodpm8TIYqZHGghkDPXIVWk,3949
 utilities/pqdm.py,sha256=idv2seRVP2f6NeSfpeEnT5A-tQezaHZKDyeu16g2-0E,3091
 utilities/psutil.py,sha256=KUlu4lrUw9Zg1V7ZGetpWpGb9DB8l_SSDWGbANFNCPU,2104
 utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 utilities/pyinstrument.py,sha256=NZCZz2nBo0BLJ9DTf7H_Q_KGxvsf2S2M3h0qYoYh2kw,804
-utilities/pytest.py,sha256=2HHfAWkzZeK2OAzL2F49EDKooMkfDoGqg8Ev4cHC_N8,7869
+utilities/pytest.py,sha256=M-Om6b3hpF9W_bEB7UFY2IzBCubSxzVQleGrgRXHtxY,7741
 utilities/pytest_regressions.py,sha256=8by5DWEL89Y469TI5AzX1pMy3NJWVtjEg2xQdOOdYuM,4169
 utilities/random.py,sha256=hZlH4gnAtoaofWswuJYjcygejrY8db4CzP-z_adO2Mo,4165
 utilities/re.py,sha256=S4h-DLL6ScMPqjboZ_uQ1BVTJajrqV06r_81D--_HCE,4573
-utilities/redis.py,sha256=2fdveFbqL2pEAeyiVuN_Je8nSM_IZHeahPduMHhFRzY,28381
+utilities/redis.py,sha256=pqzl5A08vaRS4Gfjxob3LWWH9c-vwlsKbvVMTjWMSh8,28364
 utilities/reprlib.py,sha256=ssYTcBW-TeRh3fhCJv57sopTZHF5FrPyyUg9yp5XBlo,3953
 utilities/scipy.py,sha256=wZJM7fEgBAkLSYYvSmsg5ac-QuwAI0BGqHVetw1_Hb0,947
 utilities/sentinel.py,sha256=A_p5jX2K0Yc5XBfoYHyBLqHsEWzE1ByOdDuzzA2pZnE,1434
 utilities/shelve.py,sha256=4OzjQI6kGuUbJciqf535rwnao-_IBv66gsT6tRGiUt0,759
 utilities/slack_sdk.py,sha256=ppFBvKgfg5IRWiIoKPtpTyzBtBF4XmwEvU3I5wLJikM,2140
 utilities/socket.py,sha256=K77vfREvzoVTrpYKo6MZakol0EYu2q1sWJnnZqL0So0,118
-utilities/sqlalchemy.py,sha256=IJKzrKUd_eBOkyK6CucDlxtHwo2vYH3t-rV2_5rAxq8,40554
+utilities/sqlalchemy.py,sha256=wtbIp6XDjKwfrvl-wfoY4FQXo_a9vSoHq5K_dYeBBeY,40541
 utilities/sqlalchemy_polars.py,sha256=5Q9HReETYg0qB6E6WQhFh4QAZlKE-IWlogj2BVif_-w,14246
 utilities/statsmodels.py,sha256=koyiBHvpMcSiBfh99wFUfSggLNx7cuAw3rwyfAhoKpQ,3410
 utilities/string.py,sha256=shmBK87zZwzGyixuNuXCiUbqzfeZ9xlrFwz6JTaRvDk,582
@@ -87,8 +87,8 @@ utilities/zoneinfo.py,sha256=FBMcUQ4662Aq8SsuCL1OAhDQiyANmVjtb-C30DRrWoE,1966
 utilities/pytest_plugins/__init__.py,sha256=U4S_2y3zgLZVfMenHRaJFBW8yqh2mUBuI291LGQVOJ8,35
 utilities/pytest_plugins/pytest_randomly.py,sha256=B1qYVlExGOxTywq2r1SMi5o7btHLk2PNdY_b1p98dkE,409
 utilities/pytest_plugins/pytest_regressions.py,sha256=9v8kAXDM2ycIXJBimoiF4EgrwbUvxTycFWJiGR_GHhM,1466
-dycw_utilities-0.156.0.dist-info/METADATA,sha256=J71C9d7mzvc5tHFVLEmcOD0NLrPbS5D8zkpnRwq5AEg,1643
-dycw_utilities-0.156.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.156.0.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
-dycw_utilities-0.156.0.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.156.0.dist-info/RECORD,,
+dycw_utilities-0.157.0.dist-info/METADATA,sha256=I1kXqyTgNiYkQ5eip4vqyC8d-JA3uZCxNWB7Dxpn5JM,1643
+dycw_utilities-0.157.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.157.0.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
+dycw_utilities-0.157.0.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.157.0.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.156.0"
+__version__ = "0.157.0"

utilities/asyncio.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import asyncio
-import sys
 from asyncio import (
     Lock,
     Queue,
@@ -36,9 +35,8 @@ from typing import (
     override,
 )
 
-from utilities.errors import ImpossibleCaseError, is_instance_error
 from utilities.functions import ensure_int, ensure_not_none
-from utilities.logging import to_logger
+from utilities.os import is_pytest
 from utilities.random import SYSTEM_RANDOM
 from utilities.sentinel import Sentinel, sentinel
 from utilities.shelve import yield_shelf
@@ -70,8 +68,6 @@ if TYPE_CHECKING:
     from utilities.types import (
         Coro,
         Delta,
-        ExceptionTypeLike,
-        LoggerLike,
         MaybeCallableBoolLike,
         MaybeType,
         PathLike,
@@ -369,8 +365,6 @@ async def get_items[T](queue: Queue[T], /, *, max_size: int | None = None) -> li
     try:
         items = [await queue.get()]
     except RuntimeError as error:  # pragma: no cover
-        from utilities.pytest import is_pytest
-
         if (not is_pytest()) or (error.args[0] != "Event loop is closed"):
             raise
         return []
@@ -400,43 +394,6 @@ def get_items_nowait[T](queue: Queue[T], /, *, max_size: int | None = None) -> l
 ##
 
 
-async def loop_until_succeed(
-    func: Callable[[], Coro[None]],
-    /,
-    *,
-    logger: LoggerLike | None = None,
-    errors: ExceptionTypeLike[Exception] | None = None,
-    sleep: Delta | None = None,
-) -> bool:
-    """Repeatedly call a coroutine until it succeeds."""
-    name = get_coroutine_name(func)
-    while True:
-        try:
-            await func()
-        except Exception as error:  # noqa: BLE001
-            if logger is not None:
-                to_logger(logger).error("Error running %r", name, exc_info=True)
-            exc_type, exc_value, traceback = sys.exc_info()
-            if (exc_type is None) or (exc_value is None):  # pragma: no cover
-                raise ImpossibleCaseError(
-                    case=[f"{exc_type=}", f"{exc_value=}"]
-                ) from None
-            sys.excepthook(exc_type, exc_value, traceback)
-            if (errors is not None) and is_instance_error(error, errors):
-                return False
-            if sleep is not None:
-                if logger is not None:
-                    to_logger(logger).info("Sleeping for %s...", sleep)
-                await sleep_td(sleep)
-            if logger is not None:
-                to_logger(logger).info("Retrying %r...", name)
-        else:
-            return True
-
-
-##
-
-
 async def put_items[T](items: Iterable[T], queue: Queue[T], /) -> None:
     """Put items into a queue; if full then wait."""
     for item in items:
@@ -589,7 +546,6 @@ __all__ = [
     "get_coroutine_name",
     "get_items",
     "get_items_nowait",
-    "loop_until_succeed",
     "put_items",
     "put_items_nowait",
     "sleep_max",

utilities/eventkit.py

@@ -48,6 +48,7 @@ def add_listener[E: Event, F: Callable](
     error: Callable[[Event, BaseException], MaybeCoro[None]] | None = None,
     ignore: TypeLike[BaseException] | None = None,
     logger: LoggerLike | None = None,
+    logger_allow_pytest: bool = False,
     decorators: MaybeIterable[Callable[[F], F]] | None = None,
     done: Callable[..., MaybeCoro[None]] | None = None,
     keep_ref: bool = False,
@@ -59,6 +60,7 @@ def add_listener[E: Event, F: Callable](
         error=error,
         ignore=ignore,
         logger=logger,
+        logger_allow_pytest=logger_allow_pytest,
         decorators=decorators,
     )
     return cast("E", event.connect(lifted, done=done, keep_ref=keep_ref))
@@ -284,6 +286,7 @@ def lift_listener[F1: Callable[..., MaybeCoro[None]], F2: Callable](
     error: Callable[[Event, BaseException], MaybeCoro[None]] | None = None,
     ignore: TypeLike[BaseException] | None = None,
     logger: LoggerLike | None = None,
+    logger_allow_pytest: bool = False,
     decorators: MaybeIterable[Callable[[F2], F2]] | None = None,
 ) -> F1:
     match error, bool(iscoroutinefunction(listener)):
@@ -297,7 +300,7 @@ def lift_listener[F1: Callable[..., MaybeCoro[None]], F2: Callable](
                 except Exception as exc:  # noqa: BLE001
                     if (ignore is not None) and isinstance(exc, ignore):
                         return
-                    to_logger(logger).exception("")
+                    to_logger(logger, allow_pytest=logger_allow_pytest).exception("")
 
             lifted = listener_no_error_sync
 
@@ -311,7 +314,7 @@ def lift_listener[F1: Callable[..., MaybeCoro[None]], F2: Callable](
                 except Exception as exc:  # noqa: BLE001
                     if (ignore is not None) and isinstance(exc, ignore):
                         return
-                    to_logger(logger).exception("")
+                    to_logger(logger, allow_pytest=logger_allow_pytest).exception("")
 
             lifted = listener_no_error_async
         case _, _:

utilities/logging.py

@@ -37,6 +37,7 @@ from utilities.atomicwrites import move_many
 from utilities.dataclasses import replace_non_sentinel
 from utilities.errors import ImpossibleCaseError
 from utilities.iterables import OneEmptyError, always_iterable, one
+from utilities.os import is_pytest
 from utilities.pathlib import ensure_suffix, to_path
 from utilities.re import (
     ExtractGroupError,
@@ -95,6 +96,7 @@ def basic_config(
     filters: MaybeIterable[_FilterType] | None = None,
     plain: bool = False,
     color_field_styles: Mapping[str, _FieldStyleKeys] | None = None,
+    allow_pytest: bool = False,
 ) -> None:
     """Do the basic config."""
     match obj:
@@ -120,7 +122,7 @@ def basic_config(
             )
         case str() as name:
             basic_config(
-                obj=to_logger(name),
+                obj=to_logger(name, allow_pytest=allow_pytest),
                 format_=format_,
                 prefix=prefix,
                 hostname=hostname,
@@ -260,6 +262,7 @@ def setup_logging(
     console_level: LogLevel = "INFO",
     console_prefix: str = "❯",  # noqa: RUF001
     console_filters: MaybeIterable[_FilterType] | None = None,
+    allow_pytest: bool = False,
     files_dir: MaybeCallablePathLike = Path.cwd,
     files_max_bytes: int = _DEFAULT_MAX_BYTES,
     files_when: _When = _DEFAULT_WHEN,
@@ -276,7 +279,7 @@ def setup_logging(
         level=console_level,
         filters=console_filters,
     )
-    logger_use = to_logger(logger)
+    logger_use = to_logger(logger, allow_pytest=allow_pytest)
     name = logger_use.name
     levels: list[LogLevel] = ["DEBUG", "INFO", "ERROR"]
     for level in levels:
@@ -575,13 +578,18 @@ class _Rotation:
 ##
 
 
-def to_logger(logger: LoggerLike | None = None, /) -> Logger:
+def to_logger(
+    logger: LoggerLike | None = None, /, *, allow_pytest: bool = False
+) -> Logger:
     """Convert to a logger."""
     match logger:
         case Logger():
             return logger
         case str() | None:
-            return getLogger(logger)
+            logger = getLogger(logger)
+            if not allow_pytest:
+                _ = logger.addFilter(lambda _: not is_pytest())
+            return logger
         case never:
             assert_never(never)
 

utilities/os.py

@@ -126,7 +126,15 @@ class GetEnvVarError(Exception):
 
 def is_debug() -> bool:
     """Check if we are in `DEBUG` mode."""
-    return get_env_var("DEBUG", nullable=True) is not None
+    return "DEBUG" in environ
+
+
+##
+
+
+def is_pytest() -> bool:
+    """Check if `pytest` is running."""
+    return "PYTEST_VERSION" in environ
 
 
 ##
@@ -164,5 +172,6 @@ __all__ = [
     "get_cpu_use",
     "get_env_var",
     "is_debug",
+    "is_pytest",
     "temp_environ",
 ]

utilities/pottery.py

@@ -1,32 +1,27 @@
 from __future__ import annotations
 
-from contextlib import nullcontext, suppress
+from contextlib import suppress
 from dataclasses import dataclass
-from functools import wraps
 from sys import maxsize
 from typing import TYPE_CHECKING, override
 
-from pottery import AIORedlock, ExtendUnlockedLock
+from pottery import AIORedlock
 from pottery.exceptions import ReleaseUnlockedLock
 from redis.asyncio import Redis
 
-from utilities.asyncio import loop_until_succeed, sleep_td, timeout_td
+from utilities.asyncio import sleep_td, timeout_td
 from utilities.contextlib import enhanced_async_context_manager
-from utilities.contextvars import yield_set_context
 from utilities.iterables import always_iterable
-from utilities.logging import to_logger
 from utilities.whenever import MILLISECOND, SECOND, to_seconds
 
 if TYPE_CHECKING:
-    from collections.abc import AsyncIterator, Callable, Iterable
-    from contextvars import ContextVar
+    from collections.abc import AsyncIterator, Iterable
 
     from whenever import Delta
 
-    from utilities.types import Coro, LoggerLike, MaybeIterable
+    from utilities.types import MaybeIterable
 
 _NUM: int = 1
-_TIMEOUT_TRY_ACQUIRE: Delta = SECOND
 _TIMEOUT_RELEASE: Delta = 10 * SECOND
 _SLEEP: Delta = MILLISECOND
 
@@ -45,79 +40,6 @@ async def extend_lock(
 ##
 
 
-@enhanced_async_context_manager
-async def try_yield_coroutine_looper(
-    redis: MaybeIterable[Redis],
-    key: str,
-    /,
-    *,
-    num: int = _NUM,
-    timeout_release: Delta = _TIMEOUT_RELEASE,
-    num_extensions: int | None = None,
-    timeout_acquire: Delta = _TIMEOUT_TRY_ACQUIRE,
-    sleep_acquire: Delta = _SLEEP,
-    throttle: Delta | None = None,
-    logger: LoggerLike | None = None,
-    sleep_error: Delta | None = None,
-    context: ContextVar[bool] | None = None,
-) -> AsyncIterator[CoroutineLooper | None]:
-    """Try acquire access to a coroutine looper."""
-    try:  # skipif-ci-and-not-linux
-        async with yield_access(
-            redis,
-            key,
-            num=num,
-            timeout_release=timeout_release,
-            num_extensions=num_extensions,
-            timeout_acquire=timeout_acquire,
-            sleep=sleep_acquire,
-            throttle=throttle,
-        ) as lock:
-            looper = CoroutineLooper(lock=lock, logger=logger, sleep=sleep_error)
-            if context is None:
-                yield looper
-            else:
-                with yield_set_context(context):
-                    yield looper
-    except _YieldAccessUnableToAcquireLockError as error:  # skipif-ci-and-not-linux
-        if logger is not None:
-            to_logger(logger).info("%s", error)
-        async with nullcontext():
-            yield
-
-
-@dataclass(order=True, unsafe_hash=True, kw_only=True)
-class CoroutineLooper:
-    """Looper, guarded by a lock, to repeatedly call a coroutine until it succeeds."""
-
-    lock: AIORedlock
-    logger: LoggerLike | None = None
-    sleep: Delta | None = None
-
-    async def __call__[**P](
-        self, func: Callable[P, Coro[None]], *args: P.args, **kwargs: P.kwargs
-    ) -> bool:
-        return await loop_until_succeed(
-            lambda: self._make_coro(func, *args, **kwargs),
-            logger=self.logger,
-            errors=ExtendUnlockedLock,
-            sleep=self.sleep,
-        )
-
-    def _make_coro[**P](
-        self, func: Callable[P, Coro[None]], /, *args: P.args, **kwargs: P.kwargs
-    ) -> Coro[None]:
-        @wraps(func)
-        async def wrapped() -> None:
-            await extend_lock(lock=self.lock)
-            return await func(*args, **kwargs)
-
-        return wrapped()
-
-
-##
-
-
 @enhanced_async_context_manager
 async def yield_access(
     redis: MaybeIterable[Redis],
@@ -212,10 +134,4 @@ class _YieldAccessUnableToAcquireLockError(YieldAccessError):
         return f"Unable to acquire any 1 of {self.num} locks for {self.key!r} after {self.timeout}"  # skipif-ci-and-not-linux
 
 
-__all__ = [
-    "CoroutineLooper",
-    "YieldAccessError",
-    "extend_lock",
-    "try_yield_coroutine_looper",
-    "yield_access",
-]
+__all__ = ["YieldAccessError", "extend_lock", "yield_access"]

utilities/pytest.py

@@ -111,14 +111,6 @@ def add_pytest_configure(config: Config, options: Iterable[tuple[str, str]], /)
 ##
 
 
-def is_pytest() -> bool:
-    """Check if `pytest` is running."""
-    return "PYTEST_VERSION" in environ
-
-
-##
-
-
 def node_id_path(
     node_id: str, /, *, root: PathLike | None = None, suffix: str | None = None
 ) -> Path:
@@ -257,7 +249,6 @@ __all__ = [
     "add_pytest_addoption",
     "add_pytest_collection_modifyitems",
     "add_pytest_configure",
-    "is_pytest",
     "node_id_path",
     "skipif_linux",
     "skipif_mac",

utilities/redis.py

@@ -25,6 +25,7 @@ from utilities.contextlib import enhanced_async_context_manager
 from utilities.errors import ImpossibleCaseError
 from utilities.functions import ensure_int, identity
 from utilities.iterables import always_iterable, one
+from utilities.os import is_pytest
 from utilities.whenever import MILLISECOND, SECOND, to_milliseconds, to_seconds
 
 if TYPE_CHECKING:
@@ -730,8 +731,6 @@ async def subscribe[T](
         try:
             _ = task.cancel()
         except RuntimeError as error:  # pragma: no cover
-            from utilities.pytest import is_pytest
-
             if (not is_pytest()) or (error.args[0] != "Event loop is closed"):
                 raise
         with suppress(CancelledError):

utilities/sqlalchemy.py

@@ -95,6 +95,7 @@ from utilities.iterables import (
     merge_str_mappings,
     one,
 )
+from utilities.os import is_pytest
 from utilities.reprlib import get_repr
 from utilities.text import secret_str, snake_case
 from utilities.types import (
@@ -976,8 +977,6 @@ async def yield_connection(
         async with timeout_td(timeout, error=error), engine.begin() as conn:
             yield conn
     except GeneratorExit:  # pragma: no cover
-        from utilities.pytest import is_pytest
-
         if not is_pytest():
             raise
         return