[PySpark] remove unnecessary use of numSlices from pyspark tests