Add IsA helper to tests to provde any_int, any_str and any_unicode

This commit is contained in:
Thomas Adamcik 2011-07-12 22:02:50 +02:00
parent b9286fb9ee
commit 2f1d32ba80

View File

@ -22,3 +22,22 @@ def path_to_data_dir(name):
path = os.path.abspath(path)
return os.path.join(path, name)
class IsA(object):
def __init__(self, klass):
self.klass = klass
def __eq__(self, rhs):
try:
return isinstance(rhs, self.klass)
except TypeError:
return type(rhs) == type(self.klass)
def __ne__(self, rhs):
return not self.__eq__(rhs)
def __repr__(self):
return str(self.klass)
any_int = IsA(int)
any_str = IsA(str)
any_unicode = IsA(unicode)