Add notebookDocumentSync to capabilities

Author: tkrabel-db

pylsp/python_lsp.py

@@ -268,9 +268,9 @@ def capabilities(self):
                 },
                 'openClose': True,
             },
-            'notebookDocumentSync' : {
+            'notebookDocumentSync': {
                 'notebookSelector': {
-                    'cells': [{ 'language': 'python' }]
+                    'cells': [{'language': 'python'}]
                 }
             },
             'workspace': {
@@ -397,7 +397,7 @@ def _lint_text_document(self, doc_uri, workspace, is_saved):
     def _lint_notebook_document(self, notebook_document, workspace):
         """
         Lint a notebook document.
-        
+
         This is a bit more complicated than linting a text document, because we need to
         send the entire notebook document to the pylsp_lint hook, but we need to send
         the diagnostics back to the client on a per-cell basis.
@@ -486,7 +486,7 @@ def m_notebook_document__did_close(self, notebookDocument=None, cellTextDocument
     def m_notebook_document__did_change(self, notebookDocument=None, change=None, **_kwargs):
         """
         Changes to the notebook document.
-        
+
         This could be one of the following:
         1. Notebook metadata changed
         2. Cell(s) added

pylsp/workspace.py

@@ -505,6 +505,7 @@ def add_cells(self, new_cells: List, start: int) -> None:
     def remove_cells(self, start: int, delete_count: int) -> None:
         del self.cells[start:start+delete_count]
 
+
 # We inherit from Document for now to get the same API. However, cell document differ from the text documents in that
 # they have a language id.
 class Cell(Document):

test/test_notebook_document.py

@@ -58,7 +58,7 @@ def client_server_pair():
     client_server_pair_obj.client._endpoint.notify("exit")
 
 
-def test_initialize(client_server_pair): # pylint: disable=redefined-outer-name
+def test_initialize(client_server_pair):  # pylint: disable=redefined-outer-name
     client, server = client_server_pair
     response = client._endpoint.request(
         "initialize",
@@ -73,7 +73,7 @@ def test_initialize(client_server_pair): # pylint: disable=redefined-outer-name
     # TODO: assert that notebook capabilities are in response
 
 
-def test_notebook_document__did_open(client_server_pair): # pylint: disable=redefined-outer-name
+def test_notebook_document__did_open(client_server_pair):  # pylint: disable=redefined-outer-name
     client, server = client_server_pair
     client._endpoint.request(
         "initialize",
@@ -157,7 +157,7 @@ def test_notebook_document__did_open(client_server_pair): # pylint: disable=rede
         mock_notify.assert_has_calls(expected_call_args)
 
 
-def test_notebook_document__did_change(client_server_pair): # pylint: disable=redefined-outer-name
+def test_notebook_document__did_change(client_server_pair):  # pylint: disable=redefined-outer-name
     client, server = client_server_pair
     client._endpoint.request(
         "initialize",
@@ -258,7 +258,7 @@ def test_notebook_document__did_change(client_server_pair): # pylint: disable=re
         )
         wait_for_condition(lambda: mock_notify.call_count >= 1)
         assert len(server.workspace.documents) == 2
-        assert not "cell_2_uri" in server.workspace.documents
+        assert "cell_2_uri" not in server.workspace.documents
         assert len(server.workspace.get_document("notebook_uri").cells) == 1
         expected_call_args = [
             call(
@@ -406,7 +406,7 @@ def test_notebook_document__did_change(client_server_pair): # pylint: disable=re
         mock_notify.assert_has_calls(expected_call_args)
 
 
-def test_notebook__did_close(client_server_pair):  # pylint: disable=redefined-outer-name
+def test_notebook__did_close(client_server_pair):   # pylint: disable=redefined-outer-name
     client, server = client_server_pair
     client._endpoint.request(
         "initialize",