Skip to content

Failure marker #5695

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
youknowone merged 6 commits into from
Apr 24, 2025
+137 −0
Merged

Failure marker #5695

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
91514b0
initial auto-upgrader
arihant2math Apr 14, 2025
0236917
platform support
arihant2math Apr 14, 2025
193a761
use ast walking
arihant2math Apr 14, 2025
5f402f2
detect testname automatically
arihant2math Apr 15, 2025
452e629
handle classes properly
arihant2math Apr 15, 2025
55802b2
add instructions to fix_test.py
arihant2math Apr 24, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
handle classes properly
  • Loading branch information
@arihant2math
arihant2math committed Apr 24, 2025
commit 452e62917b486e67f66a08e1e31d651464576578
34 changes: 33 additions & 1 deletion scripts/fix_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import argparse
import ast
import itertools
import platform
from pathlib import Path

Expand Down Expand Up @@ -71,6 +72,37 @@ def modify_test(file: str, test: list[str], for_platform: bool = False) -> str:
break
return "\n".join(lines)

def modify_test_v2(file: str, test: list[str], for_platform: bool = False) -> str:
a = ast.parse(file)
lines = file.splitlines()
fixture = "@unittest.expectedFailure"
for key, node in ast.iter_fields(a):
if key == "body":
for i, n in enumerate(node):
match n:
case ast.ClassDef():
if len(test) == 2 and test[0] == n.name:
# look through body for function def
for i, fn in enumerate(n.body):
match fn:
case ast.FunctionDef():
if fn.name == test[-1]:
assert not for_platform
indent = " " * fn.col_offset
lines.insert(fn.lineno - 1, indent + fixture)
lines.insert(fn.lineno - 1, indent + "# TODO: RUSTPYTHON")
break
case ast.FunctionDef():
if n.name == test[0] and len(test) == 1:
assert not for_platform
indent = " " * n.col_offset
lines.insert(n.lineno - 1, indent + fixture)
lines.insert(n.lineno - 1, indent + "# TODO: RUSTPYTHON")
break
if i > 500:
exit()
return "\n".join(lines)

def run_test(test_name):
print(f"Running test: {test_name}")
rustpython_location = "./target/release/rustpython"
Expand All @@ -87,7 +119,7 @@ def run_test(test_name):
for test in tests.tests:
if test.result == "fail" or test.result == "error":
print("Modifying test:", test.name)
f = modify_test(f, path_to_test(test.path), args.platform)
f = modify_test_v2(f, path_to_test(test.path), args.platform)
with open(args.path, "w") as file:
# TODO: Find validation method, and make --force override it
file.write(f)
Loading