diff options
author | Ankur Dave <ankurdave@gmail.com> | 2011-04-15 21:40:54 -0700 |
---|---|---|
committer | Ankur Dave <ankurdave@gmail.com> | 2011-05-03 15:40:41 -0700 |
commit | c18fa3ebc6848d2da19ac2f68c9e22870e135ecd (patch) | |
tree | cd3987a5ad9d6bba8362f7b12ee017db81c81c68 /bagel/src/test | |
parent | 1c8ca0ebe1537c8f424722294794a66ff123f132 (diff) | |
download | spark-c18fa3ebc6848d2da19ac2f68c9e22870e135ecd.tar.gz spark-c18fa3ebc6848d2da19ac2f68c9e22870e135ecd.tar.bz2 spark-c18fa3ebc6848d2da19ac2f68c9e22870e135ecd.zip |
Package combiner functions into a trait
Diffstat (limited to 'bagel/src/test')
-rw-r--r-- | bagel/src/test/scala/bagel/BagelSuite.scala | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/bagel/src/test/scala/bagel/BagelSuite.scala b/bagel/src/test/scala/bagel/BagelSuite.scala index 72aecb7fd8..29f5f0c358 100644 --- a/bagel/src/test/scala/bagel/BagelSuite.scala +++ b/bagel/src/test/scala/bagel/BagelSuite.scala @@ -20,10 +20,7 @@ class BagelSuite extends FunSuite with Assertions { val msgs = sc.parallelize(Array[(String, TestMessage)]()) val numSupersteps = 5 val result = - Pregel.run(sc, verts, msgs, - Pregel.defaultCreateCombiner[TestMessage], - Pregel.defaultMergeMsg[TestMessage], - Pregel.defaultMergeCombiners[TestMessage], 1) { + Pregel.run(sc, verts, msgs, new DefaultCombiner[TestMessage], 1) { (self: TestVertex, msgs: Option[ArrayBuffer[TestMessage]], superstep: Int) => (new TestVertex(self.id, superstep < numSupersteps - 1, self.age + 1), Array[TestMessage]()) } @@ -37,10 +34,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, - Pregel.defaultCreateCombiner[TestMessage], - Pregel.defaultMergeMsg[TestMessage], - Pregel.defaultMergeCombiners[TestMessage], 1) { + Pregel.run(sc, verts, msgs, new DefaultCombiner[TestMessage], 1) { (self: TestVertex, msgs: Option[ArrayBuffer[TestMessage]], superstep: Int) => val msgsOut = msgs match { |