dycw-utilities 0.151.10__tar.gz → 0.151.12__tar.gz

{dycw_utilities-0.151.10 → dycw_utilities-0.151.12}/PKG-INFO

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

{dycw_utilities-0.151.10 → dycw_utilities-0.151.12}/pyproject.toml

@@ -49,9 +49,6 @@ dev = [
   "pytest-cov >=6.2.1, <6.3",
   "pytest-timeout >=2.4.0, <2.5",
 ]
-dnspython = [
-  "dnspython >=2.7.0, <2.8",
-]
 eventkit = [
   "eventkit >=1.0.3, <1.1",
 ]
@@ -214,7 +211,7 @@ statsmodels = [
   "statsmodels >=0.14.4, <0.15",
 ]
 typed-settings = [
-  "typed-settings >=24.6.0, <24.7",
+  "typed-settings >=25.0.0, <25.1",
 ]
 tzdata = [
   "tzdata >=2025.2, <2025.3",
@@ -232,7 +229,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.151.10"
+version = "0.151.12"
 
 [project.entry-points.pytest11]
 pytest-randomly = "utilities.pytest_plugins.pytest_randomly"
@@ -265,7 +262,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.151.10"
+current_version = "0.151.12"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.151.10 → dycw_utilities-0.151.12}/src/tests/test_period.py

@@ -21,9 +21,9 @@ from utilities.hypothesis import (
 )
 from utilities.period import (
     DatePeriod,
+    PeriodDict,
     TimePeriod,
     ZonedDateTimePeriod,
-    _PeriodAsDict,
     _PeriodExactEqArgumentsError,
     _PeriodInvalidError,
     _PeriodTimeZoneError,
@@ -154,7 +154,7 @@ class TestDatePeriod:
         period = DatePeriod(start, end)
         dict1 = period.to_dict()
         dstart, dend = dict1["start"], dict1["end"]
-        dict2 = _PeriodAsDict(
+        dict2 = PeriodDict(
             start=data.draw(sampled_from([dstart, dstart.py_date()])),
             end=data.draw(sampled_from([dend, dend.py_date()])),
         )
@@ -240,7 +240,7 @@ class TestTimePeriod:
         period = TimePeriod(start, end)
         dict1 = period.to_dict()
         dstart, dend = dict1["start"], dict1["end"]
-        dict2 = _PeriodAsDict(
+        dict2 = PeriodDict(
             start=data.draw(sampled_from([dstart, dstart.py_time()])),
             end=data.draw(sampled_from([dend, dend.py_time()])),
         )
@@ -478,7 +478,7 @@ class TestZonedDateTimePeriod:
         period = ZonedDateTimePeriod(start, end)
         dict1 = period.to_dict()
         dstart, dend = dict1["start"], dict1["end"]
-        dict2 = _PeriodAsDict(
+        dict2 = PeriodDict(
             start=data.draw(sampled_from([dstart, dstart.py_datetime()])),
             end=data.draw(sampled_from([dend, dend.py_datetime()])),
         )

{dycw_utilities-0.151.10 → dycw_utilities-0.151.12}/src/tests/test_polars.py

@@ -147,6 +147,7 @@ from utilities.polars import (
     cross_rolling_quantile,
     dataclass_to_dataframe,
     dataclass_to_schema,
+    decreasing_horizontal,
     deserialize_dataframe,
     deserialize_series,
     drop_null_struct_series,
@@ -159,6 +160,7 @@ from utilities.polars import (
     get_expr_name,
     get_frequency_spectrum,
     get_series_number_of_decimals,
+    increasing_horizontal,
     insert_after,
     insert_before,
     insert_between,
@@ -1594,6 +1596,39 @@ class TestGetSeriesNumberOfDecimals:
             _ = get_series_number_of_decimals(Series(dtype=Float64))
 
 
+class TestIncreasingAndDecreasingHorizontal:
+    def test_main(self) -> None:
+        df = DataFrame(
+            data=[(1, 2, 3), (1, 3, 2), (2, 1, 3), (2, 3, 1), (3, 1, 2), (3, 2, 1)],
+            schema={"x": Int64, "y": Int64, "z": Int64},
+            orient="row",
+        ).with_columns(
+            increasing_horizontal("x", "y", "z").alias("inc"),
+            decreasing_horizontal("x", "y", "z").alias("dec"),
+        )
+        inc = Series(
+            name="inc", values=[True, False, False, False, False, False], dtype=Boolean
+        )
+        assert_series_equal(df["inc"], inc)
+        dec = Series(
+            name="dec", values=[False, False, False, False, False, True], dtype=Boolean
+        )
+        assert_series_equal(df["dec"], dec)
+
+    def test_empty(self) -> None:
+        df = (
+            Series(name="x", values=[1, 2, 3], dtype=Int64)
+            .to_frame()
+            .with_columns(
+                increasing_horizontal().alias("inc"),
+                decreasing_horizontal().alias("dec"),
+            )
+        )
+        expected = Series(values=[True, True, True], dtype=Boolean)
+        assert_series_equal(df["inc"], expected, check_names=False)
+        assert_series_equal(df["dec"], expected, check_names=False)
+
+
 class TestInsertBeforeOrAfter:
     df: ClassVar[DataFrame] = DataFrame(schema={"a": Int64, "b": Int64, "c": Int64})
 

{dycw_utilities-0.151.10 → dycw_utilities-0.151.12}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.151.10"
+__version__ = "0.151.12"

{dycw_utilities-0.151.10 → dycw_utilities-0.151.12}/src/utilities/period.py

@@ -2,7 +2,16 @@ from __future__ import annotations
 
 import datetime as dt
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any, Self, TypedDict, assert_never, overload, override
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Self,
+    TypedDict,
+    TypeVar,
+    assert_never,
+    overload,
+    override,
+)
 from zoneinfo import ZoneInfo
 
 from whenever import Date, DateDelta, PlainDateTime, Time, TimeDelta, ZonedDateTime
@@ -16,8 +25,14 @@ from utilities.zoneinfo import UTC, ensure_time_zone, get_time_zone_name
 if TYPE_CHECKING:
     from utilities.types import TimeZoneLike
 
+_TDate_co = TypeVar("_TDate_co", bound=Date | dt.date, covariant=True)
+_TTime_co = TypeVar("_TTime_co", bound=Time | dt.time, covariant=True)
+_TDateTime_co = TypeVar(
+    "_TDateTime_co", bound=ZonedDateTime | dt.datetime, covariant=True
+)
 
-class _PeriodAsDict[T: Date | Time | ZonedDateTime | dt.date | dt.time | dt.datetime](
+
+class PeriodDict[T: Date | Time | ZonedDateTime | dt.date | dt.time | dt.datetime](
     TypedDict
 ):
     start: T
@@ -85,7 +100,7 @@ class DatePeriod:
         return f"{fc(start)}-{fc(end)}"
 
     @classmethod
-    def from_dict(cls, mapping: _PeriodAsDict[Date | dt.date], /) -> Self:
+    def from_dict(cls, mapping: PeriodDict[_TDate_co], /) -> Self:
         """Convert the dictionary to a period."""
         match mapping["start"]:
             case Date() as start:
@@ -109,9 +124,9 @@ class DatePeriod:
         """Replace elements of the period."""
         return replace_non_sentinel(self, start=start, end=end)
 
-    def to_dict(self) -> _PeriodAsDict[Date]:
+    def to_dict(self) -> PeriodDict[Date]:
         """Convert the period to a dictionary."""
-        return _PeriodAsDict(start=self.start, end=self.end)
+        return PeriodDict(start=self.start, end=self.end)
 
 
 @dataclass(repr=False, order=True, unsafe_hash=True, kw_only=False)
@@ -140,7 +155,7 @@ class TimePeriod:
         return DatePeriod(start, end).at((self.start, self.end), time_zone=time_zone)
 
     @classmethod
-    def from_dict(cls, mapping: _PeriodAsDict[Time | dt.time], /) -> Self:
+    def from_dict(cls, mapping: PeriodDict[_TTime_co], /) -> Self:
         """Convert the dictionary to a period."""
         match mapping["start"]:
             case Time() as start:
@@ -164,9 +179,9 @@ class TimePeriod:
         """Replace elements of the period."""
         return replace_non_sentinel(self, start=start, end=end)
 
-    def to_dict(self) -> _PeriodAsDict[Time]:
+    def to_dict(self) -> PeriodDict[Time]:
         """Convert the period to a dictionary."""
-        return _PeriodAsDict(start=self.start, end=self.end)
+        return PeriodDict(start=self.start, end=self.end)
 
 
 @dataclass(repr=False, order=True, unsafe_hash=True, kw_only=False)
@@ -271,7 +286,7 @@ class ZonedDateTimePeriod:
         return f"{fc(start.to_plain())}-{fc(end, fmt='%Y%m%dT%H')}"
 
     @classmethod
-    def from_dict(cls, mapping: _PeriodAsDict[ZonedDateTime | dt.datetime], /) -> Self:
+    def from_dict(cls, mapping: PeriodDict[_TDateTime_co], /) -> Self:
         """Convert the dictionary to a period."""
         match mapping["start"]:
             case ZonedDateTime() as start:
@@ -303,9 +318,9 @@ class ZonedDateTimePeriod:
         """The time zone of the period."""
         return ZoneInfo(self.start.tz)
 
-    def to_dict(self) -> _PeriodAsDict[ZonedDateTime]:
+    def to_dict(self) -> PeriodDict[ZonedDateTime]:
         """Convert the period to a dictionary."""
-        return _PeriodAsDict(start=self.start, end=self.end)
+        return PeriodDict(start=self.start, end=self.end)
 
     def to_tz(self, time_zone: TimeZoneLike, /) -> Self:
         """Convert the time zone."""
@@ -346,4 +361,10 @@ class _PeriodExactEqArgumentsError(PeriodError):
         return f"Invalid arguments; got {self.args}"
 
 
-__all__ = ["DatePeriod", "PeriodError", "TimePeriod", "ZonedDateTimePeriod"]
+__all__ = [
+    "DatePeriod",
+    "PeriodDict",
+    "PeriodError",
+    "TimePeriod",
+    "ZonedDateTimePeriod",
+]

{dycw_utilities-0.151.10 → dycw_utilities-0.151.12}/src/utilities/polars.py

@@ -7,7 +7,7 @@ from collections.abc import Set as AbstractSet
 from contextlib import suppress
 from dataclasses import asdict, dataclass
 from functools import partial, reduce
-from itertools import chain, product
+from itertools import chain, pairwise, product
 from math import ceil, log
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Literal, assert_never, cast, overload, override
@@ -1471,6 +1471,37 @@ class _GetSeriesNumberOfDecimalsAllNullError(GetSeriesNumberOfDecimalsError):
 ##
 
 
+@overload
+def increasing_horizontal(*columns: ExprLike) -> Expr: ...
+@overload
+def increasing_horizontal(*columns: Series) -> Series: ...
+@overload
+def increasing_horizontal(*columns: IntoExprColumn) -> Expr | Series: ...
+def increasing_horizontal(*columns: IntoExprColumn) -> Expr | Series:
+    """Check if a set of columns are increasing."""
+    columns2 = ensure_expr_or_series_many(*columns)
+    if len(columns2) == 0:
+        return lit(value=True, dtype=Boolean)
+    return all_horizontal(prev < curr for prev, curr in pairwise(columns2))
+
+
+@overload
+def decreasing_horizontal(*columns: ExprLike) -> Expr: ...
+@overload
+def decreasing_horizontal(*columns: Series) -> Series: ...
+@overload
+def decreasing_horizontal(*columns: IntoExprColumn) -> Expr | Series: ...
+def decreasing_horizontal(*columns: IntoExprColumn) -> Expr | Series:
+    """Check if a set of columns are decreasing."""
+    columns2 = ensure_expr_or_series_many(*columns)
+    if len(columns2) == 0:
+        return lit(value=True, dtype=Boolean)
+    return all_horizontal(prev > curr for prev, curr in pairwise(columns2))
+
+
+##
+
+
 def insert_after(df: DataFrame, column: str, value: IntoExprColumn, /) -> DataFrame:
     """Insert a series after an existing column; not in-place."""
     columns = df.columns
@@ -2472,6 +2503,7 @@ __all__ = [
     "cross",
     "dataclass_to_dataframe",
     "dataclass_to_schema",
+    "decreasing_horizontal",
     "deserialize_dataframe",
     "drop_null_struct_series",
     "ensure_data_type",
@@ -2483,6 +2515,7 @@ __all__ = [
     "get_expr_name",
     "get_frequency_spectrum",
     "get_series_number_of_decimals",
+    "increasing_horizontal",
     "insert_after",
     "insert_before",
     "insert_between",

dycw_utilities-0.151.10/src/tests/test_dnspython.py

@@ -1,14 +0,0 @@
-from __future__ import annotations
-
-from ipaddress import IPv4Address
-
-from tests.conftest import SKIPIF_CI_AND_MAC
-from utilities.dnspython import nslookup
-
-
-class TestNSLookup:
-    @SKIPIF_CI_AND_MAC
-    def test_main(self) -> None:
-        result = nslookup("localhost")
-        expected = [IPv4Address("127.0.0.1")]
-        assert result == expected

dycw_utilities-0.151.10/src/utilities/dnspython.py

@@ -1,17 +0,0 @@
-from __future__ import annotations
-
-from ipaddress import IPv4Address
-
-from dns.resolver import resolve
-
-from utilities.functools import cache
-
-
-@cache
-def nslookup(address: str, /) -> list[IPv4Address]:
-    """Look up a set of addresses."""
-    ans = resolve(address)  # skipif-ci-and-mac
-    return [IPv4Address(str(rd)) for rd in ans]  # skipif-ci-and-mac
-
-
-__all__ = ["nslookup"]