dycw-utilities 0.166.29__tar.gz → 0.166.30__tar.gz

{dycw_utilities-0.166.29 → dycw_utilities-0.166.30}/PKG-INFO

@@ -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 → dycw_utilities-0.166.30}/pyproject.toml

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

{dycw_utilities-0.166.29 → dycw_utilities-0.166.30}/src/tests/test_pytest.py

@@ -302,6 +302,19 @@ class TestRunFrac:
             )
         self._run_test(testdir)
 
+    @mark.flaky
+    def test_predicate(self, *, testdir: Testdir) -> None:
+        _ = testdir.makepyfile(
+            """
+            from utilities.pytest import run_frac
+
+            @run_frac(predicate=False)
+            def test_main() -> None:
+                assert True
+            """
+        )
+        testdir.runpytest().assert_outcomes(passed=1)
+
     def _run_test(self, testdir: Testdir, /) -> None:
         result = testdir.runpytest()
         try:

{dycw_utilities-0.166.29 → dycw_utilities-0.166.30}/src/utilities/__init__.py

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

{dycw_utilities-0.166.29 → dycw_utilities-0.166.30}/src/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%})")