[SPARK-3519] add distinct(n) to PySpark

Added missing rdd.distinct(numPartitions) and associated tests
Jenkins failed because the Python style checks didn't pass; the problem is that PEP8 requires whitespace after commas and space around operators like `*`.