Générez des tests unitaires pour une fonction Python donnée.
|
Vous pouvez interroger un modèle directement et tester les résultats renvoyés lors de l'utilisation de différentes valeurs de paramètre en utilisant la console Cloud ou en appelant directement l'API Vertex AI.
|
|||||||||||
Invite forme libre
Create unit tests for the following quick_sort function.
```python
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)
```
Réponse
```python
import unittest
class QuickSortTest(unittest.TestCase):
def test_empty_array(self):
self.assertEqual(quick_sort([]), [])
def test_single_element_array(self):
self.assertEqual(quick_sort([1]), [1])
def test_sorted_array(self):
self.assertEqual(quick_sort([1, 2, 3, 4, 5]), [1, 2, 3, 4, 5])
def test_unsorted_array(self):
self.assertEqual(quick_sort([5, 2, 1, 3, 4]), [1, 2, 3, 4, 5])
def test_array_with_duplicates(self):
self.assertEqual(quick_sort([5, 2, 1, 3, 4, 5]), [1, 2, 3, 4, 5, 5])
def test_large_array(self):
arr = [i for i in range(1000)]
self.assertEqual(quick_sort(arr), arr)
```
|
|
||||||||||