Implement __eq__ for RandomStateType and RandomGeneratorType

aesara/tensor/random/type.py

@@ -93,6 +93,12 @@ class RandomStateType(RandomType):
 
         return _eq(sa, sb)
 
+    def __eq__(self, other):
+        return type(self) == type(other)
+
+    def __hash__(self):
+        return hash(type(self))
+
 
 # Register `RandomStateType`'s C code for `ViewOp`.
 aesara.compile.register_view_op_c_code(
@@ -174,6 +180,12 @@ class RandomGeneratorType(RandomType):
 
         return _eq(sa, sb)
 
+    def __eq__(self, other):
+        return type(self) == type(other)
+
+    def __hash__(self):
+        return hash(type(self))
+
 
 # Register `RandomGeneratorType`'s C code for `ViewOp`.
 aesara.compile.register_view_op_c_code(

tests/tensor/random/test_type.py

@@ -39,8 +39,9 @@ class TestRandomStateType:
         rng_pkl = pickle.dumps(rng_r)
         rng_unpkl = pickle.loads(rng_pkl)
 
-        assert isinstance(rng_unpkl, type(rng_r))
-        assert isinstance(rng_unpkl.type, type(rng_r.type))
+        assert rng_r != rng_unpkl
+        assert rng_r.type == rng_unpkl.type
+        assert hash(rng_r.type) == hash(rng_unpkl.type)
 
     def test_repr(self):
         assert repr(random_state_type) == "RandomStateType"
@@ -129,8 +130,9 @@ class TestRandomGeneratorType:
         rng_pkl = pickle.dumps(rng_r)
         rng_unpkl = pickle.loads(rng_pkl)
 
-        assert isinstance(rng_unpkl, type(rng_r))
-        assert isinstance(rng_unpkl.type, type(rng_r.type))
+        assert rng_r != rng_unpkl
+        assert rng_r.type == rng_unpkl.type
+        assert hash(rng_r.type) == hash(rng_unpkl.type)
 
     def test_repr(self):
         assert repr(random_generator_type) == "RandomGeneratorType"