Add visitkeys as an alias to visit.

Author: jakirkham

docs/api/hierarchy.rst

@@ -16,6 +16,7 @@ Groups (``zarr.hierarchy``)
     .. automethod:: array_keys
     .. automethod:: arrays
     .. automethod:: visit
+    .. automethod:: visitkeys
     .. automethod:: visitvalues
     .. automethod:: visititems
     .. automethod:: create_group

zarr/hierarchy.py

@@ -57,6 +57,7 @@ class Group(MutableMapping):
     array_keys
     arrays
     visit
+    visitkeys
     visitvalues
     visititems
     create_group
@@ -497,6 +498,12 @@ def visit(self, func):
         base_len = len(self.name)
         return self.visitvalues(lambda o: func(o.name[base_len:].lstrip("/")))
 
+    def visitkeys(self, func):
+        """An alias for :py:meth:`~Group.visit`.
+        """
+
+        return self.visit(func)
+
     def visititems(self, func):
         """Run ``func`` on each object's path and the object itself.
 

zarr/tests/test_hierarchy.py

@@ -521,6 +521,24 @@ def visitor4(name, obj):
             "baz",
         ], items)
 
+        del items[:]
+        g1.visitkeys(visitor3)
+        eq([
+            "a",
+            "a/b",
+            "a/b/c",
+            "foo",
+            "foo/bar",
+            "foo/baz",
+        ], items)
+
+        del items[:]
+        g1["foo"].visitkeys(visitor3)
+        eq([
+            "bar",
+            "baz",
+        ], items)
+
         del items[:]
         g1.visititems(visitor3)
         eq([
@@ -563,9 +581,11 @@ def visitor1(val, *args):
                 return True  # pragma: no cover
 
         eq(None, g1.visit(visitor0))
+        eq(None, g1.visitkeys(visitor0))
         eq(None, g1.visitvalues(visitor0))
         eq(None, g1.visititems(visitor0))
         eq(True, g1.visit(visitor1))
+        eq(True, g1.visitkeys(visitor1))
         eq(True, g1.visitvalues(visitor1))
         eq(True, g1.visititems(visitor1))