test automation code snippets in python