Writing tests for factorial computation (part a5f5)

Разберём решение и сразу накроем его тестами — это хороший стиль работы с кодом.

Задача

Вычислить факториал натурального числа 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 · нескольких · факториал · Разберём · Проверка · Решение · вызовов · работы · print(factorial(6 · Пример · правках · привычка · задача · примерах · python · тестами · Сложность · factorial(n · range(2 · накроем · Хорошая · кодом · включая · циклы