diff options
Diffstat (limited to 'core/src/test/scala')
-rw-r--r-- | core/src/test/scala/org/apache/spark/scheduler/cluster/mesos/MesosClusterSchedulerSuite.scala | 1 | ||||
-rw-r--r-- | core/src/test/scala/org/apache/spark/storage/DiskBlockObjectWriterSuite.scala | 8 |
2 files changed, 4 insertions, 5 deletions
diff --git a/core/src/test/scala/org/apache/spark/scheduler/cluster/mesos/MesosClusterSchedulerSuite.scala b/core/src/test/scala/org/apache/spark/scheduler/cluster/mesos/MesosClusterSchedulerSuite.scala index 0260759027..87d9080de5 100644 --- a/core/src/test/scala/org/apache/spark/scheduler/cluster/mesos/MesosClusterSchedulerSuite.scala +++ b/core/src/test/scala/org/apache/spark/scheduler/cluster/mesos/MesosClusterSchedulerSuite.scala @@ -31,7 +31,6 @@ import org.scalatest.mock.MockitoSugar import org.apache.spark.{LocalSparkContext, SparkConf, SparkFunSuite} import org.apache.spark.deploy.Command import org.apache.spark.deploy.mesos.MesosDriverDescription -import org.apache.spark.scheduler.cluster.mesos.Utils class MesosClusterSchedulerSuite extends SparkFunSuite with LocalSparkContext with MockitoSugar { diff --git a/core/src/test/scala/org/apache/spark/storage/DiskBlockObjectWriterSuite.scala b/core/src/test/scala/org/apache/spark/storage/DiskBlockObjectWriterSuite.scala index 059c2c2444..684e978d11 100644 --- a/core/src/test/scala/org/apache/spark/storage/DiskBlockObjectWriterSuite.scala +++ b/core/src/test/scala/org/apache/spark/storage/DiskBlockObjectWriterSuite.scala @@ -110,13 +110,13 @@ class DiskBlockObjectWriterSuite extends SparkFunSuite with BeforeAndAfterEach { writer.write(Long.box(20), Long.box(30)) val firstSegment = writer.commitAndGet() assert(firstSegment.length === file.length()) - assert(writeMetrics.shuffleBytesWritten === file.length()) + assert(writeMetrics.bytesWritten === file.length()) writer.write(Long.box(40), Long.box(50)) writer.revertPartialWritesAndClose() assert(firstSegment.length === file.length()) - assert(writeMetrics.shuffleBytesWritten === file.length()) + assert(writeMetrics.bytesWritten === file.length()) } test("calling revertPartialWritesAndClose() after commit() should have no effect") { @@ -128,11 +128,11 @@ class DiskBlockObjectWriterSuite extends SparkFunSuite with BeforeAndAfterEach { writer.write(Long.box(20), Long.box(30)) val firstSegment = writer.commitAndGet() assert(firstSegment.length === file.length()) - assert(writeMetrics.shuffleBytesWritten === file.length()) + assert(writeMetrics.bytesWritten === file.length()) writer.revertPartialWritesAndClose() assert(firstSegment.length === file.length()) - assert(writeMetrics.shuffleBytesWritten === file.length()) + assert(writeMetrics.bytesWritten === file.length()) } test("calling revertPartialWritesAndClose() on a closed block writer should have no effect") { |