switch assertion to an explicit ValueError

This commit is contained in:
Benjamin Peterson 2012-09-25 11:48:50 -04:00
parent 0dea936e78
commit ba66c2280f
2 changed files with 3 additions and 3 deletions

View file

@ -274,9 +274,9 @@ def find_root(node):
"""Find the top level namespace.""" """Find the top level namespace."""
# Scamper up to the top level namespace # Scamper up to the top level namespace
while node.type != syms.file_input: while node.type != syms.file_input:
assert node.parent, "Tree is insane! root found before "\
"file_input node was found."
node = node.parent node = node.parent
if not node:
raise ValueError("root found before file_input node was found.")
return node return node
def does_tree_import(package, name, node): def does_tree_import(package, name, node):

View file

@ -445,7 +445,7 @@ class RefactoringTool(object):
try: try:
find_root(node) find_root(node)
except AssertionError: except ValueError:
# this node has been cut off from a # this node has been cut off from a
# previous transformation ; skip # previous transformation ; skip
continue continue