Your question is UDFs in PySpark Unit Tests. Take a moment with it on the right.
Talk me through your thinking if you like. When you're confident, submit your answer and I'll grade it like a real screen (7/10 or better passes).
Describe a time you implemented PySpark UDFs and refactored them into a reusable function or class method that transforms an input DataFrame.