mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 14:04:26 +08:00
57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
"""
|
|
A helper module for testing, improves compatibility for testing (as
|
|
``jedi._compatibility``) as well as introducing helper functions.
|
|
"""
|
|
|
|
from contextlib import contextmanager
|
|
|
|
import os
|
|
import pytest
|
|
from os.path import abspath, dirname, join
|
|
from functools import partial, wraps
|
|
from jedi import Project
|
|
|
|
test_dir = dirname(abspath(__file__))
|
|
test_dir_project = Project(test_dir)
|
|
root_dir = dirname(test_dir)
|
|
example_dir = join(test_dir, 'examples')
|
|
|
|
sample_int = 1 # This is used in completion/imports.py
|
|
|
|
skip_if_windows = partial(pytest.param,
|
|
marks=pytest.mark.skipif("sys.platform=='win32'"))
|
|
skip_if_not_windows = partial(pytest.param,
|
|
marks=pytest.mark.skipif("sys.platform!='win32'"))
|
|
|
|
|
|
def get_example_dir(*names):
|
|
return join(example_dir, *names)
|
|
|
|
|
|
def cwd_at(path):
|
|
"""
|
|
Decorator to run function at `path`.
|
|
|
|
:type path: str
|
|
:arg path: relative path from repository root (e.g., ``'jedi'``).
|
|
"""
|
|
def decorator(func):
|
|
@wraps(func)
|
|
def wrapper(Script, **kwargs):
|
|
with set_cwd(path):
|
|
return func(Script, **kwargs)
|
|
return wrapper
|
|
return decorator
|
|
|
|
|
|
@contextmanager
|
|
def set_cwd(path, absolute_path=False):
|
|
repo_root = os.path.dirname(test_dir)
|
|
|
|
oldcwd = os.getcwd()
|
|
os.chdir(os.path.join(repo_root, path))
|
|
try:
|
|
yield
|
|
finally:
|
|
os.chdir(oldcwd)
|