pyodide/tools/pytest_wrapper.py

39 lines
953 B
Python
Executable File

#!/usr/bin/env python3
import subprocess
from typing import List
import sys
args = sys.argv[1:]
def remove_num_threads_option(args: List[str]) -> None:
"""Remove -n <n> from argument list"""
for i in range(0, len(args)):
if args[i] == "-n":
del args[i : i + 2]
break
if __name__ == "__main__":
try:
subprocess.run(["pytest"] + args, check=True)
sys.exit(0)
except subprocess.CalledProcessError:
pass
# Failed tests. Look up number of failed tests
with open(".pytest_cache/v/cache/lastfailed") as f:
num_failed = sum(1 for line in f) - 2
if num_failed > 9:
print("More than 9 tests failed. Not rerunning")
sys.exit(1)
print("Rerunnning failed tests sequentially")
remove_num_threads_option(args)
try:
subprocess.run(["pytest", "--lf"] + args, check=True)
except subprocess.CalledProcessError:
sys.exit(1)