Fix small formatting issues

Author: FrancescAlted

bench/ctable/extend.py

@@ -114,4 +114,3 @@ class RowModel(BaseModel):
 print(f"{'Python list of lists':<30} {t_from_list:>12.4f} {'1.00x':>18}")
 print(f"{'NumPy structured array':<30} {t_from_np:>12.4f} {t_from_list / t_from_np:>17.2f}x")
 print(f"{'Existing CTable':<30} {t_from_ctable:>12.4f} {t_from_list / t_from_ctable:>17.2f}x")
-

src/blosc2/__init__.py

@@ -596,7 +596,7 @@ def _raise(exc):
 """
 
 # Delayed imports for avoiding overwriting of python builtins
-from .ctable import CTable, Column
+from .ctable import Column, CTable
 from .ndarray import (
     abs,
     acos,

src/blosc2/ctable.py

@@ -200,23 +200,20 @@ def __getitem__(self, key: int | slice | list | np.ndarray):
             pos_true = _find_physical_index(self._valid_rows, key)
             return self._raw_col[int(pos_true)]
 
-
-
         elif isinstance(key, slice):
             real_pos = blosc2.where(self._valid_rows, np.arange(len(self._valid_rows))).compute()
             start, stop, step = key.indices(len(real_pos))
             mask = blosc2.zeros(len(self._table._valid_rows), dtype=np.bool_)
             if step == 1:
                 phys_start = real_pos[start]
                 phys_stop = real_pos[stop - 1]
-                mask[phys_start: phys_stop + 1] = True
+                mask[phys_start : phys_stop + 1] = True
             else:
                 lindices = np.arange(start, stop, step)
                 phys_indices = real_pos[lindices]
                 mask[phys_indices[:]] = True
             return Column(self._table, self._col_name, mask=mask)
 
-
         elif isinstance(key, (list, tuple, np.ndarray)):
             real_pos = blosc2.where(self._valid_rows, np.arange(len(self._valid_rows))).compute()
             phys_indices = np.array([real_pos[i] for i in key], dtype=np.int64)
@@ -264,11 +261,11 @@ def __iter__(self):
                 val = np.frombuffer(info.repeated_value, dtype=arr.dtype)[0]
                 if not val:
                     continue
-                yield from self._raw_col[chunk_start: chunk_start + actual_size]
+                yield from self._raw_col[chunk_start : chunk_start + actual_size]
                 continue
 
-            mask_chunk = arr[chunk_start: chunk_start + actual_size]
-            data_chunk = self._raw_col[chunk_start: chunk_start + actual_size]
+            mask_chunk = arr[chunk_start : chunk_start + actual_size]
+            data_chunk = self._raw_col[chunk_start : chunk_start + actual_size]
             yield from data_chunk[mask_chunk]
 
     def __len__(self):
@@ -308,7 +305,9 @@ def to_numpy(self):
 
 
 class CTable(Generic[RowT]):
-    def __init__(self, row_type: type[RowT], new_data=None, expected_size: int = 1_048_576, compact: bool = False) -> None:
+    def __init__(
+        self, row_type: type[RowT], new_data=None, expected_size: int = 1_048_576, compact: bool = False
+    ) -> None:
         self._row_type = row_type
         self._cols: dict[str, blosc2.NDArray] = {}
         self._n_rows: int = 0

tests/ctable/test_column.py

@@ -10,8 +10,8 @@
 import numpy as np
 import pytest
 from pydantic import BaseModel, Field
-import blosc2
 
+import blosc2
 from blosc2 import CTable
 
 
@@ -171,12 +171,14 @@ def test_column_iter():
 
     tabla3 = CTable(RowModel, new_data=DATA20)
     tabla3.delete([0, 5, 10, 15])
+    # fmt: off
     expected_score = [
         10.0, 20.0, 30.0, 40.0,
         60.0, 70.0, 80.0, 90.0,
         110.0, 120.0, 130.0, 140.0,
         160.0, 170.0, 180.0, 190.0,
     ]
+    # fmt: on
     assert list(tabla3.score) == expected_score
 
 
@@ -264,7 +266,7 @@ def test_to_array_full_column():
     col = tabla.id
 
     expected = np.array([i for i in range(20) if i not in {0, 10, 19}], dtype=np.int64)
-    np.testing.assert_array_equal(col[0:len(col)].to_numpy(), expected)
+    np.testing.assert_array_equal(col[0 : len(col)].to_numpy(), expected)
 
 
 def test_to_array_mask_does_not_include_deleted():
@@ -289,6 +291,5 @@ def test_column_view_mask_is_independent():
     np.testing.assert_array_equal(view_a.to_numpy(), np.arange(0, 5, dtype=np.int64))
 
 
-
 if __name__ == "__main__":
     pytest.main(["-v", __file__])