Нужно ловить все исключения, не только `AssertionError`.