При проверке функций всегда учитывай, как они вызываются в run.py и  в других функциях. Проверяй соответствие всех аргументов и возвращаемых значений.