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

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

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

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

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=mFYxtUKg-3GS-hWddHQ9eTJIsbmZslAJSALV6E4Y0KE,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
@@ -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.28.dist-info/METADATA,sha256=Z2XfjjQWdmANA-pvGFW8ucEYbpiZN3TusCqN2bhHdx8,1700
-dycw_utilities-0.166.28.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.166.28.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
-dycw_utilities-0.166.28.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.166.28.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.28"
+__version__ = "0.166.29"

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",