EY interview question

Python code and case study