diff options
author | Ankur Dave <ankurdave@gmail.com> | 2011-05-09 15:23:21 -0700 |
---|---|---|
committer | Ankur Dave <ankurdave@gmail.com> | 2011-05-09 15:23:21 -0700 |
commit | f40a0898a7f627f0d66f8393f724b518c50fba09 (patch) | |
tree | ad1de231aaab965568b2079e4a43e541c9539b05 /bagel/src/test | |
parent | c1104058c6201d5a2b59dff1babcb71523f156fe (diff) | |
download | spark-f40a0898a7f627f0d66f8393f724b518c50fba09.tar.gz spark-f40a0898a7f627f0d66f8393f724b518c50fba09.tar.bz2 spark-f40a0898a7f627f0d66f8393f724b518c50fba09.zip |
Rename bagel to spark.bagel and Pregel to Bagel
Diffstat (limited to 'bagel/src/test')
-rw-r--r-- | bagel/src/test/scala/bagel/BagelSuite.scala | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/bagel/src/test/scala/bagel/BagelSuite.scala b/bagel/src/test/scala/bagel/BagelSuite.scala index 53a93a6b80..1b47fc9ed5 100644 --- a/bagel/src/test/scala/bagel/BagelSuite.scala +++ b/bagel/src/test/scala/bagel/BagelSuite.scala @@ -1,4 +1,4 @@ -package bagel +package spark.bagel import org.scalatest.{FunSuite, Assertions} import org.scalatest.prop.Checkers @@ -10,7 +10,7 @@ import scala.collection.mutable.ArrayBuffer import spark._ -import bagel.Pregel._ +import spark.bagel.Bagel._ @serializable class TestVertex(val id: String, val active: Boolean, val age: Int) extends Vertex @serializable class TestMessage(val targetId: String) extends Message @@ -22,7 +22,7 @@ class BagelSuite extends FunSuite with Assertions { val msgs = sc.parallelize(Array[(String, TestMessage)]()) val numSupersteps = 5 val result = - Pregel.run(sc, verts, msgs)()(addAggregatorArg { + Bagel.run(sc, verts, msgs)()(addAggregatorArg { (self: TestVertex, msgs: Option[ArrayBuffer[TestMessage]], superstep: Int) => (new TestVertex(self.id, superstep < numSupersteps - 1, self.age + 1), Array[TestMessage]()) }) @@ -36,7 +36,7 @@ class BagelSuite extends FunSuite with Assertions { val msgs = sc.parallelize(Array("a" -> new TestMessage("a"))) val numSupersteps = 5 val result = - Pregel.run(sc, verts, msgs)()(addAggregatorArg { + Bagel.run(sc, verts, msgs)()(addAggregatorArg { (self: TestVertex, msgs: Option[ArrayBuffer[TestMessage]], superstep: Int) => val msgsOut = msgs match { |