Skip to content

Commit 96b932b

Browse files
committed
added seperate function to test cocktail shaker sort
1 parent 1c9c67c commit 96b932b

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

pydatastructs/linear_data_structures/tests/test_algorithms.py

+19-2
Original file line numberDiff line numberDiff line change
@@ -109,8 +109,25 @@ def test_counting_sort():
109109
assert counting_sort(arr)._data == expected_arr
110110

111111
def test_cocktail_shaker_sort():
112-
_test_common_sort(cocktail_shaker_sort)
113-
_test_common_sort(cocktail_shaker_sort, backend=Backend.CPP)
112+
array = OneDimensionalArray(int, [5, 3, 8, 6, 2])
113+
cocktail_shaker_sort(array, backend=Backend.PYTHON)
114+
115+
assert array._data[:5] == [2, 3, 5, 6, 8]
116+
array = OneDimensionalArray(int, [10, 1, 7, 4, 3])
117+
cocktail_shaker_sort(array, backend=Backend.CPP)
118+
assert array._data[:5] == [1, 3, 4, 7, 10]
119+
120+
array = OneDimensionalArray(int, [20, 15, 10, 5, 0])
121+
cocktail_shaker_sort(array, backend=Backend.CPP)
122+
assert array._data[:5] == [0, 5, 10, 15, 20]
123+
124+
array = OneDimensionalArray(int, [1, 2, 3, 4, 5])
125+
cocktail_shaker_sort(array, backend=Backend.CPP)
126+
assert array._data[:5] == [1, 2, 3, 4, 5]
127+
128+
array = OneDimensionalArray(int, [5, 4, 3, 2, 1])
129+
cocktail_shaker_sort(array, backend=Backend.CPP)
130+
assert array._data[:5] == [1, 2, 3, 4, 5]
114131

115132
def test_quick_sort():
116133
_test_common_sort(quick_sort)

0 commit comments

Comments
 (0)