mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 14:04:26 +08:00
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
import os
|
|
|
|
from . import base
|
|
from . import run
|
|
from . import refactor
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
parser.addoption(
|
|
"--integration-case-dir",
|
|
default=os.path.join(base.test_dir, 'completion'),
|
|
help="Directory in which integration test case files locate.")
|
|
parser.addoption(
|
|
"--refactor-case-dir",
|
|
default=os.path.join(base.test_dir, 'refactor'),
|
|
help="Directory in which refactoring test case files locate.")
|
|
parser.addoption(
|
|
"--test-files", "-T", default=[], action='append',
|
|
help=(
|
|
"Specify test files using FILE_NAME[:LINE[,LINE[,...]]]. "
|
|
"For example: -T generators.py:10,13,19. "
|
|
"Note that you can use -m to specify the test case by id."))
|
|
parser.addoption(
|
|
"--thirdparty",
|
|
help="Include integration tests that requires third party modules.")
|
|
|
|
|
|
def parse_test_files_option(opt):
|
|
"""
|
|
Parse option passed to --test-files into a key-value pair.
|
|
|
|
>>> parse_test_files_option('generators.py:10,13,19')
|
|
('generators.py', [10, 13, 19])
|
|
"""
|
|
opt = str(opt)
|
|
if ':' in opt:
|
|
(f_name, rest) = opt.split(':', 1)
|
|
return (f_name, list(map(int, rest.split(','))))
|
|
else:
|
|
return (opt, [])
|
|
|
|
|
|
def pytest_generate_tests(metafunc):
|
|
"""
|
|
:type metafunc: _pytest.python.Metafunc
|
|
"""
|
|
test_files = dict(map(parse_test_files_option,
|
|
metafunc.config.option.test_files))
|
|
if 'case' in metafunc.fixturenames:
|
|
base_dir = metafunc.config.option.integration_case_dir
|
|
thirdparty = metafunc.config.option.thirdparty
|
|
metafunc.parametrize(
|
|
'case',
|
|
run.collect_dir_tests(base_dir, test_files, thirdparty))
|
|
if 'refactor_case' in metafunc.fixturenames:
|
|
base_dir = metafunc.config.option.refactor_case_dir
|
|
metafunc.parametrize(
|
|
'refactor_case',
|
|
refactor.collect_dir_tests(base_dir, test_files))
|