[SPARK-3335] [SQL] [PySpark] support broadcast in Python UDF

After this patch, broadcast can be used in Python UDF.
Actually, hold on a second: why did you move this test case from the doctests to `tests.py`?