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

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

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

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

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=ykhEqK5jGq8CZixk5jWUzsXv0OSPtB7rnq3h3-_XblA,61
+utilities/__init__.py,sha256=1ZO_cavh1o4bHb63EXlMpG07TU7njs0GEuTie2-NSEo,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=wulDgPzABPJq6ArOiE96oWRf18cSCMj7arRtXJ8LbgU,9649
+utilities/pytest.py,sha256=4KzMZYg5Zzfyjxrw8VlnZNYCD2S0Uee1JIFMIPyIkO8,10076
 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.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,,
+dycw_utilities-0.166.30.dist-info/METADATA,sha256=lYac7uqJTy4ACG9-MmZeajuG8tiZBsBMh5kU0irWduQ,1700
+dycw_utilities-0.166.30.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.166.30.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
+dycw_utilities-0.166.30.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.166.30.dist-info/RECORD,,

utilities/__init__.py

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

utilities/pytest.py

@@ -29,7 +29,8 @@ from utilities.platform import (
     IS_WINDOWS,
 )
 from utilities.random import bernoulli
-from utilities.types import MaybeCoro, Seed
+from utilities.text import to_bool
+from utilities.types import MaybeCallableBoolLike, MaybeCoro, Seed
 from utilities.whenever import SECOND, get_now_local
 
 if TYPE_CHECKING:
@@ -168,21 +169,31 @@ class _NodeIdToPathNotGetTailError(NodeIdToPathError):
 
 
 def run_frac[F: Callable[..., MaybeCoro[None]]](
-    *, frac: float = 0.5, seed: Seed | None = None
+    *,
+    predicate: MaybeCallableBoolLike | None = 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))
+    return cast(
+        "Any", partial(_run_frac_inner, predicate=predicate, frac=frac, seed=seed)
+    )
 
 
 def _run_frac_inner[F: Callable[..., MaybeCoro[None]]](
-    func: F, /, *, frac: float = 0.5, seed: Seed | None = None
+    func: F,
+    /,
+    *,
+    predicate: MaybeCallableBoolLike | None = None,
+    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)
+                _skipif_frac(predicate=predicate, frac=frac, seed=seed)
                 cast("Callable[..., None]", func)(*args, **kwargs)
 
             return cast("Any", run_frac_sync)
@@ -191,7 +202,7 @@ def _run_frac_inner[F: Callable[..., MaybeCoro[None]]](
 
             @wraps(func)
             async def run_frac_async(*args: Any, **kwargs: Any) -> None:
-                _skipif_frac(frac=frac, seed=seed)
+                _skipif_frac(predicate=predicate, frac=frac, seed=seed)
                 await cast("Callable[..., Coro[None]]", func)(*args, **kwargs)
 
             return cast("Any", run_frac_async)
@@ -200,8 +211,17 @@ def _run_frac_inner[F: Callable[..., MaybeCoro[None]]](
             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):
+def _skipif_frac(
+    *,
+    predicate: MaybeCallableBoolLike | None = None,
+    frac: float = 0.5,
+    seed: Seed | None = None,
+) -> None:
+    if skip is None:
+        return  # pragma: no cover
+    if ((predicate is None) or to_bool(predicate)) and bernoulli(
+        true=1 - frac, seed=seed
+    ):
         _ = skip(reason=f"{_get_name()} skipped (run {frac:.0%})")