mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 22:14:27 +08:00
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
import tempfile
|
|
import shutil
|
|
|
|
import jedi
|
|
|
|
|
|
collect_ignore = ["setup.py"]
|
|
|
|
|
|
# The following hooks (pytest_configure, pytest_unconfigure) are used
|
|
# to modify `jedi.settings.cache_directory` because `clean_jedi_cache`
|
|
# has no effect during doctests. Without these hooks, doctests uses
|
|
# user's cache (e.g., ~/.cache/jedi/). We should remove this
|
|
# workaround once the problem is fixed in py.test.
|
|
#
|
|
# See:
|
|
# - https://github.com/davidhalter/jedi/pull/168
|
|
# - https://bitbucket.org/hpk42/pytest/issue/275/
|
|
|
|
jedi_cache_directory_orig = None
|
|
jedi_cache_directory_temp = None
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
parser.addoption("--jedi-debug", "-D", action='store_true',
|
|
help="Enables Jedi's debug output.")
|
|
|
|
parser.addoption("--warning-is-error", action='store_true',
|
|
help="Warnings are treated as errors.")
|
|
|
|
|
|
def pytest_configure(config):
|
|
global jedi_cache_directory_orig, jedi_cache_directory_temp
|
|
jedi_cache_directory_orig = jedi.settings.cache_directory
|
|
jedi_cache_directory_temp = tempfile.mkdtemp(prefix='jedi-test-')
|
|
jedi.settings.cache_directory = jedi_cache_directory_temp
|
|
|
|
if config.option.jedi_debug:
|
|
jedi.set_debug_function()
|
|
|
|
if config.option.warning_is_error:
|
|
import warnings
|
|
warnings.simplefilter("error")
|
|
|
|
|
|
def pytest_unconfigure(config):
|
|
global jedi_cache_directory_orig, jedi_cache_directory_temp
|
|
jedi.settings.cache_directory = jedi_cache_directory_orig
|
|
shutil.rmtree(jedi_cache_directory_temp)
|