def collectPossiblyUninitializedLocalVariablesInScope(pyAstNode):
boundVariables = PyAstFreeVariableAnalyses.collectBoundVariablesInScope(pyAstNode)
visitor = _PossiblyUninitializedLocalVariablesInScopeVisitor(
pyAstNode,
boundVariables)
possiblyUninitializedVariables = visitor.getPossiblyUninitializedLocalVariablesInScope()
return possiblyUninitializedVariables
def collectPossiblyUninitializedLocalVariablesInScope(pyAstNode):
boundVariables = PyAstFreeVariableAnalyses.collectBoundVariablesInScope(
pyAstNode)
visitor = _PossiblyUninitializedLocalVariablesInScopeVisitor(
pyAstNode, boundVariables)
possiblyUninitializedVariables = visitor.getPossiblyUninitializedLocalVariablesInScope(
)
return possiblyUninitializedVariables
def test_CollectBoundVariablesInScope_tryExceptWith(self):
tree = ast.parse(
textwrap.dedent("""
try:
with y as z:
x += 1
except Exception as e:
print e.message
except ValueError:
print "Wrong Value"
"""))
self.assertEqual(
set(['x', 'e', 'z']),
PyAstFreeVariableAnalyses.collectBoundVariablesInScope(tree))
def test_CollectBoundVariablesInScope_tryExceptWith(self):
tree = ast.parse(
textwrap.dedent(
"""
try:
with y as z:
x += 1
except Exception as e:
print e.message
except ValueError:
print "Wrong Value"
"""
)
)
self.assertEqual(set(["x", "e", "z"]), PyAstFreeVariableAnalyses.collectBoundVariablesInScope(tree))
def test_CollectBoundVariablesInScope_classDef(self):
tree = ast.parse(
textwrap.dedent(
"""
v0 = z
class c(x):
v1
v2 = 2
v3 += 3
(v4, v5), v6 = w
v7 = zz
"""
)
)
self.assertEqual(set(["v0", "v7"]), PyAstFreeVariableAnalyses.collectBoundVariablesInScope(tree))
self.assertEqual(set(["c"]), PyAstFreeVariableAnalyses.collectBoundNamesInScope(tree))
def test_CollectBoundVariablesInScope_classDef(self):
tree = ast.parse(
textwrap.dedent("""
v0 = z
class c(x):
v1
v2 = 2
v3 += 3
(v4, v5), v6 = w
v7 = zz
"""))
self.assertEqual(
set(['v0', 'v7']),
PyAstFreeVariableAnalyses.collectBoundVariablesInScope(tree))
self.assertEqual(
set(['c']),
PyAstFreeVariableAnalyses.collectBoundNamesInScope(tree))
def test_CollectBoundVariablesInScope_block(self):
tree = ast.parse(
textwrap.dedent("""
v1
v2 = 2
v3 += 3
(v4, v5), v6 = w
for v7 in range(0,10):
v8 = z
v9 = [v10 for v11 in v12]
v13 = { v14 for v15 in v16 }
v17 = { v18:v19 for v20 in v21 for v22 in v23 }
"""))
self.assertEqual(
set([
'v2', 'v3', 'v4', 'v5', 'v6', 'v7', 'v8', 'v9', 'v11', 'v13',
'v15', 'v17', 'v20', 'v22'
]), PyAstFreeVariableAnalyses.collectBoundVariablesInScope(tree))
def test_CollectBoundVariablesInScope_block(self):
tree = ast.parse(
textwrap.dedent(
"""
v1
v2 = 2
v3 += 3
(v4, v5), v6 = w
for v7 in range(0,10):
v8 = z
v9 = [v10 for v11 in v12]
v13 = { v14 for v15 in v16 }
v17 = { v18:v19 for v20 in v21 for v22 in v23 }
"""
)
)
self.assertEqual(
set(["v2", "v3", "v4", "v5", "v6", "v7", "v8", "v9", "v11", "v13", "v15", "v17", "v20", "v22"]),
PyAstFreeVariableAnalyses.collectBoundVariablesInScope(tree),
)