It was tough and you need to be prepared with python skills too.
Python coding , pyspark coding , optimization skills are required
You will have to appear a coding test first post that interview with commence.
Interview questions [1]
Question 1
Python code using loop of medium difficulty
Pyspark code