averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

Consistent --output; Improved coverage: main()

Browse Source 
Coverage at 99% according to coverage.py

* `compare_perf_tests.py` now always outputs the same format to stdout as is written to `--output` file
* Added integration test for the main() function
* Added tests for console output (and suppressed it leaking during testing)
* Fixed file name in test’s file header
This commit is contained in:
Pavol Vaskovic
2017-06-04 18:31:06 +02:00
parent 9265a71ac6
commit dea7d8fe77
2 changed files with 127 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
benchmark/scripts/compare_perf_tests.py
View File

@@ -268,7 +268,7 @@ class ReportFormatter(object):
widths = self._column_widths()
def justify_columns(contents):
return tuple(map(lambda (w, c): c.ljust(w), zip(widths, contents)))
return tuple([c.ljust(w) for w, c in zip(widths, contents)])
def row(contents):
return ROW.format(*justify_columns(contents))
@@ -409,34 +409,18 @@ def main():
args.delta_threshold)
formatter = ReportFormatter(comparator, args.old_branch, args.new_branch,
args.changes_only)
formats = {
'markdown': formatter.markdown,
'git': formatter.git,
'html': formatter.html
}
if args.format:
if args.format.lower() != 'markdown':
print(formatter.git())
else:
print(formatter.markdown())
report = formats[args.format]()
print(report)
if args.format:
if args.format.lower() == 'html':
if args.output:
write_to_file(args.output, formatter.html())
else:
print('Error: missing --output flag.')
sys.exit(1)
elif args.format.lower() == 'markdown':
if args.output:
write_to_file(args.output, formatter.markdown())
elif args.format.lower() != 'git':
print('{0} is unknown format.'.format(args.format))
sys.exit(1)
def write_to_file(file_name, data):
"""
Write data to given file
"""
with open(file_name, 'w') as f:
f.write(data)
if args.output:
with open(args.output, 'w') as f:
f.write(report)
if __name__ == '__main__':