From 50cf288e34a5cf9d4359215e7b440d200a5e7a78 Mon Sep 17 00:00:00 2001 From: mvhb Date: Sat, 5 Oct 2019 21:50:21 -0300 Subject: [PATCH 1/2] adding input option and incresing the number of doctest --- sorts/stooge_sort.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/sorts/stooge_sort.py b/sorts/stooge_sort.py index 8cccd1e8c657..d4ceb718427d 100644 --- a/sorts/stooge_sort.py +++ b/sorts/stooge_sort.py @@ -1,11 +1,17 @@ def stooge_sort(arr): """ - >>> arr = [2, 4, 5, 3, 1] - >>> stooge_sort(arr) - >>> print(arr) - [1, 2, 3, 4, 5] + Examples: + >>> stooge_sort([0, 5, 3, 2, 2]) + [0, 2, 2, 3, 5] + + >>> stooge_sort([]) + [] + + >>> stooge_sort([-2, -5, -45]) + [-45, -5, -2] """ stooge(arr, 0, len(arr) - 1) + return arr def stooge(arr, i, h): @@ -29,3 +35,8 @@ def stooge(arr, i, h): # Recursively sort first 2/3 elements stooge(arr, i, (h - t)) + +if __name__ == "__main__": + user_input = input("Enter numbers separated by a comma:\n").strip() + unsorted = [int(item) for item in user_input.split(",")] + print(stooge_sort(unsorted)) From ed13069474f51834bd1e16270f188ad1ec3f6870 Mon Sep 17 00:00:00 2001 From: mvhb Date: Sun, 6 Oct 2019 13:17:10 -0300 Subject: [PATCH 2/2] mixing positive and negative numbers in the same test case --- sorts/stooge_sort.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/sorts/stooge_sort.py b/sorts/stooge_sort.py index d4ceb718427d..089b01a4def1 100644 --- a/sorts/stooge_sort.py +++ b/sorts/stooge_sort.py @@ -1,14 +1,11 @@ def stooge_sort(arr): """ Examples: - >>> stooge_sort([0, 5, 3, 2, 2]) - [0, 2, 2, 3, 5] + >>> stooge_sort([18.1, 0, -7.1, -1, 2, 2]) + [-7.1, -1, 0, 2, 2, 18.1] >>> stooge_sort([]) [] - - >>> stooge_sort([-2, -5, -45]) - [-45, -5, -2] """ stooge(arr, 0, len(arr) - 1) return arr