dycw-utilities 0.112.13__tar.gz → 0.113.0__tar.gz

{dycw_utilities-0.112.13 → dycw_utilities-0.113.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.112.13
+Version: 0.113.0
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12
@@ -137,7 +137,6 @@ Provides-Extra: zzz-test-pickle
 Requires-Dist: atomicwrites<1.5,>=1.4.1; extra == 'zzz-test-pickle'
 Provides-Extra: zzz-test-platform
 Provides-Extra: zzz-test-polars
-Requires-Dist: dacite<1.10,>=1.9.2; extra == 'zzz-test-polars'
 Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-polars'
 Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-polars'
 Provides-Extra: zzz-test-pqdm

{dycw_utilities-0.112.13 → dycw_utilities-0.113.0}/pyproject.toml

@@ -21,7 +21,6 @@ dev = [
   "concurrent-log-handler >= 0.9.25, < 0.10",
   "cryptography >= 44.0.2, < 44.1",
   "cvxpy >= 1.6.5, < 1.7",
-  "dacite >= 1.9.2, < 1.10",
   "eventkit >= 1.0.3, < 1.1",
   "fastapi >= 0.115.11, < 0.116",
   "fpdf2 >= 2.8.3, < 2.9",
@@ -93,7 +92,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.112.13"
+version = "0.113.0"
 
 [project.optional-dependencies]
 test = [
@@ -231,7 +230,6 @@ zzz-test-pathlib = []
 zzz-test-pickle = ["atomicwrites >= 1.4.1, < 1.5"]
 zzz-test-platform = []
 zzz-test-polars = [
-  "dacite >= 1.9.2, < 1.10",
   "polars-lts-cpu >= 1.29.0, < 1.30",
   "whenever >= 0.7.3, < 0.8",
 ]
@@ -336,7 +334,7 @@ zzz-test-zoneinfo = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.112.13"
+current_version = "0.113.0"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.112.13 → dycw_utilities-0.113.0}/src/tests/test_more_itertools.py

@@ -7,6 +7,7 @@ from pytest import mark, param, raises
 from utilities.more_itertools import (
     Split,
     bucket_mapping,
+    partition_list,
     partition_typeguard,
     peekable,
     yield_splits,
@@ -35,6 +36,19 @@ class TestBucketMapping:
         assert mapping["c"] == ["c1", "c2"]
 
 
+class TestPartitionList:
+    def test_main(self) -> None:
+        def is_even(x: int, /) -> bool:
+            return x % 2 == 0
+
+        iterable = [1, 2, 3, 4]
+        false, true = partition_list(is_even, iterable)
+        assert isinstance(false, list)
+        assert false == [1, 3]
+        assert isinstance(true, list)
+        assert true == [2, 4]
+
+
 class TestPartitionTypeguard:
     def test_main(self) -> None:
         def is_int(x: Any, /) -> TypeGuard[int]:

{dycw_utilities-0.112.13 → dycw_utilities-0.113.0}/src/tests/test_polars.py

@@ -87,7 +87,6 @@ from utilities.polars import (
     IsNullStructSeriesError,
     SetFirstRowAsColumnsError,
     StructFromDataClassError,
-    YieldStructSeriesElementsError,
     _check_polars_dataframe_predicates,
     _check_polars_dataframe_schema_list,
     _check_polars_dataframe_schema_set,
@@ -117,7 +116,6 @@ from utilities.polars import (
     _IsNearEventBeforeError,
     _ReifyExprsEmptyError,
     _ReifyExprsSeriesNonUniqueError,
-    _yield_struct_series_element_remove_nulls,
     ac_halflife,
     acf,
     adjust_frequencies,
@@ -167,8 +165,6 @@ from utilities.polars import (
     uniform,
     unique_element,
     week_num,
-    yield_struct_series_dataclasses,
-    yield_struct_series_elements,
     zoned_datetime,
 )
 from utilities.random import get_state
@@ -2561,154 +2557,6 @@ class TestWeekNum:
         assert_series_equal(result, expected)
 
 
-class TestYieldStructSeriesDataclasses:
-    def test_main(self) -> None:
-        @dataclass(kw_only=True, slots=True)
-        class Row:
-            lower: int | None = None
-            upper: int | None = None
-
-        series = Series(
-            name="series",
-            values=[(1, 1), (2, None), (None, 3), (None, None)],
-            dtype=Struct({"lower": Int64, "upper": Int64}),
-        )
-        result = list(yield_struct_series_dataclasses(series, Row))
-        expected = [
-            Row(lower=1, upper=1),
-            Row(lower=2, upper=None),
-            Row(lower=None, upper=3),
-            None,
-        ]
-        assert result == expected
-
-    def test_nested(self) -> None:
-        @dataclass(kw_only=True, slots=True)
-        class Inner:
-            lower: int
-            upper: int
-
-        @dataclass(kw_only=True, slots=True)
-        class Outer:
-            a: int | None = None
-            b: int | None = None
-            inner: Inner | None = None
-
-        series = Series(
-            name="series",
-            values=[
-                {"a": 1, "b": 2, "inner": {"lower": 3, "upper": 4}},
-                {"a": 1, "b": 2, "inner": None},
-                {"a": None, "b": None, "inner": {"lower": 3, "upper": 4}},
-                {"a": None, "b": None, "inner": None},
-            ],
-            dtype=Struct({
-                "a": Int64,
-                "b": Int64,
-                "inner": Struct({"lower": Int64, "upper": Int64}),
-            }),
-        )
-        result = list(
-            yield_struct_series_dataclasses(
-                series, Outer, forward_references={"Inner": Inner}
-            )
-        )
-        expected = [
-            Outer(a=1, b=2, inner=Inner(lower=3, upper=4)),
-            Outer(a=1, b=2, inner=None),
-            Outer(a=None, b=None, inner=Inner(lower=3, upper=4)),
-            None,
-        ]
-        assert result == expected
-
-
-class TestYieldStructSeriesElements:
-    def test_main(self) -> None:
-        series = Series(
-            name="series",
-            values=[(1, 1), (2, None), (None, 3), (None, None)],
-            dtype=Struct({"lower": Int64, "upper": Int64}),
-        )
-        result = list(yield_struct_series_elements(series))
-        expected = [
-            {"lower": 1, "upper": 1},
-            {"lower": 2, "upper": None},
-            {"lower": None, "upper": 3},
-            None,
-        ]
-        assert result == expected
-
-    def test_nested(self) -> None:
-        series = Series(
-            name="series",
-            values=[
-                {"a": 1, "b": 2, "inner": {"lower": 3, "upper": 4}},
-                {"a": 1, "b": 2, "inner": None},
-                {"a": None, "b": None, "inner": {"lower": 3, "upper": 4}},
-                {"a": None, "b": None, "inner": None},
-            ],
-            dtype=Struct({
-                "a": Int64,
-                "b": Int64,
-                "inner": Struct({"lower": Int64, "upper": Int64}),
-            }),
-        )
-        result = list(yield_struct_series_elements(series))
-        expected = [
-            {"a": 1, "b": 2, "inner": {"lower": 3, "upper": 4}},
-            {"a": 1, "b": 2, "inner": None},
-            {"a": None, "b": None, "inner": {"lower": 3, "upper": 4}},
-            None,
-        ]
-        assert result == expected
-
-    @given(
-        case=sampled_from([
-            (None, None),
-            (1, 1),
-            ({"a": 1, "b": 2}, {"a": 1, "b": 2}),
-            ({"a": 1, "b": None}, {"a": 1, "b": None}),
-            ({"a": None, "b": None}, None),
-            (
-                {"a": 1, "b": 2, "inner": {"lower": 3, "upper": 4}},
-                {"a": 1, "b": 2, "inner": {"lower": 3, "upper": 4}},
-            ),
-            (
-                {"a": 1, "b": 2, "inner": {"lower": None, "upper": None}},
-                {"a": 1, "b": 2, "inner": None},
-            ),
-            (
-                {"a": None, "b": None, "inner": {"lower": 3, "upper": 4}},
-                {"a": None, "b": None, "inner": {"lower": 3, "upper": 4}},
-            ),
-            ({"a": None, "b": None, "inner": {"lower": None, "upper": None}}, None),
-        ])
-    )
-    def test_remove_nulls(self, *, case: tuple[Any, Any]) -> None:
-        obj, expected = case
-        result = _yield_struct_series_element_remove_nulls(obj)
-        assert result == expected
-
-    def test_error_struct_dtype(self) -> None:
-        series = Series(name="series", values=[1, 2, 3, None], dtype=Int64)
-        with raises(
-            YieldStructSeriesElementsError,
-            match="Series must have Struct-dtype; got Int64",
-        ):
-            _ = list(yield_struct_series_elements(series))
-
-    def test_error_null_elements(self) -> None:
-        series = Series(
-            name="series",
-            values=[{"value": 1}, {"value": 2}, {"value": 3}, None],
-            dtype=Struct({"value": Int64}),
-        )
-        with raises(
-            YieldStructSeriesElementsError, match="Series must not have nulls; got .*"
-        ):
-            _ = list(yield_struct_series_elements(series, strict=True))
-
-
 class TestZonedDateTime:
     @given(time_zone=sampled_from([HongKong, UTC]))
     def test_main(self, *, time_zone: ZoneInfo) -> None:

{dycw_utilities-0.112.13 → dycw_utilities-0.113.0}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.112.13"
+__version__ = "0.113.0"

{dycw_utilities-0.112.13 → dycw_utilities-0.113.0}/src/utilities/more_itertools.py

@@ -32,6 +32,9 @@ _T = TypeVar("_T")
 _U = TypeVar("_U")
 
 
+##
+
+
 @overload
 def bucket_mapping(
     iterable: Iterable[_T], func: Callable[[_T], THashable], /, *, list: Literal[True]
@@ -55,6 +58,20 @@ def bucket_mapping(
     return mapping
 
 
+##
+
+
+def partition_list(
+    pred: Callable[[_T], bool], iterable: Iterable[_T], /
+) -> tuple[list[_T], list[_T]]:
+    """Partition with lists."""
+    false, true = partition(pred, iterable)
+    return list(false), list(true)
+
+
+##
+
+
 def partition_typeguard(
     pred: Callable[[_T], TypeGuard[_U]], iterable: Iterable[_T], /
 ) -> tuple[Iterator[_T], Iterator[_U]]:
@@ -64,6 +81,9 @@ def partition_typeguard(
     return false, true
 
 
+##
+
+
 class peekable(_peekable, Generic[_T]):  # noqa: N801
     """Peekable which supports dropwhile/takewhile methods."""
 
@@ -98,6 +118,9 @@ class peekable(_peekable, Generic[_T]):  # noqa: N801
             yield next(self)
 
 
+##
+
+
 @dataclass(kw_only=True, slots=True)
 class Split(Generic[_T]):
     """An iterable split into head/tail."""
@@ -184,4 +207,11 @@ def _yield_splits3(
         )
 
 
-__all__ = ["Split", "bucket_mapping", "partition_typeguard", "peekable", "yield_splits"]
+__all__ = [
+    "Split",
+    "bucket_mapping",
+    "partition_list",
+    "partition_typeguard",
+    "peekable",
+    "yield_splits",
+]

{dycw_utilities-0.112.13 → dycw_utilities-0.113.0}/src/utilities/polars.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import datetime as dt
 import enum
-from collections.abc import Callable, Iterable, Iterator, Mapping, Sequence
+from collections.abc import Callable, Iterator, Sequence
 from collections.abc import Set as AbstractSet
 from contextlib import suppress
 from dataclasses import asdict, dataclass
@@ -15,7 +15,6 @@ from typing import (
     Any,
     Generic,
     Literal,
-    TypeGuard,
     TypeVar,
     assert_never,
     cast,
@@ -109,10 +108,9 @@ from utilities.warnings import suppress_warnings
 from utilities.zoneinfo import UTC, ensure_time_zone, get_time_zone_name
 
 if TYPE_CHECKING:
-    from collections.abc import Callable, Iterator, Sequence
+    from collections.abc import Callable, Iterable, Iterator, Mapping, Sequence
     from collections.abc import Set as AbstractSet
 
-    from dacite.data import Data
     from polars._typing import (
         IntoExpr,  # pyright: ignore[reportPrivateImportUsage]
         IntoExprColumn,  # pyright: ignore[reportPrivateImportUsage]
@@ -126,13 +124,7 @@ if TYPE_CHECKING:
 
     from utilities.numpy import NDArrayB, NDArrayF
     from utilities.statsmodels import ACFMissing
-    from utilities.types import (
-        Dataclass,
-        MaybeIterable,
-        StrMapping,
-        TDataclass,
-        TimeZoneLike,
-    )
+    from utilities.types import Dataclass, MaybeIterable, StrMapping, TimeZoneLike
 
 
 _T = TypeVar("_T")
@@ -2134,116 +2126,6 @@ def week_num(column: IntoExprColumn, /, *, start: WeekDay = "mon") -> Expr | Ser
 ##
 
 
-@overload
-def yield_struct_series_elements(
-    series: Series, /, *, strict: Literal[True]
-) -> Iterator[Mapping[str, Any]]: ...
-@overload
-def yield_struct_series_elements(
-    series: Series, /, *, strict: Literal[False]
-) -> Iterator[Mapping[str, Any] | None]: ...
-@overload
-def yield_struct_series_elements(
-    series: Series, /, *, strict: bool = False
-) -> Iterator[Mapping[str, Any] | None]: ...
-def yield_struct_series_elements(
-    series: Series, /, *, strict: bool = False
-) -> Iterator[Mapping[str, Any] | None]:
-    """Yield the elements of a struct-dtype Series as optional mappings."""
-    if not isinstance(series.dtype, Struct):
-        raise _YieldStructSeriesElementsDTypeError(series=series)
-    if strict and series.is_null().any():
-        raise _YieldStructSeriesElementsNullElementsError(series=series)
-    for value in series:
-        yield _yield_struct_series_element_remove_nulls(value)
-
-
-def _yield_struct_series_element_remove_nulls(obj: Any, /) -> Any:
-    if not _yield_struct_series_element_is_mapping_of_str(obj):
-        return obj
-    if any(_yield_struct_series_element_is_mapping_of_str(v) for v in obj.values()):
-        result = {
-            k: _yield_struct_series_element_remove_nulls(v) for k, v in obj.items()
-        }
-        if result == obj:
-            return result
-        return _yield_struct_series_element_remove_nulls(result)
-    return None if all(v is None for v in obj.values()) else obj
-
-
-def _yield_struct_series_element_is_mapping_of_str(
-    obj: Any, /
-) -> TypeGuard[Mapping[str, Any]]:
-    return isinstance(obj, Mapping) and is_iterable_of(obj, str)
-
-
-@dataclass(kw_only=True, slots=True)
-class YieldStructSeriesElementsError(Exception):
-    series: Series
-
-
-@dataclass(kw_only=True, slots=True)
-class _YieldStructSeriesElementsDTypeError(YieldStructSeriesElementsError):
-    @override
-    def __str__(self) -> str:
-        return f"Series must have Struct-dtype; got {self.series.dtype}"
-
-
-@dataclass(kw_only=True, slots=True)
-class _YieldStructSeriesElementsNullElementsError(YieldStructSeriesElementsError):
-    @override
-    def __str__(self) -> str:
-        return f"Series must not have nulls; got {self.series}"
-
-
-##
-
-
-@overload
-def yield_struct_series_dataclasses(
-    series: Series,
-    cls: type[TDataclass],
-    /,
-    *,
-    forward_references: dict[str, Any] | None = None,
-    check_types: bool = True,
-    strict: Literal[True],
-) -> Iterator[TDataclass]: ...
-@overload
-def yield_struct_series_dataclasses(
-    series: Series,
-    cls: type[TDataclass],
-    /,
-    *,
-    forward_references: dict[str, Any] | None = None,
-    check_types: bool = True,
-    strict: bool = False,
-) -> Iterator[TDataclass | None]: ...
-def yield_struct_series_dataclasses(
-    series: Series,
-    cls: type[TDataclass],
-    /,
-    *,
-    forward_references: dict[str, Any] | None = None,
-    check_types: bool = True,
-    strict: bool = False,
-) -> Iterator[TDataclass | None]:
-    """Yield the elements of a struct-dtype Series as dataclasses."""
-    from dacite import Config, from_dict
-
-    config = Config(
-        forward_references=forward_references, check_types=check_types, strict=True
-    )
-    for value in yield_struct_series_elements(series, strict=strict):
-        if value is None:
-            yield None
-        else:
-            yield from_dict(cls, cast("Data", value), config=config)
-
-
-##
-
-
 def zoned_datetime(
     *, time_unit: TimeUnit = "us", time_zone: TimeZoneLike = UTC
 ) -> Datetime:
@@ -2272,7 +2154,6 @@ __all__ = [
     "IsNullStructSeriesError",
     "SetFirstRowAsColumnsError",
     "StructFromDataClassError",
-    "YieldStructSeriesElementsError",
     "acf",
     "adjust_frequencies",
     "append_dataclass",
@@ -2319,7 +2200,5 @@ __all__ = [
     "try_reify_expr",
     "uniform",
     "unique_element",
-    "yield_struct_series_dataclasses",
-    "yield_struct_series_elements",
     "zoned_datetime",
 ]