[subset] Fix for f0eb7e98ca0642f17ac82433db8c957fffd6a9d3

This commit is contained in:
Behdad Esfahbod 2016-01-13 17:00:29 +00:00
parent f0eb7e98ca
commit 0ec2911c47

View File

@ -312,7 +312,10 @@ def _add_method(*clazzes):
"""Returns a decorator function that adds a new method to one or
more classes."""
def wrapper(method):
done = []
for clazz in clazzes:
if clazz in done: continue # Support multiple names of a clazz
done.append(clazz)
assert clazz.__name__ != 'DefaultTable', \
'Oops, table class not found.'
assert not hasattr(clazz, method.__name__), \