feat: define == operator

seung-lab · Jul 22, 2024 · e0cf298 · e0cf298
1 parent 65fdfc2
commit e0cf298
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/crackle/array.py b/crackle/array.py
@@ -79,6 +79,9 @@ def renumber(self, start:int = 0) -> bytes:
   def decompress(self, label:Optional[int] = None) -> bytes:
     return decompress(self.binary, label)
 
+  def __eq__(self, other:int) -> bool:
+    return self.min() == other and self.max() == other
+
   def __add__(self, other:int) -> bytes:
     return CrackleArray(add_scalar(self.binary, other))