pre-commit run

Author: dougbrn

src/nested_pandas/nestedframe/core.py

@@ -924,7 +924,7 @@ def reduce(self, func, *args, infer_nesting=True, **kwargs) -> NestedFrame:  # t
         results_nf = NestedFrame(results, index=self.index)
 
         if infer_nesting:
-            # find potential nested structures
+            # find potential nested structures from columns
             nested_cols = []
             for column in results_nf.columns:
                 if isinstance(column, str) and "." in column:
@@ -937,7 +937,9 @@ def reduce(self, func, *args, infer_nesting=True, **kwargs) -> NestedFrame:  # t
                 layer_cols = [col for col in results_nf.columns if col.startswith(f"{layer}.")]
                 rename_df = results_nf[layer_cols].rename(columns=lambda x: x.split(".", 1)[1])
                 nested_col = pack_lists(rename_df, name=layer)
-                results_nf = results_nf[[col for col in results_nf.columns if not col.startswith(f"{layer}.")]].join(nested_col)
+                results_nf = results_nf[
+                    [col for col in results_nf.columns if not col.startswith(f"{layer}.")]
+                ].join(nested_col)
 
         return results_nf
 

tests/nested_pandas/nestedframe/test_nestedframe.py

@@ -1024,39 +1024,46 @@ def cols_allclose(col1, col2):
     def test_reduce_infer_nesting():
         """Test that nesting inference works in reduce"""
 
-        ndf = generate_data(3,20, seed=1)
+        ndf = generate_data(3, 20, seed=1)
 
         # Test simple case
         def complex_output(flux):
-            return {"max_flux":np.max(flux), "lc.flux_quantiles":np.quantile(flux, [0.1,0.2,0.3,0.4,0.5])}
-        
+            return {
+                "max_flux": np.max(flux),
+                "lc.flux_quantiles": np.quantile(flux, [0.1, 0.2, 0.3, 0.4, 0.5]),
+            }
+
         result = ndf.reduce(complex_output, "nested.flux")
         assert list(result.columns) == ["max_flux", "lc"]
-        assert list(result.lc.nest.fields) == ['flux_quantiles']
+        assert list(result.lc.nest.fields) == ["flux_quantiles"]
 
         # Test multi-column nested output
         def complex_output(flux):
-            return {"max_flux":np.max(flux),
-                    "lc.flux_quantiles":np.quantile(flux, [0.1,0.2,0.3,0.4,0.5]), 
-                    "lc.labels":[0.1,0.2,0.3,0.4,0.5]}
+            return {
+                "max_flux": np.max(flux),
+                "lc.flux_quantiles": np.quantile(flux, [0.1, 0.2, 0.3, 0.4, 0.5]),
+                "lc.labels": [0.1, 0.2, 0.3, 0.4, 0.5],
+            }
 
         result = ndf.reduce(complex_output, "nested.flux")
         assert list(result.columns) == ["max_flux", "lc"]
-        assert list(result.lc.nest.fields) == ['flux_quantiles', "labels"]
+        assert list(result.lc.nest.fields) == ["flux_quantiles", "labels"]
 
         # Test integer names
         def complex_output(flux):
-            return np.max(flux), np.quantile(flux, [0.1,0.2,0.3,0.4,0.5]),[0.1,0.2,0.3,0.4,0.5]
-        
+            return np.max(flux), np.quantile(flux, [0.1, 0.2, 0.3, 0.4, 0.5]), [0.1, 0.2, 0.3, 0.4, 0.5]
+
         result = ndf.reduce(complex_output, "nested.flux")
         assert list(result.columns) == [0, 1, 2]
 
-        # Test multiple nested structure output
+        # Test multiple nested structures output
         def complex_output(flux):
-            return {"max_flux":np.max(flux),
-                    "lc.flux_quantiles":np.quantile(flux, [0.1,0.2,0.3,0.4,0.5]),
-                    "lc.labels":[0.1,0.2,0.3,0.4,0.5],
-                    "meta.colors":["green", "red", "blue"]}
+            return {
+                "max_flux": np.max(flux),
+                "lc.flux_quantiles": np.quantile(flux, [0.1, 0.2, 0.3, 0.4, 0.5]),
+                "lc.labels": [0.1, 0.2, 0.3, 0.4, 0.5],
+                "meta.colors": ["green", "red", "blue"],
+            }
 
         result = ndf.reduce(complex_output, "nested.flux")
         assert list(result.columns) == ["max_flux", "lc", "meta"]
@@ -1065,8 +1072,10 @@ def complex_output(flux):
 
         # Test only nested structure output
         def complex_output(flux):
-            return {"lc.flux_quantiles":np.quantile(flux, [0.1,0.2,0.3,0.4,0.5]),
-                    "lc.labels":[0.1,0.2,0.3,0.4,0.5]}
+            return {
+                "lc.flux_quantiles": np.quantile(flux, [0.1, 0.2, 0.3, 0.4, 0.5]),
+                "lc.labels": [0.1, 0.2, 0.3, 0.4, 0.5],
+            }
 
         result = ndf.reduce(complex_output, "nested.flux")
         assert list(result.columns) == ["lc"]