Fix update_mir_test_checks.py to run on python3
Split off from D58817
Differential Revision: https://reviews.llvm.org/D58820
llvm-svn: 355268
diff --git a/llvm/utils/update_mir_test_checks.py b/llvm/utils/update_mir_test_checks.py
index 56d236d..cf7528f 100755
--- a/llvm/utils/update_mir_test_checks.py
+++ b/llvm/utils/update_mir_test_checks.py
@@ -62,6 +62,8 @@
with open(ir) as ir_file:
stdout = subprocess.check_output('{} {}'.format(self.bin, args),
shell=True, stdin=ir_file)
+ if sys.version_info[0] > 2:
+ stdout = stdout.decode()
# Fix line endings to unix CR style.
stdout = stdout.replace('\r\n', '\n')
return stdout
@@ -408,7 +410,7 @@
log('Writing {} lines to {}...'.format(len(output_lines), test), verbose)
with open(test, 'wb') as fd:
- fd.writelines([l + '\n' for l in output_lines])
+ fd.writelines(['{}\n'.format(l).encode('utf-8') for l in output_lines])
def main():