diff --git a/wsme/tests/test_types.py b/wsme/tests/test_types.py index bc82516..058ff84 100644 --- a/wsme/tests/test_types.py +++ b/wsme/tests/test_types.py @@ -318,3 +318,7 @@ class TestTypes(unittest.TestCase): c = C1(s=six.u('test')) assert c.s == six.u('test') + + def test_array_eq(self): + l = [types.ArrayType(str)] + assert types.ArrayType(str) in l diff --git a/wsme/types.py b/wsme/types.py index 5735169..9c3e009 100644 --- a/wsme/types.py +++ b/wsme/types.py @@ -30,6 +30,9 @@ class ArrayType(object): def __hash__(self): return hash(self.item_type) + def __eq__(self, other): + return self.item_type == other.item_type + @property def item_type(self): if isinstance(self._item_type, weakref.ref):