1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
"""
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 functools import partial, wraps
from jedi import Project
from pathlib import Path
test_dir = Path(__file__).absolute().parent
test_dir_project = Project(test_dir)
root_dir = test_dir.parent
example_dir = test_dir.joinpath('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 example_dir.joinpath(*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 = test_dir.parent
oldcwd = Path.cwd()
os.chdir(repo_root.joinpath(path))
try:
yield
finally:
os.chdir(oldcwd)
|
