Clean up tests

Signed-off-by: Bradley Reynolds <[email protected]>

Author: shenanigansd

events/advent_of_code/2022/03/python/aoc_2022_03.py

@@ -1,8 +1,18 @@
 from collections import Counter
+from itertools import batched
 from pathlib import Path
 from string import ascii_lowercase, ascii_uppercase
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from collections.abc import Iterable
+
+
+def flatten[T](
+    iterable: Iterable[Iterable[T]],
+) -> Iterable[T]:
+    return [item for sub_iter in iterable for item in sub_iter]
 
-from darbia.utils.iterables import chunks, flatten
 
 priorities = dict(
     zip(ascii_lowercase + ascii_uppercase, range(1, 52 + 1), strict=False),
@@ -22,7 +32,7 @@ def part_one(values: list[str]) -> int:
 
 def part_two(values: list[str]) -> int:
     total = 0
-    for group in chunks(values, 3):
+    for group in batched(values, 3, strict=True):
         counter = Counter(flatten([list(set(chunk)) for chunk in group]))
         for character, occurrences in counter.items():
             if occurrences == 3:

events/advent_of_code/2022/05/python/aoc_2022_05.py

@@ -1,10 +1,9 @@
 import re
 from dataclasses import dataclass
+from itertools import batched
 from pathlib import Path
 from typing import TYPE_CHECKING
 
-from darbia.utils.iterables import chunks
-
 if TYPE_CHECKING:
     from collections.abc import Iterable
 
@@ -27,7 +26,7 @@ def build_from(cls, text: str) -> Ship:
         ship = Ship({index: [] for index in range(1, stack_count + 1)})
         for line in lines:
             stack = 1
-            for chunk in chunks(line, 4):
+            for chunk in batched(line, 4, strict=False):
                 if chunk[0] == "[":
                     ship.stacks[stack].append(chunk[1])
                 stack += 1  # noqa: SIM113

events/advent_of_code/2022/10/python/aoc_2022_10.py

@@ -1,7 +1,6 @@
+from itertools import batched
 from pathlib import Path
 
-from darbia.utils.iterables import chunks
-
 
 def run_cycles(text: str) -> dict[int, int]:
     cycles = {}
@@ -45,7 +44,7 @@ def part_two(text: str) -> str:
     cycles = run_cycles(text)
     for cycle, x in cycles.items():
         pixels.append("#" if x - 1 <= (cycle - 1) % 40 <= x + 1 else " ")
-    return "\n".join("".join(row) for row in chunks(pixels, 40))
+    return "\n".join("".join(row) for row in batched(pixels, 40, strict=True))
 
 
 if __name__ == "__main__":

pyproject.toml

@@ -21,6 +21,7 @@ repository = "https://github.com/shenanigansd/scratchpad"
 
 [project.optional-dependencies]
 dev = [
+    "nox",
     "prek",
     "ruff",
     "mypy",