Allow for type hinting in construct() method

src/pythonParsing.ts

@@ -1,5 +1,5 @@
 import * as crypto from "crypto";
-import { TextDocument, Range } from "vscode";
+import { Range, TextDocument } from "vscode";
 
 /**
  * Cache is a simple key-value store that keeps a maximum number of entries.
@@ -206,7 +206,7 @@ export class ManimClass {
   /**
    * Regular expression to match the construct() method definition.
    */
-  private static CONSTRUCT_METHOD_REGEX = /^\s*def\s+construct\s*\(self\)\s*:/;
+  private static CONSTRUCT_METHOD_REGEX = /^\s*def\s+construct\s*\(self\)\s*(->\s*None)?\s*:/;
 
   /**
    * The 0-based line number where the Manim Class is defined.

tests/cellRanges.test.ts

@@ -1,15 +1,15 @@
-import { workspace, Range } from "vscode";
 import { describe, it } from "mocha";
-import { uriInWorkspace } from "./utils/testRunner";
+import { Range, workspace } from "vscode";
 import { ManimCellRanges } from "../src/pythonParsing";
+import { uriInWorkspace } from "./utils/testRunner";
 
 describe("Manim Cell Ranges", function () {
   // in the expected ranges we only care about the start and end lines
   // line numbers are 0-based here
   const tests = [
     {
       filename: "detection_basic.py",
-      expectedRanges: [[5, 7], [9, 10]],
+      expectedRanges: [[5, 7], [9, 10], [16, 18]],
     },
     {
       filename: "detection_class_definition.py",

tests/fixtures/detection_basic.py

@@ -11,6 +11,14 @@ def construct(self):
         print("And even more code")
 
 
+class BasicNotebookWithType(Scene):
+
+    def construct(self) -> None:
+        ## Cell inside construct(self) marked with "None" type
+        print("With some code None")
+        print("With some more code None")
+
+
 class NoManimScene(Scene):
     def constructtttt(self):
         ## Should not be detected as Manim Cell