Remove a TestCase class usage

This commit is contained in:
Dave Halter
2019-07-10 19:32:19 -07:00
parent e889a4923e
commit cef23f44cd

View File

@@ -261,33 +261,29 @@ def test_is_definition_import(names, code, expected):
assert [name.is_definition() for name in ns] == expected assert [name.is_definition() for name in ns] == expected
class TestParent(TestCase): def test_parent(Script):
@pytest.fixture(autouse=True) def _parent(source, line=None, column=None):
def init(self, Script): def_, = Script(dedent(source), line, column).goto_assignments()
self.Script = Script
def _parent(self, source, line=None, column=None):
def_, = self.Script(dedent(source), line, column).goto_assignments()
return def_.parent() return def_.parent()
def test_parent(self): parent = _parent('foo=1\nfoo')
parent = self._parent('foo=1\nfoo') assert parent.type == 'module'
assert parent.type == 'module'
parent = self._parent(''' parent = _parent('''
def spam(): def spam():
if 1: if 1:
y=1 y=1
y''') y''')
assert parent.name == 'spam' assert parent.name == 'spam'
assert parent.parent().type == 'module' assert parent.parent().type == 'module'
def test_on_function(self):
parent = self._parent('''\ def test_parent_on_function(Script):
def spam(): code = 'def spam():\n pass'
pass''', 1, len('def spam')) def_, = Script(code, line=1, column=len('def spam')).goto_assignments()
assert parent.name == '' parent = def_.parent()
assert parent.type == 'module' assert parent.name == ''
assert parent.type == 'module'
def test_parent_on_completion(Script): def test_parent_on_completion(Script):