Merge branch 'release/1.0.1' into lint/organize-imports

Author: Splines

src/pythonParsing.ts

@@ -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

@@ -9,7 +9,7 @@ describe("Manim Cell Ranges", function () {
   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