test: update expected context values for statement-type helpers

Author: KRRT7

tests/test_code_context_extractor.py

@@ -978,7 +978,12 @@ def test_repo_helper() -> None:
     code_ctx = get_code_optimization_context(function_to_optimize, project_root)
     read_write_context, read_only_context = code_ctx.read_writable_code, code_ctx.read_only_context_code
     hashing_context = code_ctx.hashing_code_context
+    path_to_globals = project_root / "globals.py"
     expected_read_write_context = f"""
+```python:{path_to_globals.relative_to(project_root)}
+# Define a global variable
+API_URL = "https://api.example.com/data"
+```
 ```python:{path_to_utils.relative_to(project_root)}
 import math
 
@@ -1071,7 +1076,12 @@ def test_repo_helper_of_helper() -> None:
     code_ctx = get_code_optimization_context(function_to_optimize, project_root)
     read_write_context, read_only_context = code_ctx.read_writable_code, code_ctx.read_only_context_code
     hashing_context = code_ctx.hashing_code_context
+    path_to_globals = project_root / "globals.py"
     expected_read_write_context = f"""
+```python:{path_to_globals.relative_to(project_root)}
+# Define a global variable
+API_URL = "https://api.example.com/data"
+```
 ```python:{path_to_utils.relative_to(project_root)}
 import math
 from transform_utils import DataTransformer
@@ -1798,6 +1808,8 @@ def calculate(self, operation, x, y):
 """
         expected_read_only_context = """
 ```python:utility_module.py
+import sys
+
 DEFAULT_PRECISION = "medium"
 
 # Try-except block with variable definitions
@@ -1808,6 +1820,17 @@ def calculate(self, operation, x, y):
     # Used variable in except block
     CALCULATION_BACKEND = "python"
 
+# Nested if-else with variable definitions
+if sys.platform.startswith('win'):
+    # Used variable in outer if
+    SYSTEM_TYPE = "windows"
+elif sys.platform.startswith('linux'):
+    # Used variable in outer elif
+    SYSTEM_TYPE = "linux"
+else:
+    # Used variable in outer else
+    SYSTEM_TYPE = "other"
+
 # Function that will be used in the main code
 def select_precision(precision, fallback_precision):
     if precision is None:
@@ -2014,6 +2037,8 @@ def get_system_details():
         relative_path = file_path.relative_to(project_root)
         expected_read_write_context = f"""
 ```python:utility_module.py
+import sys
+
 DEFAULT_PRECISION = "medium"
 
 # Try-except block with variable definitions
@@ -2024,6 +2049,17 @@ def get_system_details():
     # Used variable in except block
     CALCULATION_BACKEND = "python"
 
+# Nested if-else with variable definitions
+if sys.platform.startswith('win'):
+    # Used variable in outer if
+    SYSTEM_TYPE = "windows"
+elif sys.platform.startswith('linux'):
+    # Used variable in outer elif
+    SYSTEM_TYPE = "linux"
+else:
+    # Used variable in outer else
+    SYSTEM_TYPE = "other"
+
 # Function that will be used in the main code
 def select_precision(precision, fallback_precision):
     if precision is None:
@@ -2064,6 +2100,8 @@ def __init__(self, precision="high", fallback_precision=None, mode="standard"):
 """
         expected_read_only_context = """
 ```python:utility_module.py
+import sys
+
 DEFAULT_PRECISION = "medium"
 
 # Try-except block with variable definitions
@@ -2073,6 +2111,17 @@ def __init__(self, precision="high", fallback_precision=None, mode="standard"):
 except ImportError:
     # Used variable in except block
     CALCULATION_BACKEND = "python"
+
+# Nested if-else with variable definitions
+if sys.platform.startswith('win'):
+    # Used variable in outer if
+    SYSTEM_TYPE = "windows"
+elif sys.platform.startswith('linux'):
+    # Used variable in outer elif
+    SYSTEM_TYPE = "linux"
+else:
+    # Used variable in outer else
+    SYSTEM_TYPE = "other"
 ```
 """
         assert read_write_context.markdown.strip() == expected_read_write_context.strip()