Formatting

Author: hombit

src/nested_pandas/nestedframe/core.py

@@ -271,7 +271,7 @@ def add_nested(
             - inner: form intersection of calling frame's index with other
               frame's index, preserving the order of the calling index.
         on : str, default: None
-            A column in the list 
+            A column in the list
         dtype : dtype or None
             NestedDtype to use for the nested column; pd.ArrowDtype or
             pa.DataType can also be used to specify the nested dtype. If None,

tests/nested_pandas/nestedframe/test_nestedframe.py

@@ -2,11 +2,10 @@
 import pandas as pd
 import pyarrow as pa
 import pytest
-from pandas.testing import assert_frame_equal
-
 from nested_pandas import NestedFrame
 from nested_pandas.datasets import generate_data
 from nested_pandas.nestedframe.core import _SeriesFromNest
+from pandas.testing import assert_frame_equal
 
 
 def test_nestedframe_construction():
@@ -188,8 +187,7 @@ def test_add_nested_with_flat_df():
 def test_add_nested_with_flat_df_and_mismatched_index():
     """Test add_nested when index values of base are missing matches in nested"""
 
-    base = NestedFrame(
-        data={"a": [1, 2, 3], "b": [2, 4, 6], "new_index": [0, 1, 3] }, index=[0, 1, 2])
+    base = NestedFrame(data={"a": [1, 2, 3], "b": [2, 4, 6], "new_index": [0, 1, 3]}, index=[0, 1, 2])
 
     nested = pd.DataFrame(
         data={
@@ -378,6 +376,7 @@ def test_add_nested_with_flat_df_and_mismatched_index():
     # we know that none of the joined values should be none
     assert not inner_res_on.isnull().values.any()
 
+
 def test_add_nested_with_series():
     """Test that add_nested correctly adds a nested column to the base df"""
 

tests/nested_pandas/series/test_accessor.py

@@ -1,14 +1,13 @@
+import nested_pandas as npd
 import numpy as np
 import pandas as pd
 import pyarrow as pa
 import pytest
-from numpy.testing import assert_array_equal
-from pandas.testing import assert_frame_equal, assert_index_equal, assert_series_equal
-
-import nested_pandas as npd
 from nested_pandas import NestedDtype
 from nested_pandas.series.ext_array import NestedExtensionArray
 from nested_pandas.series.packer import pack_flat, pack_seq
+from numpy.testing import assert_array_equal
+from pandas.testing import assert_frame_equal, assert_index_equal, assert_series_equal
 
 
 def test_registered():
@@ -983,10 +982,11 @@ def test_values():
     for value in series.nest.values():
         assert_series_equal(value, series.nest[value.name])
 
+
 def test_get_list_index():
     """Test that the get_list_index() method works."""
     # First check that an empty NestedSeries returns an empty list index.
-    empty_struct_array = pa.StructArray.from_arrays(arrays=[],names=[])
+    empty_struct_array = pa.StructArray.from_arrays(arrays=[], names=[])
     empty_series = pd.Series(empty_struct_array, dtype=NestedDtype(empty_struct_array.type), index=[])
     assert len(empty_series) == 0
     assert len(empty_series.array.get_list_index()) == 0

tests/nested_pandas/series/test_packer.py

@@ -2,11 +2,10 @@
 import pandas as pd
 import pyarrow as pa
 import pytest
-from numpy.testing import assert_array_equal
-from pandas.testing import assert_frame_equal, assert_series_equal
-
 from nested_pandas import NestedDtype
 from nested_pandas.series import packer
+from numpy.testing import assert_array_equal
+from pandas.testing import assert_frame_equal, assert_series_equal
 
 
 def offsets_reused(nested_series):
@@ -67,6 +66,7 @@ def test_pack_with_flat_df_and_index():
     offsets_reused(series)
     assert_series_equal(series, desired)
 
+
 def test_pack_with_flat_df_and_on():
     """Test packing a dataframe on a column"""
     df = pd.DataFrame(
@@ -97,6 +97,7 @@ def test_pack_with_flat_df_and_on():
     offsets_reused(series)
     assert_series_equal(series, desired)
 
+
 def test_pack_with_flat_df_and_on_and_index():
     """Test packing a dataframe on a column while also specifying an index"""
     df = pd.DataFrame(
@@ -125,6 +126,7 @@ def test_pack_with_flat_df_and_on_and_index():
     offsets_reused(series)
     assert_series_equal(series, desired)
 
+
 def test_pack_with_series_of_dfs():
     """Test pack(pd.Series([pd.DataFrame(), ...]))."""
     input_series = pd.Series(
@@ -184,6 +186,7 @@ def test_pack_flat():
     offsets_reused(actual)
     assert_series_equal(actual, desired)
 
+
 def test_pack_flat_with_on():
     """Test pack_flat() where you pack on a given column."""
     df = pd.DataFrame(
@@ -202,13 +205,13 @@ def test_pack_flat_with_on():
             # Index 0: # All of the values where column 'c' is 0
             (
                 np.array([8, 1, 3, 5]),  # values from column 'a'
-                np.array([1, 0, 0, 0])   # values from column 'b'
+                np.array([1, 0, 0, 0]),  # values from column 'b'
             ),
             # Index 1: # All of the values where column 'c' is 1
             (
                 np.array([7, 9, 2, 4, 6]),  # values from column 'a'
-                np.array([0, 0, 1, 1, 1])   # values from column 'b'
-            )
+                np.array([0, 0, 1, 1, 1]),  # values from column 'b'
+            ),
         ],
         index=[0, 1],
         dtype=NestedDtype.from_fields(dict(a=pa.int64(), b=pa.int64())),
@@ -217,6 +220,7 @@ def test_pack_flat_with_on():
     offsets_reused(actual)
     assert_series_equal(actual, desired)
 
+
 def test_pack_sorted_df_into_struct():
     """Test pack_sorted_df_into_struct()."""
     df = pd.DataFrame(