import unittest
def quick_sort(arr):
if len(arr) <= 1:
return arr
else:
pivot = arr[0]
left = [x for x in arr[1:] if x < pivot]
right = [x for x in arr[1:] if x >= pivot]
return quick_sort(left) + [pivot] + quick_sort(right)
class TestQuickSort(unittest.TestCase):
def test_empty_array(self):
self.assertEqual(quick_sort([]), [])
def test_single_element_array(self):
self.assertEqual(quick_sort([5]), [5])
def test_sorted_array(self):
self.assertEqual(quick_sort([1, 2, 3, 4, 5]), [1, 2, 3, 4, 5])
def test_reverse_sorted_array(self):
self.assertEqual(quick_sort([5, 4, 3, 2, 1]), [1, 2, 3, 4, 5])
def test_unsorted_array(self):
self.assertEqual(quick_sort([3, 7, 8, 5, 2, 1, 9, 4, 6]), [1, 2, 3, 4, 5, 6, 7, 8, 9])
def test_array_with_duplicates(self):
self.assertEqual(quick_sort([3, 7, 8, 5, 2, 1, 9, 4, 6, 3, 7]), [1, 2, 3, 3, 4, 5, 6, 7, 7, 8, 9])
if __name__ == '__main__':
unittest.main()
测试说明：
test_empty_array：检查函数是否正确处理空数组。test_single_element_array：检查函数是否正确处理具有单个元素的数组。test_sorted_array：检查函数是否对已排序的数组正确排序。test_reverse_sorted_array：检查函数是否对反向排序的数组正确排序。test_unsorted_array：检查函数是否对未排序的数组正确排序。test_array_with_duplicates：检查函数是否正确处理具有重复元素的数组。
运行测试：
如要运行测试，请将代码保存为 Python 文件（例如，test_quick_sort.py），然后从命令行运行该文件：
python test_quick_sort.py
系统将执行测试并输出结果。如果所有测试都通过，您应该会看到类似如下内容的输出：
....
----------------------------------------------------------------------
Ran 6 tests in 0.001s
OK
