From 9e896be375a5c0270bbdf45a2532e59bcb813efa Mon Sep 17 00:00:00 2001
From: Ankur Dave <ankurdave@gmail.com>
Date: Tue, 26 Nov 2013 15:58:55 -0800
Subject: [PATCH] Test edge filtering in subgraph (test fails)

---
 graph/src/test/scala/org/apache/spark/graph/GraphSuite.scala | 1 +
 1 file changed, 1 insertion(+)

diff --git a/graph/src/test/scala/org/apache/spark/graph/GraphSuite.scala b/graph/src/test/scala/org/apache/spark/graph/GraphSuite.scala
index ffb47fa310..8eab9b2bc6 100644
--- 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)
     }
   }