fix: Allow consecutive calls to Table.group_by, closes #765

CHANGELOG.rst

@@ -1,3 +1,8 @@
+1.7.2
+-----
+
+* Fix consecutive calls to :meth:`.Table.group_by`. (#765)
+
 1.7.1 - Jan 4, 2023
 -------------------
 

agate/tableset/__init__.py

@@ -148,7 +148,12 @@ def _proxy(self, method_name, *args, **kwargs):
         tables = []
 
         for key, table in self.items():
-            tables.append(getattr(table, method_name)(*args, **kwargs))
+            result = getattr(table, method_name)(*args, **kwargs)
+            if isinstance(result, TableSet):
+                for table in result.values():
+                    tables.append(table)
+            else:
+                tables.append(result)
 
         return self._fork(
             tables,

docs/conf.py

@@ -36,6 +36,7 @@
 
 autodoc_default_options = {
     'members': None,
+    'member-order': 'bysource',
     'show-inheritance': True,
 }
 

tests/test_table/test_group_by.py

@@ -108,3 +108,8 @@ def test_group_by_bad_column(self):
 
         with self.assertRaises(KeyError):
             table.group_by('bad')
+
+    def test_group_by_twice(self):
+        table = Table(self.rows, self.column_names, self.column_types)
+
+        repr(table.group_by('one').group_by('two'))