Update test_runner.py
Browse files- test_runner.py +4 -4
test_runner.py
CHANGED
|
@@ -69,9 +69,9 @@ class TestRunner:
|
|
| 69 |
pass
|
| 70 |
|
| 71 |
def _get_defined_function_name(self, code: str) -> Optional[str]:
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
try:
|
| 76 |
tree = ast.parse(code)
|
| 77 |
first_func = None
|
|
@@ -178,4 +178,4 @@ class TestRunner:
|
|
| 178 |
a = random.randint(-100, 100)
|
| 179 |
b = random.randint(-100, 100)
|
| 180 |
cases.append(([a, b], a + b))
|
| 181 |
-
return cases
|
|
|
|
| 69 |
pass
|
| 70 |
|
| 71 |
def _get_defined_function_name(self, code: str) -> Optional[str]:
|
| 72 |
+
"""Extract the target function name from the code.
|
| 73 |
+
Looks for a function named 'fix' first; otherwise returns the first function found.
|
| 74 |
+
"""
|
| 75 |
try:
|
| 76 |
tree = ast.parse(code)
|
| 77 |
first_func = None
|
|
|
|
| 178 |
a = random.randint(-100, 100)
|
| 179 |
b = random.randint(-100, 100)
|
| 180 |
cases.append(([a, b], a + b))
|
| 181 |
+
return cases
|