mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 22:14:27 +08:00
306 lines
8.0 KiB
Python
306 lines
8.0 KiB
Python
"""
|
|
Testing of docstring related issues and especially ``jedi.docstrings``.
|
|
"""
|
|
|
|
from textwrap import dedent
|
|
import jedi
|
|
import pytest
|
|
from ..helpers import unittest
|
|
|
|
try:
|
|
import numpydoc # NOQA
|
|
except ImportError:
|
|
numpydoc_unavailable = True
|
|
else:
|
|
numpydoc_unavailable = False
|
|
|
|
try:
|
|
import numpy
|
|
except ImportError:
|
|
numpy_unavailable = True
|
|
else:
|
|
numpy_unavailable = False
|
|
|
|
|
|
class TestDocstring(unittest.TestCase):
|
|
def test_function_doc(self):
|
|
defs = jedi.Script("""
|
|
def func():
|
|
'''Docstring of `func`.'''
|
|
func""").goto_definitions()
|
|
self.assertEqual(defs[0].docstring(), 'func()\n\nDocstring of `func`.')
|
|
|
|
def test_class_doc(self):
|
|
defs = jedi.Script("""
|
|
class TestClass():
|
|
'''Docstring of `TestClass`.'''
|
|
TestClass""").goto_definitions()
|
|
self.assertEqual(defs[0].docstring(), 'Docstring of `TestClass`.')
|
|
|
|
def test_instance_doc(self):
|
|
defs = jedi.Script("""
|
|
class TestClass():
|
|
'''Docstring of `TestClass`.'''
|
|
tc = TestClass()
|
|
tc""").goto_definitions()
|
|
self.assertEqual(defs[0].docstring(), 'Docstring of `TestClass`.')
|
|
|
|
@unittest.skip('need evaluator class for that')
|
|
def test_attribute_docstring(self):
|
|
defs = jedi.Script("""
|
|
x = None
|
|
'''Docstring of `x`.'''
|
|
x""").goto_definitions()
|
|
self.assertEqual(defs[0].docstring(), 'Docstring of `x`.')
|
|
|
|
@unittest.skip('need evaluator class for that')
|
|
def test_multiple_docstrings(self):
|
|
defs = jedi.Script("""
|
|
def func():
|
|
'''Original docstring.'''
|
|
x = func
|
|
'''Docstring of `x`.'''
|
|
x""").goto_definitions()
|
|
docs = [d.docstring() for d in defs]
|
|
self.assertEqual(docs, ['Original docstring.', 'Docstring of `x`.'])
|
|
|
|
def test_completion(self):
|
|
assert jedi.Script('''
|
|
class DocstringCompletion():
|
|
#? []
|
|
""" asdfas """''').completions()
|
|
|
|
def test_docstrings_type_dotted_import(self):
|
|
s = """
|
|
def func(arg):
|
|
'''
|
|
:type arg: random.Random
|
|
'''
|
|
arg."""
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
assert 'seed' in names
|
|
|
|
def test_docstrings_param_type(self):
|
|
s = """
|
|
def func(arg):
|
|
'''
|
|
:param str arg: some description
|
|
'''
|
|
arg."""
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
assert 'join' in names
|
|
|
|
def test_docstrings_type_str(self):
|
|
s = """
|
|
def func(arg):
|
|
'''
|
|
:type arg: str
|
|
'''
|
|
arg."""
|
|
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
assert 'join' in names
|
|
|
|
def test_docstring_instance(self):
|
|
# The types hint that it's a certain kind
|
|
s = dedent("""
|
|
class A:
|
|
def __init__(self,a):
|
|
'''
|
|
:type a: threading.Thread
|
|
'''
|
|
|
|
if a is not None:
|
|
a.start()
|
|
|
|
self.a = a
|
|
|
|
|
|
def method_b(c):
|
|
'''
|
|
:type c: A
|
|
'''
|
|
|
|
c.""")
|
|
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
assert 'a' in names
|
|
assert '__init__' in names
|
|
assert 'mro' not in names # Exists only for types.
|
|
|
|
def test_docstring_keyword(self):
|
|
completions = jedi.Script('assert').completions()
|
|
self.assertIn('assert', completions[0].docstring())
|
|
|
|
# ---- Numpy Style Tests ---
|
|
|
|
@pytest.mark.skipif(numpydoc_unavailable,
|
|
reason='numpydoc module is unavailable')
|
|
def test_numpydoc_parameters():
|
|
s = dedent('''
|
|
def foobar(x, y):
|
|
"""
|
|
Parameters
|
|
----------
|
|
x : int
|
|
y : str
|
|
"""
|
|
y.''')
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
assert 'isupper' in names
|
|
assert 'capitalize' in names
|
|
|
|
@pytest.mark.skipif(numpydoc_unavailable,
|
|
reason='numpydoc module is unavailable')
|
|
def test_numpydoc_parameters_set_of_values():
|
|
s = dedent('''
|
|
def foobar(x, y):
|
|
"""
|
|
Parameters
|
|
----------
|
|
x : {'foo', 'bar', 100500}, optional
|
|
"""
|
|
x.''')
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
assert 'isupper' in names
|
|
assert 'capitalize' in names
|
|
assert 'numerator' in names
|
|
|
|
@pytest.mark.skipif(numpydoc_unavailable,
|
|
reason='numpydoc module is unavailable')
|
|
def test_numpydoc_parameters_alternative_types():
|
|
s = dedent('''
|
|
def foobar(x, y):
|
|
"""
|
|
Parameters
|
|
----------
|
|
x : int or str or list
|
|
"""
|
|
x.''')
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
assert 'isupper' in names
|
|
assert 'capitalize' in names
|
|
assert 'numerator' in names
|
|
assert 'append' in names
|
|
|
|
def test_numpydoc_returns():
|
|
s = dedent('''
|
|
def foobar():
|
|
"""
|
|
Returns
|
|
----------
|
|
x : int
|
|
y : str
|
|
"""
|
|
return x
|
|
|
|
def bazbiz():
|
|
z = foobar()
|
|
z.''')
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
assert 'isupper' in names
|
|
assert 'capitalize' in names
|
|
assert 'numerator' in names
|
|
|
|
@pytest.mark.skipif(numpydoc_unavailable,
|
|
reason='numpydoc module is unavailable')
|
|
def test_numpydoc_returns_set_of_values():
|
|
s = dedent('''
|
|
def foobar():
|
|
"""
|
|
Returns
|
|
----------
|
|
x : {'foo', 'bar', 100500}
|
|
"""
|
|
return x
|
|
|
|
def bazbiz():
|
|
z = foobar()
|
|
z.''')
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
assert 'isupper' in names
|
|
assert 'capitalize' in names
|
|
assert 'numerator' in names
|
|
|
|
@pytest.mark.skipif(numpydoc_unavailable,
|
|
reason='numpydoc module is unavailable')
|
|
def test_numpydoc_returns_alternative_types():
|
|
s = dedent('''
|
|
def foobar():
|
|
"""
|
|
Returns
|
|
----------
|
|
int or list of str
|
|
"""
|
|
return x
|
|
|
|
def bazbiz():
|
|
z = foobar()
|
|
z.''')
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
assert 'isupper' not in names
|
|
assert 'capitalize' not in names
|
|
assert 'numerator' in names
|
|
assert 'append' in names
|
|
|
|
def test_numpydoc_returns_list_of():
|
|
s = dedent('''
|
|
def foobar():
|
|
"""
|
|
Returns
|
|
----------
|
|
list of str
|
|
"""
|
|
return x
|
|
|
|
def bazbiz():
|
|
z = foobar()
|
|
z.''')
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
assert 'append' in names
|
|
assert 'isupper' not in names
|
|
assert 'capitalize' not in names
|
|
|
|
@pytest.mark.skipif(numpydoc_unavailable,
|
|
reason='numpydoc module is unavailable')
|
|
def test_numpydoc_returns_obj():
|
|
s = dedent('''
|
|
def foobar(x, y):
|
|
"""
|
|
Returns
|
|
----------
|
|
int or random.Random
|
|
"""
|
|
return x + y
|
|
|
|
def bazbiz():
|
|
z = foobar(x, y)
|
|
z.''')
|
|
script = jedi.Script(s)
|
|
names = [c.name for c in script.completions()]
|
|
assert 'numerator' in names
|
|
assert 'seed' in names
|
|
|
|
@pytest.mark.skipif(numpydoc_unavailable or numpy_unavailable,
|
|
reason='numpydoc or numpy module is unavailable')
|
|
def test_numpy_returns():
|
|
s = dedent('''
|
|
import numpy
|
|
x = numpy.asarray([])
|
|
x.d''')
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
print(names)
|
|
assert 'diagonal' in names
|
|
|
|
@pytest.mark.skipif(numpydoc_unavailable or numpy_unavailable,
|
|
reason='numpydoc or numpy module is unavailable')
|
|
def test_numpy_comp_returns():
|
|
s = dedent('''
|
|
import numpy
|
|
x = numpy.array([])
|
|
x.d''')
|
|
names = [c.name for c in jedi.Script(s).completions()]
|
|
print(names)
|
|
assert 'diagonal' in names
|
|
|