[SPARK-3752][SQL]: Add tests for different UDF's

Use the idomatic syntax for tuple types: `Seq[(Int, Int, Int)]`