pyqtgraph/flowchart/eq.py
2012-05-11 18:05:41 -04:00

30 lines
820 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)))