Add __hash__ to ManimColor (#4051)

* add __hash__ to ManimColor * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update core.py * add test * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * fix: correct typehint * fix test --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Benjamin Hackl <[email protected]>
ManimCommunity · Jan 19, 2025 · 86a9719 · 86a9719
1 parent 97f818e
commit 86a9719
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/manim/utils/color/core.py b/manim/utils/color/core.py
@@ -1030,6 +1030,9 @@ def __xor__(self, other: Self) -> Self:
             self._internal_from_integer(self.to_integer() ^ int(other), 1.0)
         )
 
+    def __hash__(self) -> int:
+        return hash(self.to_hex(with_alpha=True))
+
 
 RGBA = ManimColor
 """RGBA Color Space"""

diff --git a/tests/module/utils/test_color.py b/tests/module/utils/test_color.py
@@ -2,7 +2,7 @@
 
 import numpy as np
 
-from manim import BLACK, Mobject, Scene, VMobject
+from manim import BLACK, RED, WHITE, ManimColor, Mobject, Scene, VMobject
 
 
 def test_import_color():
@@ -49,3 +49,9 @@ def test_set_color():
     assert m.color.to_hex() == "#FFFFFF"
     m.set_color(BLACK)
     assert m.color.to_hex() == "#000000"
+
+
+def test_color_hash():
+    assert hash(WHITE) == hash(ManimColor([1.0, 1.0, 1.0, 1.0]))
+    assert hash(WHITE) == hash("#FFFFFFFF")
+    assert hash(WHITE) != hash(RED)