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