Genera test unità di ordinamento rapido

Genera test delle unità per una determinata funzione Python.

Puoi eseguire query direttamente su un modello e testare i risultati restituiti quando utilizzi valori parametro diversi con la console Cloud o chiamando direttamente l'API Vertex AI.

Richiesta in formato libero


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)
```

                

Risposta


```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)

```

              
Modello: gemini-1.0-pro-001
Temperatura: 0.2
Numero massimo di token di output: 2048
TopK: 40
TopP: 0,95