Writing tests for computing an average (part 3a4d)

Посмотрим на задачу глазами разработчика: сначала решение, затем убеждаемся в его корректности.

Задача

Найти среднее арифметическое списка чисел и разобрать, что делать с пустым списком.

Решение

def average(arr: list[float]) -> float:
    if not arr:
        raise ValueError("пустой список не имеет среднего")
    return sum(arr) / len(arr)


print(average([2, 4, 6, 8]))

Как это работает

Сумма элементов делится на их количество. Отдельно обрабатывается пустой список: деление на ноль недопустимо, поэтому выбрасывается понятная ошибка.

Пример работы

5.0

Сложность

O(n) по времени, O(1) по памяти.

Проверка тестами

Хорошая привычка — проверять решение сразу на нескольких примерах, включая крайние случаи.

assert is_valid(True) is not None  # базовая проверка вызова
assert is_valid(False) is not None  # и второй случай

Такой минимум уже защищает от случайной ошибки при правках кода.

Тесты вида assert удобны для учебных задач: они короткие, наглядные и не требуют фреймворков.

Читайте также:

ошибки · Проверка · чисел · привычка · работает · Посмотрим · Сумма · Такой · правках · случайной · работы · корректности · len(arr · Решение · требуют · пустым · ошибка · наглядные · ValueError("пустой · арифметическое · фреймворков · количество · недопустимо · тестами · убеждаемся · списком · нескольких · памяти · float · пустой