Unittest für Schnellsortierung generieren

Unittests für eine bestimmte Python-Funktion generieren.

Sie können ein Modell direkt abfragen und die zurückgegebenen Ergebnisse testen, wenn Sie verschiedene Parameterwerte mit der Cloud Console oder durch direkten Aufruf der Vertex AI API verwenden.

Freiform-Prompt


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

                

Antwort


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

```

              
Modell: gemini-1.0-pro-001
Temperatur: 0,2
Max. Ausgabetokens: 2.048
TopK: 40
TopP: 0,95