Test edge filtering in subgraph (test fails)

2013-11-26 15:58:55 -08:00 · 2013-11-26 15:58:55 -08:00 · 9e896be375
parent 137294e2ab
commit 9e896be375
1 changed files with 1 additions and 0 deletions
--- a/graph/src/test/scala/org/apache/spark/graph/GraphSuite.scala
+++ b/graph/src/test/scala/org/apache/spark/graph/GraphSuite.scala
@ -139,6 +139,7 @@ class GraphSuite extends FunSuite with LocalSparkContext {
      val subgraph = star.subgraph(vpred = (vid, attr) => vid % 2 == 0)
      assert(subgraph.vertices.collect().toSet ===
        (0 to n / 2).map(x => (x * 2, "defaultValue")).toSet)
+      assert(subgraph.edges.collect().toSet === (1 to n / 2).map(x => Edge(0, x * 2)).toSet)
    }
  }