Add IsA helper to tests to provde any_int, any_str and any_unicode
This commit is contained in:
parent
b9286fb9ee
commit
2f1d32ba80
@ -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)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user