forked from OSchip/llvm-project
106 lines
4.0 KiB
Python
Executable File
106 lines
4.0 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)
|