Код самопроверки лучше вставлять в 
```python
if __name__ == "__main__":
    pass
```