Fix async function display (#16443)

* updated regex * added tests
jupyterlab · Jun 6, 2024 · eb1eca6 · eb1eca6
1 parent 2c718c1
commit eb1eca6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/packages/fileeditor/src/toc/python.ts b/packages/fileeditor/src/toc/python.ts
@@ -17,9 +17,9 @@ try {
   // https://github.com/tc39/proposal-regexp-match-indices was accepted
   // in May 2021 (https://github.com/tc39/proposals/blob/main/finished-proposals.md)
   // So we will fallback to the polyfill regexp-match-indices if not available
-  KEYWORDS = new RegExp('^\\s*(class |def |from |import )', 'd');
+  KEYWORDS = new RegExp('^\\s*(class |def |async def |from |import )', 'd');
 } catch {
-  KEYWORDS = new RegExp('^\\s*(class |def |from |import )');
+  KEYWORDS = new RegExp('^\\s*(class |def |async def |from |import )');
 }
 
 /**

diff --git a/packages/fileeditor/test/python.spec.ts b/packages/fileeditor/test/python.spec.ts
@@ -25,6 +25,16 @@ describe('@jupyterlab/fileeditor', () => {
             }
           ]
         ],
+        [
+          'async def async_function():',
+          [
+            {
+              text: 'async def async_function():',
+              level: 1,
+              line: 0
+            }
+          ]
+        ],
         [
           'class Klass:',
           [