Skip to content
Snippets Groups Projects
Commit c42f5d17 authored by Kay Ousterhout's avatar Kay Ousterhout
Browse files

Fixed broken tests

parent a5f8f54e
No related branches found
No related tags found
No related merge requests found
...@@ -30,11 +30,8 @@ class SparkListenerSuite extends FunSuite with LocalSparkContext with ShouldMatc ...@@ -30,11 +30,8 @@ class SparkListenerSuite extends FunSuite with LocalSparkContext with ShouldMatc
/** Length of time to wait while draining listener events. */ /** Length of time to wait while draining listener events. */
val WAIT_TIMEOUT_MILLIS = 10000 val WAIT_TIMEOUT_MILLIS = 10000
before {
sc = new SparkContext("local", "DAGSchedulerSuite")
}
test("basic creation of StageInfo") { test("basic creation of StageInfo") {
sc = new SparkContext("local", "DAGSchedulerSuite")
val listener = new SaveStageInfo val listener = new SaveStageInfo
sc.addSparkListener(listener) sc.addSparkListener(listener)
val rdd1 = sc.parallelize(1 to 100, 4) val rdd1 = sc.parallelize(1 to 100, 4)
...@@ -55,6 +52,7 @@ class SparkListenerSuite extends FunSuite with LocalSparkContext with ShouldMatc ...@@ -55,6 +52,7 @@ class SparkListenerSuite extends FunSuite with LocalSparkContext with ShouldMatc
} }
test("StageInfo with fewer tasks than partitions") { test("StageInfo with fewer tasks than partitions") {
sc = new SparkContext("local", "DAGSchedulerSuite")
val listener = new SaveStageInfo val listener = new SaveStageInfo
sc.addSparkListener(listener) sc.addSparkListener(listener)
val rdd1 = sc.parallelize(1 to 100, 4) val rdd1 = sc.parallelize(1 to 100, 4)
...@@ -70,6 +68,7 @@ class SparkListenerSuite extends FunSuite with LocalSparkContext with ShouldMatc ...@@ -70,6 +68,7 @@ class SparkListenerSuite extends FunSuite with LocalSparkContext with ShouldMatc
} }
test("local metrics") { test("local metrics") {
sc = new SparkContext("local", "DAGSchedulerSuite")
val listener = new SaveStageInfo val listener = new SaveStageInfo
sc.addSparkListener(listener) sc.addSparkListener(listener)
sc.addSparkListener(new StatsReportListener) sc.addSparkListener(new StatsReportListener)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment