Thoughtworks interview question

Live coding in pyspark, TDD.