VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-06 22:14:27 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Cleanup the docstring tests

Browse Source
This commit is contained in:
Dave Halter
2017-12-29 19:47:28 +01:00
parent bf73fcbed4
commit c7266d65c1
1 changed files with 136 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

88
test/test_evaluate/test_docstring.py
View File

@@ -15,82 +15,89 @@ else:
numpydoc_unavailable = False
try:
import numpy
import numpy # NOQA
except ImportError:
numpy_unavailable = True
else:
numpy_unavailable = False
class TestDocstring(unittest.TestCase):
def test_function_doc(self):
defs = jedi.Script("""
def test_function_doc(Script):
defs = Script("""
def func():
'''Docstring of `func`.'''
func""").goto_definitions()
self.assertEqual(defs[0].docstring(), 'func()\n\nDocstring of `func`.')
assert defs[0].docstring() == 'func()\n\nDocstring of `func`.'
def test_class_doc(self):
defs = jedi.Script("""
def test_class_doc(Script):
defs = Script("""
class TestClass():
'''Docstring of `TestClass`.'''
TestClass""").goto_definitions()
self.assertEqual(defs[0].docstring(), 'Docstring of `TestClass`.')
assert defs[0].docstring() == 'Docstring of `TestClass`.'
def test_instance_doc(self):
defs = jedi.Script("""
def test_instance_doc(Script):
defs = Script("""
class TestClass():
'''Docstring of `TestClass`.'''
tc = TestClass()
tc""").goto_definitions()
self.assertEqual(defs[0].docstring(), 'Docstring of `TestClass`.')
assert defs[0].docstring() == 'Docstring of `TestClass`.'
@unittest.skip('need evaluator class for that')
def test_attribute_docstring(self):
defs = jedi.Script("""
@unittest.skip('need evaluator class for that')
def test_attribute_docstring(Script):
defs = Script("""
x = None
'''Docstring of `x`.'''
x""").goto_definitions()
self.assertEqual(defs[0].docstring(), 'Docstring of `x`.')
assert defs[0].docstring() == 'Docstring of `x`.'
@unittest.skip('need evaluator class for that')
def test_multiple_docstrings(self):
defs = jedi.Script("""
@unittest.skip('need evaluator class for that')
def test_multiple_docstrings(Script):
defs = 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`.'])
assert docs == ['Original docstring.', 'Docstring of `x`.']
def test_completion(self):
assert jedi.Script('''
def test_completion(Script):
assert Script('''
class DocstringCompletion():
#? []
""" asdfas """''').completions()
def test_docstrings_type_dotted_import(self):
def test_docstrings_type_dotted_import(Script):
s = """
def func(arg):
'''
:type arg: random.Random
'''
arg."""
names = [c.name for c in jedi.Script(s).completions()]
names = [c.name for c in Script(s).completions()]
assert 'seed' in names
def test_docstrings_param_type(self):
def test_docstrings_param_type(Script):
s = """
def func(arg):
'''
:param str arg: some description
'''
arg."""
names = [c.name for c in jedi.Script(s).completions()]
names = [c.name for c in Script(s).completions()]
assert 'join' in names
def test_docstrings_type_str(self):
def test_docstrings_type_str(Script):
s = """
def func(arg):
'''
@@ -98,10 +105,11 @@ class TestDocstring(unittest.TestCase):
'''
arg."""
names = [c.name for c in jedi.Script(s).completions()]
names = [c.name for c in Script(s).completions()]
assert 'join' in names
def test_docstring_instance(self):
def test_docstring_instance(Script):
# The types hint that it's a certain kind
s = dedent("""
class A:
@@ -123,14 +131,16 @@ class TestDocstring(unittest.TestCase):
c.""")
names = [c.name for c in jedi.Script(s).completions()]
names = [c.name for c in 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())
def test_docstring_keyword(Script):
completions = Script('assert').completions()
assert 'assert' in completions[0].docstring()
# ---- Numpy Style Tests ---
@@ -150,6 +160,7 @@ def test_numpydoc_parameters():
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():
@@ -166,6 +177,7 @@ def test_numpydoc_parameters_set_of_values():
assert 'capitalize' in names
assert 'numerator' in names
@pytest.mark.skipif(numpydoc_unavailable,
reason='numpydoc module is unavailable')
def test_numpydoc_parameters_alternative_types():
@@ -183,6 +195,7 @@ def test_numpydoc_parameters_alternative_types():
assert 'numerator' in names
assert 'append' in names
@pytest.mark.skipif(numpydoc_unavailable,
reason='numpydoc module is unavailable')
def test_numpydoc_returns():
@@ -204,6 +217,7 @@ def test_numpydoc_returns():
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():
@@ -224,6 +238,7 @@ def test_numpydoc_returns_set_of_values():
assert 'capitalize' in names
assert 'numerator' in names
@pytest.mark.skipif(numpydoc_unavailable,
reason='numpydoc module is unavailable')
def test_numpydoc_returns_alternative_types():
@@ -245,6 +260,7 @@ def test_numpydoc_returns_alternative_types():
assert 'numerator' in names
assert 'append' in names
@pytest.mark.skipif(numpydoc_unavailable,
reason='numpydoc module is unavailable')
def test_numpydoc_returns_list_of():
@@ -265,6 +281,7 @@ def test_numpydoc_returns_list_of():
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():
@@ -285,6 +302,7 @@ def test_numpydoc_returns_obj():
assert 'numerator' in names
assert 'seed' in names
@pytest.mark.skipif(numpydoc_unavailable,
reason='numpydoc module is unavailable')
def test_numpydoc_yields():
@@ -302,30 +320,28 @@ def test_numpydoc_yields():
z = foobar():
z.''')
names = [c.name for c in jedi.Script(s).completions()]
print('names',names)
assert 'isupper' in names
assert 'capitalize' in names
assert 'numerator' 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