dycw-utilities 0.109.12__tar.gz → 0.109.13__tar.gz

{dycw_utilities-0.109.12 → dycw_utilities-0.109.13}/PKG-INFO

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

{dycw_utilities-0.109.12 → dycw_utilities-0.109.13}/pyproject.toml

@@ -91,7 +91,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.109.12"
+version = "0.109.13"
 
 [project.optional-dependencies]
 test = [
@@ -334,7 +334,7 @@ zzz-test-zoneinfo = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.109.12"
+current_version = "0.109.13"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.109.12 → dycw_utilities-0.109.13}/src/tests/test_iterables.py

@@ -97,6 +97,7 @@ from utilities.iterables import (
     ensure_iterable_not_str,
     expanding_window,
     filter_include_and_exclude,
+    group_consecutive_integers,
     groupby_lists,
     hashable_to_iterable,
     is_iterable,
@@ -118,6 +119,7 @@ from utilities.iterables import (
     sum_mappings,
     take,
     transpose,
+    ungroup_consecutive_integers,
     unique_everseen,
 )
 from utilities.sentinel import Sentinel, sentinel
@@ -890,6 +892,18 @@ class TestFilterIncludeAndExclude:
         assert result == expected
 
 
+class TestGroupAndUngroupConsecutiveIntegers:
+    @given(xs=lists(integers(), unique=True).map(sorted))
+    def test_main(self, *, xs: list[int]) -> None:
+        result = list(ungroup_consecutive_integers(group_consecutive_integers(xs)))
+        assert result == xs
+
+    def test_example(self) -> None:
+        result = list(group_consecutive_integers([1, 2, 3, 6, 7, 10, 11, 12]))
+        expected = [(1, 3), (6, 7), (10, 12)]
+        assert result == expected
+
+
 class TestGroupbyLists:
     iterable: ClassVar[str] = "AAAABBBCCDAABB"
 

{dycw_utilities-0.109.12 → dycw_utilities-0.109.13}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.109.12"
+__version__ = "0.109.13"

{dycw_utilities-0.109.12 → dycw_utilities-0.109.13}/src/utilities/iterables.py

@@ -19,7 +19,7 @@ from enum import Enum
 from functools import cmp_to_key, partial, reduce
 from itertools import accumulate, chain, groupby, islice, pairwise, product
 from math import isnan
-from operator import add, or_
+from operator import add, itemgetter, or_
 from typing import (
     TYPE_CHECKING,
     Any,
@@ -861,6 +861,24 @@ def filter_include_and_exclude(
 ##
 
 
+def group_consecutive_integers(iterable: Iterable[int], /) -> Iterable[tuple[int, int]]:
+    """Group consecutive integers."""
+    integers = sorted(iterable)
+    for _, group in groupby(enumerate(integers), key=lambda x: x[1] - x[0]):
+        as_list = list(map(itemgetter(1), group))
+        yield as_list[0], as_list[-1]
+
+
+def ungroup_consecutive_integers(
+    iterable: Iterable[tuple[int, int]], /
+) -> Iterable[int]:
+    """Ungroup consecutive integers."""
+    return chain.from_iterable(range(start, end + 1) for start, end in iterable)
+
+
+##
+
+
 @overload
 def groupby_lists(
     iterable: Iterable[_T], /, *, key: None = None
@@ -1504,6 +1522,7 @@ __all__ = [
     "ensure_iterable_not_str",
     "expanding_window",
     "filter_include_and_exclude",
+    "group_consecutive_integers",
     "groupby_lists",
     "hashable_to_iterable",
     "is_iterable",
@@ -1525,5 +1544,6 @@ __all__ = [
     "sum_mappings",
     "take",
     "transpose",
+    "ungroup_consecutive_integers",
     "unique_everseen",
 ]