Writing tests for bracket sequence validation (part 1f42)

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

Задача

Проверить, правильно ли расставлены скобки в строке из символов ()[]{}.

Решение

def is_balanced(s: str) -> bool:
    pairs = {')': '(', ']': '[', '}': '{'}
    stack = []
    for ch in s:
        if ch in '([{':
            stack.append(ch)
        elif ch in pairs:
            if not stack or stack.pop() != pairs[ch]:
                return False
    return not stack


print(is_balanced("([]{})") )

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

Это классическое применение стека: открывающая скобка кладётся наверх, закрывающая сверяется с вершиной. Если в конце стек пуст — последовательность верна.

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

True

Сложность

O(n) по времени, O(n) по памяти в худшем случае.

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

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

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

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

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

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

корректности · stack.append(ch · символов · минимум · памяти · assert · защищает · привычка · Решение · сначала · базовая · ошибки · глазами · второй · Тесты · требуют · правильно · работает · python · скобки · кладётся · наверх · разработчика · применение · Проверить · классическое · фреймворков · случай · stack