ENH: Make HDFStore iterable

closes pandas-dev#12221
toobaz · Feb 7, 2016 · 7fa8a8a · 7fa8a8a
1 parent 70f79ce
commit 7fa8a8a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/pandas/io/pytables.py b/pandas/io/pytables.py
@@ -488,6 +488,9 @@ def keys(self):
         """
         return [n._v_pathname for n in self.groups()]
 
+    def __iter__(self):
+        return iter(self.keys())
+
     def items(self):
         """
         iterate on key->group

diff --git a/pandas/io/tests/test_pytables.py b/pandas/io/tests/test_pytables.py
@@ -389,8 +389,9 @@ def test_keys(self):
             store['d'] = tm.makePanel()
             store['foo/bar'] = tm.makePanel()
             self.assertEqual(len(store), 5)
-            self.assertTrue(set(
-                store.keys()) == set(['/a', '/b', '/c', '/d', '/foo/bar']))
+            expected = set(['/a', '/b', '/c', '/d', '/foo/bar'])
+            self.assertTrue(set(store.keys()) == expected)
+            self.assertTrue(set(store) == expected)
 
     def test_repr(self):