Minor update to the scala tests

LucaCanali · May 3, 2023 · bafe083 · bafe083
1 parent f41c72b
commit bafe083
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/test/scala/ch/cern/sparkmeasure/StageMetricsTest.scala b/src/test/scala/ch/cern/sparkmeasure/StageMetricsTest.scala
@@ -31,7 +31,7 @@ class StageMetricsTest extends FunSuite with BeforeAndAfterAll {
   }
 
   test("test StageMetrics") {
-    val stageMetrics = ch.cern.sparkmeasure.StageMetrics(spark)
+    val stageMetrics = StageMetrics(spark)
     stageMetrics.runAndMeasure {
       spark.sql("select count(*) from range(100) cross join range(100) cross join range(100)").show()
     }

diff --git a/src/test/scala/ch/cern/sparkmeasure/TaskMetricsTest.scala b/src/test/scala/ch/cern/sparkmeasure/TaskMetricsTest.scala
@@ -31,7 +31,7 @@ class TaskMetricsTest extends FunSuite with BeforeAndAfterAll {
   }
 
   test("test TaskMetrics") {
-    val taskMetrics = ch.cern.sparkmeasure.TaskMetrics(spark)
+    val taskMetrics = TaskMetrics(spark)
     taskMetrics.runAndMeasure {
       spark.sql("select count(*) from range(100) cross join range(100) cross join range(100)").show()
     }