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

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

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

@@ -71,7 +71,7 @@ http-test = [
   "orjson",
 ]
 hypothesis = [
-  "hypothesis >=6.138.15, <6.139",
+  "hypothesis >=6.138.16, <6.139",
 ]
 hypothesis-test = [
   "libcst",
@@ -161,7 +161,7 @@ pyinstrument = [
 ]
 pytest = [
   "pytest >=8.4.2, <8.5",
-  "pytest-asyncio >=1.1.1, <1.2",
+  "pytest-asyncio >=1.2.0, <1.3",
   "pytest-randomly >=4.0.1, <4.1",
   "pytest-timeout >=2.4.0, <2.5",
   "pytest-xdist >=3.8.0, <3.9",
@@ -246,7 +246,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.166.27"
+version = "0.166.29"
 
 [project.entry-points.pytest11]
 pytest-randomly = "utilities.pytest_plugins.pytest_randomly"
@@ -258,9 +258,9 @@ logging = [
 ]
 test = [
   "dycw-pytest-only >=2.1.1, <2.2",
-  "hypothesis >=6.138.15, <6.139",
+  "hypothesis >=6.138.16, <6.139",
   "pytest >=8.4.2, <8.5",
-  "pytest-asyncio >=1.1.1, <1.2",
+  "pytest-asyncio >=1.2.0, <1.3",
   "pytest-cov >=7.0.0, <7.1",
   "pytest-instafail >=0.5.0, <0.6",
   "pytest-lazy-fixtures >=1.3.4, <1.4",
@@ -279,7 +279,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.166.27"
+current_version = "0.166.29"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.166.27 → dycw_utilities-0.166.29}/src/tests/test_os.py

@@ -134,6 +134,10 @@ class TestIsPytest:
     def test_main(self) -> None:
         assert is_pytest()
 
+    def test_off(self) -> None:
+        with temp_environ(PYTEST_VERSION=None):
+            assert not is_pytest()
+
 
 class TestTempEnviron:
     @given(key=text.map(_prefix), value=text)

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

@@ -257,6 +257,59 @@ class TestPytestOptions:
         result.stdout.re_match_lines(list(matches))
 
 
+class TestRunFrac:
+    @mark.flaky
+    def test_basic(self, *, testdir: Testdir) -> None:
+        _ = testdir.makepyfile(
+            """
+            from utilities.pytest import run_frac
+
+            @run_frac()
+            def test_main() -> None:
+                assert True
+            """
+        )
+        self._run_test(testdir)
+
+    @mark.flaky
+    @mark.parametrize("asyncio_first", [param(True), param(False)])
+    def test_async(self, *, testdir: Testdir, asyncio_first: bool) -> None:
+        if asyncio_first:
+            _ = testdir.makepyfile(
+                """
+                from pytest import mark
+
+                from utilities.pytest import run_frac
+
+                @mark.asyncio
+                @run_frac()
+                async def test_main() -> None:
+                    assert True
+                """
+            )
+        else:
+            _ = testdir.makepyfile(
+                """
+                from pytest import mark
+
+                from utilities.pytest import run_frac
+
+                @run_frac()
+                @mark.asyncio
+                async def test_main() -> None:
+                    assert True
+                """
+            )
+        self._run_test(testdir)
+
+    def _run_test(self, testdir: Testdir, /) -> None:
+        result = testdir.runpytest()
+        try:
+            result.assert_outcomes(passed=1)
+        except AssertionError:
+            result.assert_outcomes(skipped=1)
+
+
 class TestThrottle:
     delta: ClassVar[float] = 0.5
 

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

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

{dycw_utilities-0.166.27 → dycw_utilities-0.166.29}/src/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
 
 
 ##

{dycw_utilities-0.166.27 → dycw_utilities-0.166.29}/src/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",