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

Added missing rdd.distinct(numPartitions) and associated tests
BTW, this two cases are about distinct(), just for different cases, it's also better to put them together.