Unittest в Python — пример

Напишите тест на функцию bubble_sort(), которая сортирует полученный список по возрастанию.

Проверьте работу функции:

1. С несортированным списком чисел, состоящим из int и float.
2. С пустым списком.

Подсказка:

• Несортированный список может быть любым. Например, [1, 3, 2.5] или [1, -3, 10, 2.5].
• Сортированный пустой список — это пустой список.

Код:

import unittest
from typing import List


def bubble_sort(array: List[float]) -> List[float]:
    """Сортировка списка методом пузырька по возрастанию."""
    length = len(array)
    for bypass in range(1, length):
        for k in range(0, length - bypass):
            if array[k] > array[k + 1]:
                array[k], array[k + 1] = array[k + 1], array[k]
    return array


class TestBubbleSort(unittest.TestCase):
    """Тестируем bubble_sort."""
    def test_int_float(self):
        # С несортированым списком чисел
        call = bubble_sort([2, 9, 1.5, 10])
        result = [1.5, 2, 9, 10]
        self.assertEqual(call, result,
                         'Функция bubble_sort не работает со списком чисел')

    def test_empty(self):
        # С пустым списком
        call = bubble_sort([])
        result = []
        self.assertEqual(call, result,
                         'Функция bubble_sort не работает с пустым списком')