It might be better to write the whole testing automation in Python. It will be more readable and easier to manage.