Fix nested markers

tthijm · Jun 12, 2024 · 8442a12 · 8442a12
1 parent ef468d5
commit 8442a12
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/cov/cov.py b/cov/cov.py
@@ -29,13 +29,13 @@ def decorator(func: Callable) -> Callable:
     return decorator
 
 
-def mark(branch_id: int, func_name = None) -> None:
-    if func_name is None:
-        func_name = inspect.stack()[1].function
+def mark(branch_id: int) -> None:
+    for frame in inspect.stack():
+        if frame.function in coverages:
+            coverage = coverages[frame.function]
 
-    coverage = coverages[func_name]
-
-    coverage.covered_branches[branch_id] = True
+            coverage.covered_branches[branch_id] = True
+            break
 
 
 def print_table() -> None: