Aurora Labs interview question

writing output validation in python