averageTime block with Scala

averageTime.scala

    def averageTime[R](block: => R, numIter: Int = 10): Unit = {
     val t0 = System.nanoTime()
     (1 to numIter).foreach( _ => block)
     val t1 = System.nanoTime()
     val averageTimeTaken = (t1 - t0) / numIter
     val timeTakenMs = averageTimeTaken / 1000000
     println("Elapsed time: " + timeTakenMs + "ms")
   }

 val testDf = spark.range(10000000).toDF.cache

 val resultSparkNative = testDf.withColumn("addOne", 'id + 1)

scala> averageTime { resultSparkNative.count }
Elapsed time: 36ms