happy linter

Signed-off-by: Itay Dafna <[email protected]>

Author: ibdafna

ipydatagrid/datagrid.py

@@ -110,7 +110,9 @@ def all_values(self):
         Returns values for all selected cells as a list.
         """
         return [
-            DataGrid._get_cell_value_by_numerical_index(self._data, cell["c"], cell["r"])
+            DataGrid._get_cell_value_by_numerical_index(
+                self._data, cell["c"], cell["r"]
+            )
             for cell in self
         ]
 
@@ -290,7 +292,9 @@ def __init__(self, dataframe, **kwargs):
     def __handle_custom_msg(self, _, content, buffers):  # noqa: U101,U100
         if content["event_type"] == "cell-changed":
             row = content["row"]
-            column = DataGrid._column_index_to_name(self._data, content["column_index"])
+            column = DataGrid._column_index_to_name(
+                self._data, content["column_index"]
+            )
             value = content["value"]
             # update data on kernel
             self._data["data"][row][column] = value
@@ -359,7 +363,6 @@ def generate_data_object(dataframe, guid_key="ipydguuid"):
 
         schema["primaryKeyUuid"] = guid_key
 
-
         return {
             "data": data,
             "schema": schema,
@@ -524,7 +527,9 @@ def selected_cells(self):
         List of selected cells. Each cell is represented as a dictionary
         with keys 'r': row and 'c': column
         """
-        return SelectionHelper(self._data, self.selections, self.selection_mode).all()
+        return SelectionHelper(
+            self._data, self.selections, self.selection_mode
+        ).all()
 
     @property
     def selected_cell_values(self):
@@ -537,7 +542,9 @@ def selected_cell_values(self):
         # Serielize to JSON table schema
         view_data_object = DataGrid.generate_data_object(view_data, "ipydguuid")
 
-        return SelectionHelper(view_data_object, self.selections, self.selection_mode).all_values()
+        return SelectionHelper(
+            view_data_object, self.selections, self.selection_mode
+        ).all_values()
 
     @property
     def selected_cell_iterator(self):

tests/test_datagrid.py

@@ -15,6 +15,7 @@ def dataframe() -> None:
 def datagrid(dataframe) -> None:
     return DataGrid(dataframe)
 
+
 @pytest.fixture
 def data_object(dataframe) -> None:
     return DataGrid.generate_data_object(dataframe, "ipydguuid")
@@ -103,33 +104,52 @@ def test_get_cell_value_by_numerical_index(
     invalid_coords: bool, data_object: dict
 ) -> None:
     if invalid_coords:
-        assert DataGrid._get_cell_value_by_numerical_index(data_object, 2, 2) is None
+        assert (
+            DataGrid._get_cell_value_by_numerical_index(data_object, 2, 2)
+            is None
+        )
     else:
-        assert DataGrid._get_cell_value_by_numerical_index(data_object, 1, 0) == 4
+        assert (
+            DataGrid._get_cell_value_by_numerical_index(data_object, 1, 0) == 4
+        )
+
 
 def test_data_object_generation(dataframe: pd.DataFrame) -> None:
     data_object = DataGrid.generate_data_object(dataframe, "ipydguuid")
     expected_output = {
-        'data': [{'index': 'One', 'A': 1, 'B': 4, 'ipydguuid': 0},
-                {'index': 'Two', 'A': 2, 'B': 5, 'ipydguuid': 1},
-                {'index': 'Three', 'A': 3, 'B': 6, 'ipydguuid': 2}],
-        'schema': {'fields': [{'name': 'index', 'type': 'string'},
-                  {'name': 'A', 'type': 'integer'},
-                  {'name': 'B', 'type': 'integer'},
-                  {'name': 'ipydguuid', 'type': 'integer'}],
-                  'primaryKey': ['index', 'ipydguuid'],
-                  'pandas_version': '0.20.0',
-                  'primaryKeyUuid': 'ipydguuid'},
-        'fields': [{'index': None}, {'A': None}, {'B': None}, {'ipydguuid': None}]}
+        "data": [
+            {"index": "One", "A": 1, "B": 4, "ipydguuid": 0},
+            {"index": "Two", "A": 2, "B": 5, "ipydguuid": 1},
+            {"index": "Three", "A": 3, "B": 6, "ipydguuid": 2},
+        ],
+        "schema": {
+            "fields": [
+                {"name": "index", "type": "string"},
+                {"name": "A", "type": "integer"},
+                {"name": "B", "type": "integer"},
+                {"name": "ipydguuid", "type": "integer"},
+            ],
+            "primaryKey": ["index", "ipydguuid"],
+            "pandas_version": "0.20.0",
+            "primaryKeyUuid": "ipydguuid",
+        },
+        "fields": [
+            {"index": None},
+            {"A": None},
+            {"B": None},
+            {"ipydguuid": None},
+        ],
+    }
 
     assert data_object == expected_output
 
+
 def test_selected_cell_values(monkeypatch, datagrid, dataframe):
     # Mocking data returned from front-end
     def mock_get_visible_data():
         return dataframe
-    
+
     monkeypatch.setattr(datagrid, "get_visible_data", mock_get_visible_data)
     datagrid.select(1, 0, 2, 1)  # Select 1A to 2B
 
-    assert datagrid.selected_cell_values == [2, 5, 3, 6]
+    assert datagrid.selected_cell_values == [2, 5, 3, 6]