mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-28 16:26:05 +00:00
b71edfaa4e
This is the first commit in a series that will reformat all the python files in the LLVM repository. Reformatting is done with `black`. See more information here: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Reviewed By: jhenderson, JDevlieghere, MatzeB Differential Revision: https://reviews.llvm.org/D150545
103 lines
3.8 KiB
Python
Executable File
103 lines
3.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# Automatically formatted with yapf (https://github.com/google/yapf)
|
|
|
|
import subprocess
|
|
import unittest
|
|
|
|
|
|
def getFinalPasses(run):
|
|
stdout = run.stdout.decode()
|
|
stdout = stdout[: stdout.rfind("\n")]
|
|
stdout = stdout[stdout.rfind("\n") + 1 :]
|
|
return stdout
|
|
|
|
|
|
class Test(unittest.TestCase):
|
|
def test_0(self):
|
|
"""Test all passes are removed except those required to crash. Verify
|
|
that PM structure is intact."""
|
|
run_args = [
|
|
"./utils/reduce_pipeline.py",
|
|
"--opt-binary=./utils/reduce_pipeline_test/fake_opt.py",
|
|
"--input=/dev/null",
|
|
"--passes=a,b,c,A(d,B(e,f),g),h,i",
|
|
"-crash-seq=b,d,f",
|
|
]
|
|
run = subprocess.run(run_args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
self.assertEqual(run.returncode, 0)
|
|
self.assertEqual(getFinalPasses(run), '-passes="b,A(d,B(f))"')
|
|
|
|
def test_1(self):
|
|
"""Test all passes are removed except those required to crash. The
|
|
required passes in this case are the first and last in that order
|
|
(a bit of a corner-case for the reduction algorithm)."""
|
|
run_args = [
|
|
"./utils/reduce_pipeline.py",
|
|
"--opt-binary=./utils/reduce_pipeline_test/fake_opt.py",
|
|
"--input=/dev/null",
|
|
"--passes=a,b,c,A(d,B(e,f),g),h,i",
|
|
"-crash-seq=a,i",
|
|
]
|
|
run = subprocess.run(run_args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
self.assertEqual(run.returncode, 0)
|
|
self.assertEqual(getFinalPasses(run), '-passes="a,i"')
|
|
|
|
def test_2_0(self):
|
|
"""Test expansion of EXPAND_a_to_f (expands into 'a,b,c,d,e,f')."""
|
|
run_args = [
|
|
"./utils/reduce_pipeline.py",
|
|
"--opt-binary=./utils/reduce_pipeline_test/fake_opt.py",
|
|
"--input=/dev/null",
|
|
"--passes=EXPAND_a_to_f",
|
|
"-crash-seq=b,e",
|
|
]
|
|
run = subprocess.run(run_args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
self.assertEqual(run.returncode, 0)
|
|
self.assertEqual(getFinalPasses(run), '-passes="b,e"')
|
|
|
|
def test_2_1(self):
|
|
"""Test EXPAND_a_to_f and the '--dont-expand-passes' option."""
|
|
run_args = [
|
|
"./utils/reduce_pipeline.py",
|
|
"--opt-binary=./utils/reduce_pipeline_test/fake_opt.py",
|
|
"--input=/dev/null",
|
|
"--passes=EXPAND_a_to_f",
|
|
"-crash-seq=EXPAND_a_to_f",
|
|
"--dont-expand-passes",
|
|
]
|
|
run = subprocess.run(run_args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
self.assertEqual(run.returncode, 0)
|
|
self.assertEqual(getFinalPasses(run), '-passes="EXPAND_a_to_f"')
|
|
|
|
def test_3(self):
|
|
"""Test that empty pass-managers get removed by default."""
|
|
run_args = [
|
|
"./utils/reduce_pipeline.py",
|
|
"--opt-binary=./utils/reduce_pipeline_test/fake_opt.py",
|
|
"--input=/dev/null",
|
|
"--passes=a,b,c,A(d,B(e,f),g),h,i",
|
|
"-crash-seq=b,d,h",
|
|
]
|
|
run = subprocess.run(run_args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
self.assertEqual(run.returncode, 0)
|
|
self.assertEqual(getFinalPasses(run), '-passes="b,A(d),h"')
|
|
|
|
def test_4(self):
|
|
"""Test the '--dont-remove-empty-pm' option."""
|
|
run_args = [
|
|
"./utils/reduce_pipeline.py",
|
|
"--opt-binary=./utils/reduce_pipeline_test/fake_opt.py",
|
|
"--input=/dev/null",
|
|
"--passes=a,b,c,A(d,B(e,f),g),h,i",
|
|
"-crash-seq=b,d,h",
|
|
"--dont-remove-empty-pm",
|
|
]
|
|
run = subprocess.run(run_args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
self.assertEqual(run.returncode, 0)
|
|
self.assertEqual(getFinalPasses(run), '-passes="b,A(d,B()),h"')
|
|
|
|
|
|
unittest.main()
|
|
exit(0)
|