dycw-utilities 0.153.9__tar.gz → 0.153.11__tar.gz

{dycw_utilities-0.153.9 → dycw_utilities-0.153.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.153.9
+Version: 0.153.11
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12
@@ -12,7 +12,7 @@ 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.137,>=6.136.6; extra == 'test'
+Requires-Dist: hypothesis<6.137,>=6.136.7; extra == 'test'
 Requires-Dist: pudb<2025.2,>=2025.1; extra == 'test'
 Requires-Dist: pytest-asyncio<1.2,>=1.1.0; extra == 'test'
 Requires-Dist: pytest-cov<6.3,>=6.2.1; extra == 'test'

{dycw_utilities-0.153.9 → dycw_utilities-0.153.11}/pyproject.toml

@@ -69,7 +69,7 @@ http-test = [
   "orjson",
 ]
 hypothesis = [
-  "hypothesis >=6.136.6, <6.137",
+  "hypothesis >=6.136.7, <6.137",
 ]
 hypothesis-test = [
   "libcst",
@@ -229,7 +229,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.153.9"
+version = "0.153.11"
 
 [project.entry-points.pytest11]
 pytest-randomly = "utilities.pytest_plugins.pytest_randomly"
@@ -241,7 +241,7 @@ logging = [
 ]
 test = [
   "dycw-pytest-only >=2.1.1, <2.2",
-  "hypothesis >=6.136.6, <6.137",
+  "hypothesis >=6.136.7, <6.137",
   "pudb >=2025.1, <2025.2",
   "pytest >=8.4.1, <8.5",
   "pytest-asyncio >=1.1.0, <1.2",
@@ -262,7 +262,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.153.9"
+current_version = "0.153.11"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.153.9 → dycw_utilities-0.153.11}/src/tests/test_math.py

@@ -14,7 +14,7 @@ from hypothesis.strategies import (
     sampled_from,
 )
 from numpy import iinfo, int8, int16, int32, int64, uint8, uint16, uint32, uint64
-from pytest import approx, raises
+from pytest import approx, mark, param, raises
 
 from utilities.hypothesis import int32s, numbers, pairs
 from utilities.math import (
@@ -102,6 +102,8 @@ from utilities.math import (
 )
 
 if TYPE_CHECKING:
+    from _pytest.mark import ParameterSet
+
     from utilities.types import MathRoundMode, Number
 
 
@@ -1151,33 +1153,32 @@ class TestRoundFloatImprecisions:
 
 
 class TestRoundToFloat:
-    @given(
-        case=sampled_from([
-            (0.0, 0.5, 0.0),
-            (0.1, 0.5, 0.0),
-            (0.2, 0.5, 0.0),
-            (0.3, 0.5, 0.5),
-            (0.4, 0.5, 0.5),
-            (0.5, 0.5, 0.5),
-            (0.6, 0.5, 0.5),
-            (0.7, 0.5, 0.5),
-            (0.8, 0.5, 1.0),
-            (0.9, 0.5, 1.0),
-            (1.0, 0.5, 1.0),
-            (1.1, 0.5, 1.0),
-            (1.2, 0.5, 1.0),
-            (1.3, 0.5, 1.5),
-            (1.4, 0.5, 1.5),
-            (1.5, 0.5, 1.5),
-            (1.6, 0.5, 1.5),
-            (1.7, 0.5, 1.5),
-            (1.8, 0.5, 2.0),
-            (1.9, 0.5, 2.0),
-            (2.0, 0.5, 2.0),
-        ])
-    )
-    def test_main(self, *, case: tuple[float, float, float]) -> None:
-        x, y, expected = case
+    cases: ClassVar[list[ParameterSet]] = [
+        param(0.0, 0.5, 0.0),
+        param(0.1, 0.5, 0.0),
+        param(0.2, 0.5, 0.0),
+        param(0.3, 0.5, 0.5),
+        param(0.4, 0.5, 0.5),
+        param(0.5, 0.5, 0.5),
+        param(0.6, 0.5, 0.5),
+        param(0.7, 0.5, 0.5),
+        param(0.8, 0.5, 1.0),
+        param(0.9, 0.5, 1.0),
+        param(1.0, 0.5, 1.0),
+        param(1.1, 0.5, 1.0),
+        param(1.2, 0.5, 1.0),
+        param(1.3, 0.5, 1.5),
+        param(1.4, 0.5, 1.5),
+        param(1.5, 0.5, 1.5),
+        param(1.6, 0.5, 1.5),
+        param(1.7, 0.5, 1.5),
+        param(1.8, 0.5, 2.0),
+        param(1.9, 0.5, 2.0),
+        param(2.0, 0.5, 2.0),
+    ]
+
+    @mark.parametrize(("x", "y", "expected"), cases)
+    def test_main(self, *, x: float, y: float, expected: float) -> None:
         result = round_to_float(x, y)
         assert result == approx(expected)
 

{dycw_utilities-0.153.9 → dycw_utilities-0.153.11}/src/tests/test_polars.py

@@ -60,6 +60,7 @@ from polars.testing import assert_frame_equal, assert_series_equal
 from pytest import mark, param, raises
 from whenever import Time, TimeZoneNotFoundError, ZonedDateTime
 
+import tests.test_math
 import utilities.polars
 from utilities.hypothesis import (
     assume_does_not_raise,
@@ -185,6 +186,7 @@ from utilities.polars import (
     read_series,
     reify_exprs,
     replace_time_zone,
+    round_to_float,
     serialize_dataframe,
     serialize_series,
     set_first_row_as_columns,
@@ -2417,6 +2419,24 @@ class TestReplaceTimeZone:
         assert_series_equal(result, series)
 
 
+class TestRoundToFloat:
+    @mark.parametrize(("x", "y", "exp_value"), tests.test_math.TestRoundToFloat.cases)
+    def test_main(self, *, x: float, y: float, exp_value: float) -> None:
+        series = Series(name="x", values=[x], dtype=Float64)
+        result = round_to_float(series, y)
+        expected = Series(name="x", values=[exp_value], dtype=Float64)
+        assert_series_equal(result, expected, check_exact=True)
+
+    def test_dataframe_name(self) -> None:
+        df = (
+            Series(name="x", values=[1.234], dtype=Float64)
+            .to_frame()
+            .with_columns(round_to_float("x", 0.1))
+        )
+        expected = Series(name="x", values=[1.2], dtype=Float64).to_frame()
+        assert_frame_equal(df, expected)
+
+
 class TestSerializeAndDeserializeDataFrame:
     cases: ClassVar[list[tuple[PolarsDataType, SearchStrategy[Any]]]] = [
         (Boolean, booleans()),

{dycw_utilities-0.153.9 → dycw_utilities-0.153.11}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.153.9"
+__version__ = "0.153.11"

{dycw_utilities-0.153.9 → dycw_utilities-0.153.11}/src/utilities/polars.py

@@ -114,6 +114,7 @@ if TYPE_CHECKING:
         JoinValidation,
         PolarsDataType,
         QuantileMethod,
+        RoundMode,
         SchemaDict,
         TimeUnit,
     )
@@ -2365,6 +2366,29 @@ def _reconstruct_dtype(obj: _DeconDType, /) -> PolarsDataType:
 ##
 
 
+@overload
+def round_to_float(
+    x: ExprLike, y: float, /, *, mode: RoundMode = "half_to_even"
+) -> Expr: ...
+@overload
+def round_to_float(
+    x: Series, y: float, /, *, mode: RoundMode = "half_to_even"
+) -> Series: ...
+@overload
+def round_to_float(
+    x: IntoExprColumn, y: float, /, *, mode: RoundMode = "half_to_even"
+) -> ExprOrSeries: ...
+def round_to_float(
+    x: IntoExprColumn, y: float, /, *, mode: RoundMode = "half_to_even"
+) -> ExprOrSeries:
+    """Round a column to the nearest multiple of another float."""
+    x = ensure_expr_or_series(x)
+    return (x / y).round(mode=mode) * y
+
+
+##
+
+
 def set_first_row_as_columns(df: DataFrame, /) -> DataFrame:
     """Set the first row of a DataFrame as its columns."""
     try:
@@ -2652,6 +2676,7 @@ __all__ = [
     "read_dataframe",
     "read_series",
     "replace_time_zone",
+    "round_to_float",
     "serialize_dataframe",
     "set_first_row_as_columns",
     "struct_dtype",