diff options
| -rw-r--r-- | src/ee/__init__.py | 3 | ||||
| -rw-r--r-- | test/test_EeVal.py | 7 |
2 files changed, 10 insertions, 0 deletions
diff --git a/src/ee/__init__.py b/src/ee/__init__.py index 1eb79d0..d3885c3 100644 --- a/src/ee/__init__.py +++ b/src/ee/__init__.py @@ -59,6 +59,9 @@ class EeVal(object): def unit(self): return self._unit + def __hash__(self): + return hash((self.__float__(), self._unit)) + def __eq__(self, other): return ((self.__float__(), self._unit) == (other.__float__(), other._unit)) diff --git a/test/test_EeVal.py b/test/test_EeVal.py index c4254fb..b96e089 100644 --- a/test/test_EeVal.py +++ b/test/test_EeVal.py @@ -7,3 +7,10 @@ def test_ordering(): assert n100 > p100 assert p100 < n100 assert [p100, n100, u1] == sorted([p100, u1, n100]) + +def test_hash(): + p100 = EeVal('100 p') + n100 = EeVal('100 n') + u1 = EeVal('1u') + + assert 3 == len(set([p100, p100, n100, u1])) |