Writing tests for factorial computation (part 7ed5)

Решение без тестов быстро ломается, поэтому добавим к задаче простые проверки.

Задача

Вычислить факториал натурального числа n: n! = 1 · 2 · … · n, а 0! = 1. Самая первая задача на циклы в учебнике.

Решение

def factorial(n: int) -> int:
    result = 1
    for i in range(2, n + 1):
        result *= i
    return result


print(factorial(6))

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

Аккумулятор result умножается на числа от 2 до n. Итеративное решение предпочтительнее рекурсивного: оно не переполняет стек вызовов.

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

720

Сложность

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

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

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

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

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

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

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

Итеративное · времени · нескольких · поэтому · минимум · проверка · Аккумулятор · короткие · натурального · Решение · предпочтительнее · учебнике · требуют · правках · работы · включая · range(2 · работает · умножается · Самая · Проверка · второй