dycw-utilities 0.166.27__py3-none-any.whl → 0.166.29__py3-none-any.whl

{dycw_utilities-0.166.27.dist-info → dycw_utilities-0.166.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.166.27
+Version: 0.166.29
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12
@@ -12,8 +12,8 @@ Provides-Extra: logging
 Requires-Dist: coloredlogs<15.1,>=15.0.1; extra == 'logging'
 Provides-Extra: test
 Requires-Dist: dycw-pytest-only<2.2,>=2.1.1; extra == 'test'
-Requires-Dist: hypothesis<6.139,>=6.138.15; extra == 'test'
-Requires-Dist: pytest-asyncio<1.2,>=1.1.1; extra == 'test'
+Requires-Dist: hypothesis<6.139,>=6.138.16; extra == 'test'
+Requires-Dist: pytest-asyncio<1.3,>=1.2.0; extra == 'test'
 Requires-Dist: pytest-cov<7.1,>=7.0.0; extra == 'test'
 Requires-Dist: pytest-instafail<0.6,>=0.5.0; extra == 'test'
 Requires-Dist: pytest-lazy-fixtures<1.4,>=1.3.4; extra == 'test'

{dycw_utilities-0.166.27.dist-info → dycw_utilities-0.166.29.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=HjzeHsUXRBIhhvlhaEdygkT-X4czww6QieOICBEASeM,61
+utilities/__init__.py,sha256=ykhEqK5jGq8CZixk5jWUzsXv0OSPtB7rnq3h3-_XblA,61
 utilities/aeventkit.py,sha256=ddoleSwW9zdc2tjX5Ge0pMKtYwV_JMxhHYOxnWX2AGM,12609
 utilities/altair.py,sha256=nHdpWt8ZwdUwRQN970MvHd5bRWokNqzHcZQEdSHKRuE,9033
 utilities/asyncio.py,sha256=60l1IwjnRGeaVphAFiwDIHyfKoZYKY-XGpptUxGiU-M,17034
@@ -41,7 +41,7 @@ utilities/numpy.py,sha256=Xn23sA2ZbVNqwUYEgNJD3XBYH6IbCri_WkHSNhg3NkY,26122
 utilities/operator.py,sha256=C3NylZWGTVWRpwYHOPVhaLgRhw0DfpS4_XQ8KfPhBLQ,3613
 utilities/optuna.py,sha256=C-fhWYiXHVPo1l8QctYkFJ4DyhbSrGorzP1dJb_qvd8,1933
 utilities/orjson.py,sha256=rII_rINPcH9X0CfL_WeFjwHgKNkKAF-TVS5rnpLHjzc,41998
-utilities/os.py,sha256=8TjFLVWlGhhEpzZ0X_vNAyhYntjeVL5WTwaQcdTaNVw,3934
+utilities/os.py,sha256=MQxP4yVsV0sg8nPQyC8rctnimJdys_PvzxqDZfVzMMs,3935
 utilities/parse.py,sha256=JcJn5yXKhIWXBCwgBdPsyu7Hvcuw6kyEdqvaebCaI9k,17951
 utilities/pathlib.py,sha256=X6pHmfT3hnBGysdTr73uHsNaBEgKviZhk7aGWvEXDXo,8912
 utilities/pickle.py,sha256=MBT2xZCsv0pH868IXLGKnlcqNx2IRVKYNpRcqiQQqxw,653
@@ -56,7 +56,7 @@ utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 utilities/pydantic_settings.py,sha256=bFr9UDrVhdqTl7O_TPutCLdsQ5rUp--nO76-_F8z0bs,5102
 utilities/pydantic_settings_sops.py,sha256=m2uB9wDw1PU2CiSL7KwxulDXT0STjOZTu3Ctc2q7jc0,1943
 utilities/pyinstrument.py,sha256=hnXaL-4HE7wWBI5JKaPfYTpsrXe68YiuZKahHV0VJn8,841
-utilities/pytest.py,sha256=pl4j9-GReiYQMZkAFOGxsHIO_DcTHbJ_VaiLPFrw57g,8370
+utilities/pytest.py,sha256=wulDgPzABPJq6ArOiE96oWRf18cSCMj7arRtXJ8LbgU,9649
 utilities/pytest_regressions.py,sha256=ocjHTtfOeiGfQAKIei8pKNd61sxN9dawrJJ9gPt2wzA,4097
 utilities/random.py,sha256=hZlH4gnAtoaofWswuJYjcygejrY8db4CzP-z_adO2Mo,4165
 utilities/re.py,sha256=S4h-DLL6ScMPqjboZ_uQ1BVTJajrqV06r_81D--_HCE,4573
@@ -91,8 +91,8 @@ utilities/zoneinfo.py,sha256=tdIScrTB2-B-LH0ukb1HUXKooLknOfJNwHk10MuMYvA,3619
 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=mnHYBfdprz50UGVkVzV1bZERZN5CFfoF8YbokGxdFwU,1639
-dycw_utilities-0.166.27.dist-info/METADATA,sha256=zD73RDL_kiYeNBGjNY5R9krPiTk165csMHzG7DzD35I,1700
-dycw_utilities-0.166.27.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.166.27.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
-dycw_utilities-0.166.27.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.166.27.dist-info/RECORD,,
+dycw_utilities-0.166.29.dist-info/METADATA,sha256=WjWEiFbbj5kEsjpaNFZrF-_GwYdhVIoUQqO5-ddjM3g,1700
+dycw_utilities-0.166.29.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.166.29.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
+dycw_utilities-0.166.29.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.166.29.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.166.27"
+__version__ = "0.166.29"

utilities/os.py

@@ -134,7 +134,7 @@ def is_debug() -> bool:
 
 def is_pytest() -> bool:
     """Check if `pytest` is running."""
-    return get_env_var("PYTEST_VERSION", default=None) is not None
+    return get_env_var("PYTEST_VERSION", nullable=True) is not None
 
 
 ##

utilities/pytest.py

@@ -28,7 +28,8 @@ from utilities.platform import (
     IS_NOT_WINDOWS,
     IS_WINDOWS,
 )
-from utilities.types import MaybeCoro
+from utilities.random import bernoulli
+from utilities.types import MaybeCoro, Seed
 from utilities.whenever import SECOND, get_now_local
 
 if TYPE_CHECKING:
@@ -166,6 +167,47 @@ class _NodeIdToPathNotGetTailError(NodeIdToPathError):
 ##
 
 
+def run_frac[F: Callable[..., MaybeCoro[None]]](
+    *, frac: float = 0.5, seed: Seed | None = None
+) -> Callable[[F], F]:
+    """Run a test only a fraction of the time.."""
+    return cast("Any", partial(_run_frac_inner, frac=frac, seed=seed))
+
+
+def _run_frac_inner[F: Callable[..., MaybeCoro[None]]](
+    func: F, /, *, frac: float = 0.5, seed: Seed | None = None
+) -> F:
+    match bool(iscoroutinefunction(func)):
+        case False:
+
+            @wraps(func)
+            def run_frac_sync(*args: Any, **kwargs: Any) -> None:
+                _skipif_frac(frac=frac, seed=seed)
+                cast("Callable[..., None]", func)(*args, **kwargs)
+
+            return cast("Any", run_frac_sync)
+
+        case True:
+
+            @wraps(func)
+            async def run_frac_async(*args: Any, **kwargs: Any) -> None:
+                _skipif_frac(frac=frac, seed=seed)
+                await cast("Callable[..., Coro[None]]", func)(*args, **kwargs)
+
+            return cast("Any", run_frac_async)
+
+        case never:
+            assert_never(never)
+
+
+def _skipif_frac(*, frac: float = 0.5, seed: Seed | None = None) -> None:
+    if (skip is not None) and bernoulli(true=1 - frac, seed=seed):
+        _ = skip(reason=f"{_get_name()} skipped (run {frac:.0%})")
+
+
+##
+
+
 def throttle[F: Callable[..., MaybeCoro[None]]](
     *, root: PathLike | None = None, delta: Delta = SECOND, on_try: bool = False
 ) -> Callable[[F], F]:
@@ -181,7 +223,6 @@ def _throttle_inner[F: Callable[..., MaybeCoro[None]]](
     delta: Delta = SECOND,
     on_try: bool = False,
 ) -> F:
-    """Throttle a test function/method."""
     if get_env_var("THROTTLE", nullable=True) is not None:
         return func
     match bool(iscoroutinefunction(func)), on_try:
@@ -276,6 +317,7 @@ __all__ = [
     "add_pytest_collection_modifyitems",
     "add_pytest_configure",
     "node_id_path",
+    "run_frac",
     "skipif_linux",
     "skipif_mac",
     "skipif_not_linux",