Added edges function.

davidrupp · Mar 18, 2013 · 75c7a68 · 75c7a68
1 parent 9f313c1
commit 75c7a68
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/graffy/core.clj b/src/graffy/core.clj
@@ -4,6 +4,7 @@
   (add-node [this n])
   (add-edge [this s d])
   (nodes [this])
+  (edges [this n])
   (neighbors [this n]))
 
 (defn make-graph []
@@ -20,6 +21,9 @@
         this)
       (nodes [this]
         (keys @state))
+      (edges [this n]
+        (for [dest (neighbors this n)]
+          [n dest]))
       (neighbors [this n]
         (or (@state n)
             #{}))

diff --git a/test/graffy/core_test.clj b/test/graffy/core_test.clj
@@ -16,3 +16,13 @@
       (is (= #{:b} (into #{} (neighbors g :a))))
       (is (= #{:a} (into #{} (neighbors g :b))))
       (is (= #{} (into #{} (neighbors g :c)))))))
+
+(deftest test-edges
+  (testing "edges"
+    (let [g (make-graph)]
+      (-> g (add-edge :a :b)
+            (add-edge :a :c)
+            (add-edge :a :d)
+            (add-edge :b :d))
+      (is (= #{[:a :b] [:a :c] [:a :d]} (into #{} (edges g :a))))
+      (is (= #{[:d :a] [:d :b]} (into #{} (edges g :d)))))))