Всегда проверяй свой код. Простое тестирование с assert