types(unknown_sort): constrain items to Comparable

sorts/unknown_sort.py

@@ -5,8 +5,14 @@
 already O(n)
 """
 
+from typing import Any, Protocol
 
-def merge_sort(collection: list) -> list:
+
+class Comparable(Protocol):
+    def __lt__(self, other: Any, /) -> bool: ...
+
+
+def merge_sort[T: Comparable](collection: list[T]) -> list[T]:
     """Pure implementation of the fastest merge sort algorithm in Python
 
     :param collection: some mutable ordered collection with heterogeneous
@@ -22,8 +28,20 @@ def merge_sort(collection: list) -> list:
 
     >>> merge_sort([-2, -5, -45])
     [-45, -5, -2]
+
+    >>> merge_sort(["banana", "apple", "cherry"])
+    ['apple', 'banana', 'cherry']
+
+    >>> merge_sort([3.14, 1.5, 2.7])
+    [1.5, 2.7, 3.14]
+
+    >>> merge_sort([1, "a"])  # doctest: +ELLIPSIS
+    Traceback (most recent call last):
+    ...
+    TypeError: ...
     """
-    start, end = [], []
+    start: list[T] = []
+    end: list[T] = []
     while len(collection) > 1:
         min_one, max_one = min(collection), max(collection)
         start.append(min_one)

tests/test_sorts.py

@@ -49,6 +49,7 @@
 from sorts.shrink_shell_sort import shell_sort as shrink_shell_sort
 from sorts.stooge_sort import stooge_sort
 from sorts.strand_sort import strand_sort
+from sorts.unknown_sort import merge_sort as unknown_merge_sort
 
 
 def test_heap_sort() -> None:
@@ -85,6 +86,7 @@ def test_heap_sort() -> None:
     shrink_shell_sort,
     stooge_sort,
     strand_sort,
+    unknown_merge_sort,
 )
 
 
@@ -153,6 +155,7 @@ def test_rec_insertion_sort(case) -> None:
         selection_sort,
         shrink_shell_sort,
         strand_sort,
+        unknown_merge_sort,
     ],
     ids=lambda f: f.__name__,
 )