Merge branch 'main' of https://github.com/approvals/ApprovalTests.Python

approvals · Nov 26, 2023 · 9c1d052 · 9c1d052
2 parents e788ea2 + abbd3ac
commit 9c1d052
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/approvaltests/inline/split_code.py b/approvaltests/inline/split_code.py
@@ -17,11 +17,13 @@ def on_method(code, method_name) -> "SplitCode":
         before = []
         after = []
         tab = "    "
+
         class State(Enum):
             BEFORE = 0
             FIRST_LINE_OF_METHOD_BODY = 1
             IN_DOCSTRING = 2
             AFTER_DOCTSTRING = 3
+
         state = State.BEFORE
 
         for line in lines:
@@ -32,7 +34,7 @@ class State(Enum):
             elif state == State.BEFORE:
                 before.append(line)
             elif state == State.FIRST_LINE_OF_METHOD_BODY:
-                tab = re.compile(r'^\s*').match(line).group()
+                tab = re.compile(r"^\s*").match(line).group()
                 if stripped_line.startswith('"""'):
                     state = State.IN_DOCSTRING
                 else: