Protium interview question

An easy level python coding question.