From bafe0838652436c83a245feba3a3706dfd0f9457 Mon Sep 17 00:00:00 2001 From: Luca Canali Date: Wed, 3 May 2023 11:02:01 +0200 Subject: [PATCH] Minor update to the scala tests --- src/test/scala/ch/cern/sparkmeasure/StageMetricsTest.scala | 2 +- src/test/scala/ch/cern/sparkmeasure/TaskMetricsTest.scala | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/test/scala/ch/cern/sparkmeasure/StageMetricsTest.scala b/src/test/scala/ch/cern/sparkmeasure/StageMetricsTest.scala index 930ee6b..82dd872 100644 --- 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 index c768893..6f0ca51 100644 --- 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() }