30 lines
818 B
Python
30 lines
818 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
from numpy import ndarray, bool_
|
||
|
|
|
||
|
|
def eq(a, b):
|
||
|
|
"""The great missing equivalence function: Guaranteed evaluation to a single bool value."""
|
||
|
|
try:
|
||
|
|
e = a==b
|
||
|
|
except ValueError:
|
||
|
|
return False
|
||
|
|
except AttributeError:
|
||
|
|
return False
|
||
|
|
except:
|
||
|
|
print "a:", str(type(a)), str(a)
|
||
|
|
print "b:", str(type(b)), str(b)
|
||
|
|
raise
|
||
|
|
t = type(e)
|
||
|
|
if t is bool:
|
||
|
|
return e
|
||
|
|
elif t is bool_:
|
||
|
|
return bool(e)
|
||
|
|
elif isinstance(e, ndarray):
|
||
|
|
try: ## disaster: if a is an empty array and b is not, then e.all() is True
|
||
|
|
if a.shape != b.shape:
|
||
|
|
return False
|
||
|
|
except:
|
||
|
|
return False
|
||
|
|
return e.all()
|
||
|
|
else:
|
||
|
|
raise Exception("== operator returned type %s" % str(type(e)))
|