Fix eq/hash for copies of types

erezsh · Oct 15, 2024 · 5cf3a25 · 5cf3a25
1 parent 7ff7204
commit 5cf3a25
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/runtype/base_types.py b/runtype/base_types.py
@@ -52,6 +52,7 @@ def __mul__(self, other: _Type):
     def __le__(self, other):
         return NotImplemented
 
+
 class AnyType(Type):
     """Represents the Any type.
 

diff --git a/runtype/pytypes.py b/runtype/pytypes.py
@@ -177,6 +177,15 @@ def cast_from(self, obj):
 
         return obj
 
+    def __eq__(self, other):
+        if type(other) != type(self):
+            return False
+        return self.kernel == other.kernel
+
+    def __hash__(self):
+        return hash((type(self), self.kernel))
+
+
 class TupleType(PythonType):
     def test_instance(self, obj, sampler=None):
         return isinstance(obj, tuple)