dycw-utilities 0.138.10__tar.gz → 0.138.12__tar.gz

{dycw_utilities-0.138.10 → dycw_utilities-0.138.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.138.10
+Version: 0.138.12
 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.136,>=6.135.14; extra == 'test'
+Requires-Dist: hypothesis<6.136,>=6.135.16; extra == 'test'
 Requires-Dist: pudb<2025.2,>=2025.1; extra == 'test'
 Requires-Dist: pytest-asyncio<1.1,>=1.0.0; extra == 'test'
 Requires-Dist: pytest-cov<6.2,>=6.1.1; extra == 'test'

{dycw_utilities-0.138.10 → dycw_utilities-0.138.12}/pyproject.toml

@@ -41,7 +41,7 @@ fastapi-test = ["httpx", "uvicorn"]
 fpdf2 = ["fpdf2 >= 2.8.3, < 2.9"]
 hashlib-test = ["orjson"]
 http-test = ["orjson"]
-hypothesis = ["hypothesis >= 6.135.14, < 6.136"]
+hypothesis = ["hypothesis >= 6.135.16, < 6.136"]
 hypothesis-test = ["libcst", "luigi", "pathvalidate", "numpy", "pytest-rerunfailures"]
 inflect = ["inflect >= 7.5.0, < 7.6"]
 jupyter-test = ["pandas", "polars"]
@@ -106,7 +106,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.138.10"
+version = "0.138.12"
 
 [project.entry-points.pytest11]
 pytest-regressions = "utilities.pytest_regressions_plugin"
@@ -117,7 +117,7 @@ logging = [
 ]
 test = [
   "dycw-pytest-only >= 2.1.1, < 2.2",
-  "hypothesis >= 6.135.14, < 6.136",
+  "hypothesis >= 6.135.16, < 6.136",
   "pudb >= 2025.1, < 2025.2",
   "pytest >= 8.3.5, < 8.4",
   "pytest-asyncio >= 1.0.0, < 1.1",
@@ -137,7 +137,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.138.10"
+current_version = "0.138.12"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.138.10 → dycw_utilities-0.138.12}/src/tests/test_polars.py

@@ -153,6 +153,7 @@ from utilities.polars import (
     nan_sum_agg,
     nan_sum_cols,
     normal,
+    order_of_magnitude,
     reify_exprs,
     replace_time_zone,
     set_first_row_as_columns,
@@ -1990,6 +1991,32 @@ class TestNormal:
         assert series.is_finite().all()
 
 
+class TestOrderOfMagnitude:
+    @given(
+        sign=sampled_from([1, -1]),
+        case=sampled_from([
+            (0.25, -0.60206, -1),
+            (0.5, -0.30103, 0),
+            (0.75, -0.1249387, 0),
+            (1.0, 0.0, 0),
+            (5.0, 0.69897, 1),
+            (10.0, 1.0, 1),
+            (50.0, 1.69897, 2),
+            (100.0, 2.0, 2),
+        ]),
+    )
+    def test_main(self, *, sign: int, case: tuple[float, float, int]) -> None:
+        x, exp_float, exp_int = case
+        x_use = Series(values=[sign * x])
+        res_float = order_of_magnitude(x_use)
+        assert res_float.dtype == Float64
+        assert_series_equal(res_float, Series([exp_float]))
+        res_int = order_of_magnitude(x_use, round_=True)
+        assert res_int.dtype == Int64
+        assert_series_equal(res_int, Series([exp_int]))
+        assert (res_int == exp_int).all()
+
+
 class TestReifyExprs:
     @given(length=hypothesis.strategies.integers(0, 10), name=text_ascii())
     def test_one_expr(self, *, length: int, name: str) -> None:

{dycw_utilities-0.138.10 → dycw_utilities-0.138.12}/src/tests/test_whenever.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import datetime as dt
 from dataclasses import dataclass, field
 from logging import DEBUG
 from typing import TYPE_CHECKING, ClassVar, Self
@@ -7,7 +8,7 @@ from zoneinfo import ZoneInfo
 
 from hypothesis import given
 from hypothesis.strategies import integers, none, sampled_from, timezones
-from pytest import raises
+from pytest import mark, param, raises
 from whenever import (
     Date,
     DateDelta,
@@ -64,6 +65,7 @@ from utilities.whenever import (
     MinMaxDateError,
     ToDaysError,
     ToNanosError,
+    ToPyTimeDeltaError,
     WheneverLogRecord,
     _FreqDayIncrementError,
     _FreqIncrementError,
@@ -87,6 +89,7 @@ from utilities.whenever import (
     to_days,
     to_local_plain,
     to_nanos,
+    to_py_time_delta,
     to_time_delta,
     to_zoned_date_time,
     two_digit_year_month,
@@ -520,6 +523,32 @@ class TestToLocalPlain:
         assert isinstance(result, PlainDateTime)
 
 
+class TestToPyTimeDelta:
+    @mark.parametrize(
+        ("delta", "expected"),
+        [
+            param(DateDelta(days=1), dt.timedelta(days=1)),
+            param(TimeDelta(microseconds=1), dt.timedelta(microseconds=1)),
+            param(
+                DateTimeDelta(days=1, microseconds=1),
+                dt.timedelta(days=1, microseconds=1),
+            ),
+        ],
+    )
+    def test_main(
+        self, *, delta: DateDelta | TimeDelta | DateTimeDelta, expected: dt.timedelta
+    ) -> None:
+        result = to_py_time_delta(delta)
+        assert result == expected
+
+    def test_error(self) -> None:
+        delta = TimeDelta(nanoseconds=1)
+        with raises(
+            ToPyTimeDeltaError, match="Time delta must not contain nanoseconds; got 1"
+        ):
+            _ = to_py_time_delta(delta)
+
+
 class TestToZonedDateTime:
     @given(date_time=zoned_datetimes())
     def test_date_time(self, *, date_time: ZonedDateTime) -> None:

{dycw_utilities-0.138.10 → dycw_utilities-0.138.12}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.138.10"
+__version__ = "0.138.12"

{dycw_utilities-0.138.10 → dycw_utilities-0.138.12}/src/utilities/polars.py

@@ -1833,6 +1833,26 @@ def normal(
 ##
 
 
+@overload
+def order_of_magnitude(column: ExprLike, /, *, round_: bool = False) -> Expr: ...
+@overload
+def order_of_magnitude(column: Series, /, *, round_: bool = False) -> Series: ...
+@overload
+def order_of_magnitude(
+    column: IntoExprColumn, /, *, round_: bool = False
+) -> Expr | Series: ...
+def order_of_magnitude(
+    column: IntoExprColumn, /, *, round_: bool = False
+) -> Expr | Series:
+    """Compute the order of magnitude of a column."""
+    column = ensure_expr_or_series(column)
+    result = column.abs().log10()
+    return result.round().cast(Int64) if round_ else result
+
+
+##
+
+
 def reify_exprs(
     *exprs: IntoExprColumn, **named_exprs: IntoExprColumn
 ) -> Expr | Series | DataFrame:
@@ -2184,6 +2204,7 @@ __all__ = [
     "nan_sum_agg",
     "nan_sum_cols",
     "normal",
+    "order_of_magnitude",
     "replace_time_zone",
     "set_first_row_as_columns",
     "struct_dtype",

{dycw_utilities-0.138.10 → dycw_utilities-0.138.12}/src/utilities/whenever.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import datetime as dt
 from collections.abc import Callable, Iterable, Mapping
 from dataclasses import dataclass
 from functools import cache
@@ -553,9 +554,10 @@ def to_local_plain(date_time: ZonedDateTime, /) -> PlainDateTime:
 
 def to_nanos(delta: DateTimeDelta, /) -> int:
     """Compute the number of nanoseconds in a date-time delta."""
-    months, days, _, _ = delta.in_months_days_secs_nanos()
-    if months != 0:
-        raise ToNanosError(months=months)
+    try:
+        days = to_days(delta.date_part())
+    except ToDaysError as error:
+        raise ToNanosError(months=error.months) from None
     return 24 * 60 * 60 * int(1e9) * days + delta.time_part().in_nanoseconds()
 
 
@@ -571,6 +573,37 @@ class ToNanosError(Exception):
 ##
 
 
+def to_py_time_delta(delta: DateDelta | TimeDelta | DateTimeDelta, /) -> dt.timedelta:
+    """Try convert a DateDelta to a standard library timedelta."""
+    match delta:
+        case DateDelta():
+            return dt.timedelta(days=to_days(delta))
+        case TimeDelta():
+            nanos = delta.in_nanoseconds()
+            micros, remainder = divmod(nanos, 1000)
+            if remainder != 0:
+                raise ToPyTimeDeltaError(nanoseconds=remainder)
+            return dt.timedelta(microseconds=micros)
+        case DateTimeDelta():
+            return to_py_time_delta(delta.date_part()) + to_py_time_delta(
+                delta.time_part()
+            )
+        case _ as never:
+            assert_never(never)
+
+
+@dataclass(kw_only=True, slots=True)
+class ToPyTimeDeltaError(Exception):
+    nanoseconds: int
+
+    @override
+    def __str__(self) -> str:
+        return f"Time delta must not contain nanoseconds; got {self.nanoseconds}"
+
+
+##
+
+
 def to_time_delta(nanos: int, /) -> TimeDelta:
     """Construct a time delta."""
     components = _to_time_delta_components(nanos)
@@ -769,6 +802,7 @@ __all__ = [
     "MinMaxDateError",
     "ToDaysError",
     "ToNanosError",
+    "ToPyTimeDeltaError",
     "WheneverLogRecord",
     "datetime_utc",
     "format_compact",
@@ -786,6 +820,7 @@ __all__ = [
     "to_days",
     "to_local_plain",
     "to_nanos",
+    "to_py_time_delta",
     "to_zoned_date_time",
     "two_digit_year_month",
 ]