Writing tests for primality testing

Writing tests for primality testing

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

Задача

Определить, является ли число n простым. Достаточно проверять делители до квадратного корня из n.

Решение

``python
def is_prime(n: int) -> bool:
    if n < 2:
        return False
    for d in range(2, int(n ** 0.5) + 1):
        if n % d == 0:
            return False
    return True

print(is_prime(29))
`

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

Если у числа есть делитель, больший корня, то у него есть и парный меньший. Поэтому перебор до √n гарантированно находит любой составной делитель.

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

`text
True
`

Сложность

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

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

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

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

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

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

View Full Gallery → https://go.cbrop.com/prse

Related:

https://isc.sans.edu/diary.html/33482#comment
https://ferment.site/m/random/t/2637
https://pastebin.com/RggyZQH9
https://discuss.tchncs.de/post/65444248
https://worknotesx4zb41.notepin.co/merge-sort-razbor