100XZX001 commited on
Commit
29a5930
·
verified ·
1 Parent(s): 667b6db

Update test_runner.py

Browse files
Files changed (1) hide show
  1. 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
- """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,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