Tests: minor updates to benchmark script for running on buildbot
* graph command accepts folder of json files as input * reset command clears log files
This commit is contained in:
parent
731926e70e
commit
fc36772b06
|
@ -4,6 +4,7 @@
|
||||||
import api
|
import api
|
||||||
import argparse
|
import argparse
|
||||||
import fnmatch
|
import fnmatch
|
||||||
|
import glob
|
||||||
import pathlib
|
import pathlib
|
||||||
import shutil
|
import shutil
|
||||||
import sys
|
import sys
|
||||||
|
@ -228,6 +229,9 @@ def cmd_reset(env: api.TestEnvironment, argv: List):
|
||||||
|
|
||||||
config.queue.write()
|
config.queue.write()
|
||||||
|
|
||||||
|
if args.test == '*':
|
||||||
|
shutil.rmtree(config.logs_dir)
|
||||||
|
|
||||||
def cmd_run(env: api.TestEnvironment, argv: List, update_only: bool):
|
def cmd_run(env: api.TestEnvironment, argv: List, update_only: bool):
|
||||||
# Run tests.
|
# Run tests.
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
|
@ -274,7 +278,17 @@ def cmd_graph(argv: List):
|
||||||
parser.add_argument('-o', '--output', type=str, required=True)
|
parser.add_argument('-o', '--output', type=str, required=True)
|
||||||
args = parser.parse_args(argv)
|
args = parser.parse_args(argv)
|
||||||
|
|
||||||
graph = api.TestGraph([pathlib.Path(path) for path in args.json_file])
|
# For directories, use all json files in the directory.
|
||||||
|
json_files = []
|
||||||
|
for path in args.json_file:
|
||||||
|
path = pathlib.Path(path)
|
||||||
|
if path.is_dir():
|
||||||
|
for filepath in glob.iglob(str(path / '*.json')):
|
||||||
|
json_files.append(pathlib.Path(filepath))
|
||||||
|
else:
|
||||||
|
json_files.append(path)
|
||||||
|
|
||||||
|
graph = api.TestGraph(json_files)
|
||||||
graph.write(pathlib.Path(args.output))
|
graph.write(pathlib.Path(args.output))
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
|
|
Loading…
Reference in New Issue