Apply f based on pred in map-zip

Author: tkocmathla

src/zipup/core.cljc

@@ -71,8 +71,8 @@
        last))
 
 (defn map-zip
-  "Applies f to every leaf node in the tree rooted at loc"
-  [f loc]
+  "Applies f to every node in the tree rooted at loc for which (pred loc) returns true"
+  [f pred loc]
   (if (zip/end? loc)
     (root-loc loc)
-    (recur f (zip/next (cond-> loc (leaf? loc) (zip/edit f))))))
+    (recur f pred (zip/next (cond-> loc (pred loc) (zip/edit f))))))

test/zipup/core_test.clj

@@ -46,9 +46,9 @@
 (deftest map-zip-test
   (is (= [2 :a 3 [4]]
          (->> (z/vector-zip [1 :a 2 [3]]) 
-              (zu/map-zip #(cond-> % (number? %) inc))
+              (zu/map-zip inc (comp number? z/node))
               z/node)))
   (is (= ["1" ":a" "2" ["3"]]
          (->> (z/vector-zip [1 :a 2 [3]])
-              (zu/map-zip str)
+              (zu/map-zip str zu/leaf?)
               z/node))))