Current File : //lib/python3/dist-packages/mypyc/test/test_commandline.py
"""Test cases for invoking mypyc on the command line.
These are slow -- do not add test cases unless you have a very good reason to do so.
"""
import glob
import os
import os.path
import re
import subprocess
import sys
from mypy.test.data import DataDrivenTestCase
from mypy.test.config import test_temp_dir
from mypy.test.helpers import normalize_error_messages
from mypyc.test.testutil import MypycDataSuite, assert_test_output
files = [
'commandline.test',
]
base_path = os.path.join(os.path.dirname(__file__), '..', '..')
python3_path = sys.executable
class TestCommandLine(MypycDataSuite):
files = files
base_path = test_temp_dir
optional_out = True
def run_case(self, testcase: DataDrivenTestCase) -> None:
# Parse options from test case description (arguments must not have spaces)
text = '\n'.join(testcase.input)
m = re.search(r'# *cmd: *(.*)', text)
assert m is not None, 'Test case missing "# cmd: <files>" section'
args = m.group(1).split()
# Write main program to run (not compiled)
program = '_%s.py' % testcase.name
program_path = os.path.join(test_temp_dir, program)
with open(program_path, 'w') as f:
f.write(text)
out = b''
try:
# Compile program
cmd = subprocess.run([sys.executable, '-m', 'mypyc', *args],
stdout=subprocess.PIPE, stderr=subprocess.STDOUT, cwd='tmp')
if 'ErrorOutput' in testcase.name or cmd.returncode != 0:
out += cmd.stdout
if cmd.returncode == 0:
# Run main program
out += subprocess.check_output(
[python3_path, program],
cwd='tmp')
finally:
suffix = 'pyd' if sys.platform == 'win32' else 'so'
so_paths = glob.glob('tmp/**/*.{}'.format(suffix), recursive=True)
for path in so_paths:
os.remove(path)
# Strip out 'tmp/' from error message paths in the testcase output,
# due to a mismatch between this test and mypy's test suite.
expected = [x.replace('tmp/', '') for x in testcase.output]
# Verify output
actual = normalize_error_messages(out.decode().splitlines())
assert_test_output(testcase, actual, 'Invalid output', expected=expected)
Mini Shell