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

Refactor TestDefinedNames

Browse Source
This commit is contained in:
Takafumi Arakaki
2013-05-19 04:46:27 +02:00
parent 29f74c245d
commit 6fadfb573c
1 changed files with 15 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
test/test_defined_names.py
View File

@@ -8,51 +8,46 @@ from .base import TestBase
class TestDefinedNames(TestBase):
def check_defined_names(self, source, names):
definitions = api.defined_names(source)
self.assertEqual([d.name for d in definitions], names)
return definitions
def test_get_definitions_flat(self):
definitions = api.defined_names("""
self.check_defined_names("""
import module
class Class:
pass
def func():
pass
data = None
""")
self.assertEqual([d.name for d in definitions],
['module', 'Class', 'func', 'data'])
""", ['module', 'Class', 'func', 'data'])
def test_dotted_assignment(self):
definitions = api.defined_names("""
self.check_defined_names("""
x = Class()
x.y.z = None
""")
self.assertEqual([d.name for d in definitions],
['x'])
""", ['x'])
def test_multiple_assignment(self):
definitions = api.defined_names("""
self.check_defined_names("""
x = y = None
""")
self.assertEqual([d.name for d in definitions],
['x', 'y'])
""", ['x', 'y'])
def test_multiple_imports(self):
definitions = api.defined_names("""
self.check_defined_names("""
from module import a, b
from another_module import *
""")
self.assertEqual([d.name for d in definitions],
['a', 'b'])
""", ['a', 'b'])
def test_nested_definitions(self):
definitions = api.defined_names("""
definitions = self.check_defined_names("""
class Class:
def f():
pass
def g():
pass
""")
self.assertEqual([d.name for d in definitions],
['Class'])
""", ['Class'])
subdefinitions = definitions[0].defined_names()
self.assertEqual([d.name for d in subdefinitions],
['f', 'g'])