Refine Shell Sort implementation documentation

sorts/shell_sort.py

@@ -1,13 +1,15 @@
-"""
+"""Shell sort implementation.
+
+Reference:
 https://en.wikipedia.org/wiki/Shellsort#Pseudocode
 """
 
 
 def shell_sort(collection: list[int]) -> list[int]:
-    """Pure implementation of shell sort algorithm in Python
-    :param collection:  Some mutable ordered collection with heterogeneous
-    comparable items inside
-    :return:  the same collection ordered by ascending
+    """Sort a list of integers in ascending order using shell sort.
+
+    :param collection: A list of integers
+    :return: The same list sorted in ascending order
 
     >>> shell_sort([0, 5, 3, 2, 2])
     [0, 2, 2, 3, 5]
@@ -16,25 +18,17 @@
     >>> shell_sort([-2, -5, -45])
     [-45, -5, -2]
     """
-    # Marcin Ciura's gap sequence
-
     gaps = [701, 301, 132, 57, 23, 10, 4, 1]
+
     for gap in gaps:
         for i in range(gap, len(collection)):
             insert_value = collection[i]
             j = i
+
             while j >= gap and collection[j - gap] > insert_value:
                 collection[j] = collection[j - gap]
                 j -= gap
-            if j != i:
-                collection[j] = insert_value
-    return collection
-
 
-if __name__ == "__main__":
-    from doctest import testmod
+            collection[j] = insert_value
 
-    testmod()
-    user_input = input("Enter numbers separated by a comma:\n").strip()
-    unsorted = [int(item) for item in user_input.split(",")]
-    print(shell_sort(unsorted))
+    return collection