From 8c6f809a867c0462b3f90ed0d87f0cc69983893d Mon Sep 17 00:00:00 2001 From: David Halter Date: Sun, 7 Oct 2012 15:16:40 +0200 Subject: [PATCH] little refactoring --- jedi/dynamic.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/jedi/dynamic.py b/jedi/dynamic.py index b113a5b7..44e2e834 100644 --- a/jedi/dynamic.py +++ b/jedi/dynamic.py @@ -520,14 +520,16 @@ def check_flow_information(flow, search_name, pos): def check_statement_information(stmt, search_name): try: ass = stmt.get_assignment_calls() - assert len(ass.values) == 1 and len(ass.values[0]) == 1 - call = ass.values[0][0] + try: + call = ass.get_only_subelement() + except AttributeError: + assert False assert type(call) == parsing.Call and str(call.name) == 'isinstance' assert bool(call.execution) # isinstance check isinst = call.execution.values - assert len(isinst) == 2 + assert len(isinst) == 2 # has two params assert len(isinst[0]) == 1 assert len(isinst[1]) == 1 assert isinstance(isinst[0][0], parsing.Call)