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