dycw-utilities 0.153.12__tar.gz → 0.153.15__tar.gz

{dycw_utilities-0.153.12 → dycw_utilities-0.153.15}/PKG-INFO

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

{dycw_utilities-0.153.12 → dycw_utilities-0.153.15}/pyproject.toml

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

{dycw_utilities-0.153.12 → dycw_utilities-0.153.15}/src/tests/test_math.py

@@ -1175,6 +1175,9 @@ class TestRoundToFloat:
         param(1.8, 0.5, 2.0),
         param(1.9, 0.5, 2.0),
         param(2.0, 0.5, 2.0),
+        param(1.0745999813079834, 5e-5, 1.0746),
+        param(1.0745500326156616, 5e-5, 1.07455),
+        param(1.0745999813079834, 5e-5, 1.0746),
     ]
 
     @mark.parametrize(("x", "y", "expected"), cases)

{dycw_utilities-0.153.12 → dycw_utilities-0.153.15}/src/tests/test_polars.py

@@ -8,6 +8,7 @@ from enum import auto
 from itertools import chain, repeat
 from math import isfinite, nan
 from pathlib import Path
+from random import Random
 from typing import TYPE_CHECKING, Any, ClassVar, Literal, assert_never, cast
 from uuid import UUID, uuid4
 
@@ -202,7 +203,6 @@ from utilities.polars import (
     zoned_datetime_dtype,
     zoned_datetime_period_dtype,
 )
-from utilities.random import get_state
 from utilities.sentinel import Sentinel, sentinel
 from utilities.tzdata import HongKong, Tokyo, USCentral, USEastern
 from utilities.whenever import get_now, get_today
@@ -1440,37 +1440,44 @@ class TestFiniteEWMMean:
         6.849902999800129,
     ]
 
-    @given(
-        case=sampled_from([
-            [
+    @mark.parametrize(
+        ("alpha", "exp_base", "min_weight", "exp_result"),
+        [
+            param(
                 0.75,
-                0.9,
                 alpha_0_75_values,
+                0.9,
                 [-8.28235294117647, -8.070588235294117, 3.2705882352941176],
-            ],
-            [
+            ),
+            param(
                 0.75,
-                0.9999,
                 alpha_0_75_values,
+                0.9999,
                 [-8.269864158112174, -8.06746317849418, 3.2331284833087284],
-            ],
-            [
+            ),
+            param(
                 0.99,
-                0.9,
                 alpha_0_99_values,
+                0.9,
                 [-8.970002970002971, -8.00970200970201, 6.849915849915851],
-            ],
-            [
+            ),
+            param(
                 0.99,
-                0.9999,
                 alpha_0_99_values,
+                0.9999,
                 [-8.970002000096999, -8.00970002000097, 6.84990300128499],
-            ],
-        ])
+            ),
+        ],
     )
-    def test_main(self, *, case: tuple[float, float, list[float], list[float]]) -> None:
-        alpha, min_weight, exp_base, exp_result = case
-        state = get_state(0)
+    def test_main(
+        self,
+        *,
+        alpha: float,
+        exp_base: list[float],
+        min_weight: float,
+        exp_result: list[float],
+    ) -> None:
+        state = Random(0)
         series = Series(values=[state.randint(-10, 10) for _ in range(100)])
         base = series.ewm_mean(alpha=alpha)
         exp_base_sr = Series(values=exp_base, dtype=Float64)

{dycw_utilities-0.153.12 → dycw_utilities-0.153.15}/src/tests/test_random.py

@@ -32,6 +32,18 @@ class TestGetDockerName:
         name = get_docker_name()
         assert search(r"^[a-z]+_[a-z]+\d$", name)
 
+    def test_deterministic(self) -> None:
+        expected = [
+            "priceless_hypatia8",
+            "quizzical_aryabhata6",
+            "priceless_wing2",
+            "dazzling_mclean2",
+            "epic_montalcini8",
+        ]
+        for exp in expected:
+            result = get_docker_name(TestGetDockerName.test_deterministic.__qualname__)
+            assert result == exp
+
 
 class TestGetState:
     @given(seed=integers() | just(SYSTEM_RANDOM))
@@ -39,6 +51,14 @@ class TestGetState:
         state = get_state(seed)
         assert isinstance(state, Random)
 
+    def test_deterministic(self) -> None:
+        expected = [6, 4, 8, 10, 6]
+        for exp in expected:
+            result = get_state(TestGetState.test_deterministic.__qualname__).randint(
+                0, 10
+            )
+            assert result == exp
+
 
 class TestShuffle:
     @given(iterable=iterables(integers()), seed=integers())
@@ -46,8 +66,18 @@ class TestShuffle:
         as_set = set(iterable)
         result = shuffle(as_set, seed=seed)
         assert set(result) == as_set
-        result2 = shuffle(as_set, seed=seed)
-        assert result == result2
+
+    def test_deterministic(self) -> None:
+        expected = [
+            [2, 3, 0, 4, 1],
+            [3, 1, 2, 0, 4],
+            [1, 0, 4, 3, 2],
+            [1, 0, 2, 3, 4],
+            [4, 0, 3, 1, 2],
+        ]
+        for exp in expected:
+            result = shuffle(range(5), seed=TestShuffle.test_deterministic.__qualname__)
+            assert result == exp
 
 
 class TestSystemRandom:

{dycw_utilities-0.153.12 → dycw_utilities-0.153.15}/src/tests/test_uuid.py

@@ -6,7 +6,7 @@ from typing import TYPE_CHECKING, Self
 from uuid import UUID
 
 from hypothesis import given
-from hypothesis.strategies import integers, none, randoms, uuids
+from hypothesis.strategies import none, randoms, uuids
 
 from utilities.dataclasses import replace_non_sentinel
 from utilities.hypothesis import pairs
@@ -25,10 +25,17 @@ class TestGetUUID:
     def test_main(self, *, seed: Random | None) -> None:
         assert isinstance(get_uuid(seed), UUID)
 
-    @given(seed=integers())
-    def test_deterministic(self, *, seed: int) -> None:
-        uuid1, uuid2 = [get_uuid(seed) for _ in range(2)]
-        assert uuid1 == uuid2
+    def test_deterministic(self) -> None:
+        expected = [
+            UUID("c1887c3c-6d8b-47e3-a6e9-e8b1e502cf8f"),
+            UUID("62333b56-33ff-4020-9126-b19ff1e86e12"),
+            UUID("db8ecae8-5dab-404b-88b0-8ed406820197"),
+            UUID("95977936-0733-413b-96db-a241c3cc55d8"),
+            UUID("24e9640f-8258-495d-899e-9139cf51a545"),
+        ]
+        for exp in expected:
+            result = get_uuid(TestGetUUID.test_deterministic.__qualname__)
+            assert result == exp
 
 
 class TestToUUID:

{dycw_utilities-0.153.12 → dycw_utilities-0.153.15}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.153.12"
+__version__ = "0.153.15"

{dycw_utilities-0.153.12 → dycw_utilities-0.153.15}/src/utilities/polars.py

@@ -2383,7 +2383,8 @@ def round_to_float(
 ) -> ExprOrSeries:
     """Round a column to the nearest multiple of another float."""
     x = ensure_expr_or_series(x)
-    return (x / y).round(mode=mode) * y
+    z = (x / y).round(mode=mode) * y
+    return z.round(decimals=number_of_decimals(y) + 1)
 
 
 ##

{dycw_utilities-0.153.12 → dycw_utilities-0.153.15}/src/utilities/random.py

@@ -3,6 +3,8 @@ from __future__ import annotations
 from random import Random, SystemRandom
 from typing import TYPE_CHECKING
 
+from utilities.functools import cache
+
 if TYPE_CHECKING:
     from collections.abc import Iterable
 
@@ -47,6 +49,7 @@ _DOCKER_SUFFIXES = [
 ##
 
 
+@cache
 def get_state(seed: Seed | None = None, /) -> Random:
     """Get a random state."""
     return seed if isinstance(seed, Random) else Random(x=seed)

{dycw_utilities-0.153.12 → dycw_utilities-0.153.15}/src/utilities/sqlalchemy.py

@@ -128,7 +128,7 @@ def check_connect(engine: Engine, /) -> bool:
     try:
         with engine.connect() as conn:
             return bool(conn.execute(_SELECT).scalar_one())
-    except (gaierror, OperationalError, ProgrammingError):
+    except (gaierror, ConnectionRefusedError, OperationalError, ProgrammingError):
         return False
 
 
@@ -139,7 +139,13 @@ async def check_connect_async(
     try:
         async with timeout_td(timeout), engine.connect() as conn:
             return bool((await conn.execute(_SELECT)).scalar_one())
-    except (gaierror, OperationalError, ProgrammingError, TimeoutError):
+    except (
+        gaierror,
+        ConnectionRefusedError,
+        OperationalError,
+        ProgrammingError,
+        TimeoutError,
+    ):
         return False