Amazon Interview Question

An Automation exercise in python