remove-n

marick · Sep 12, 2010 · c6e88b4 · c6e88b4
1 parent 823395c
commit c6e88b4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/midje/fact_body_transformation.clj b/src/midje/fact_body_transformation.clj
@@ -45,6 +45,11 @@
 
 ;; Editing
 
+(defn remove-n [n loc]
+  (if (= n 0)
+    loc
+    (recur (- n 1) (zip/remove loc))))
+
 (defn remove-n [loc n]
   (if (= n 0)
     loc
@@ -71,7 +76,7 @@
 	additions (overrides right-hand)
 	edited-loc (zip/edit loc
 			     (fn [loc] `(expect ~loc => ~(zip/node right-hand))))]
-    (-> edited-loc zip/right zip/right zip/remove zip/remove)))
+    (-> edited-loc zip/right zip/right (remove-n 2))))
 
 ;; The meat of it.