Make Winner hashable

dustalov · Aug 24, 2024 · cb7e3d4 · cb7e3d4
1 parent b23781c
commit cb7e3d4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/python/evalica/test_evalica.py b/python/evalica/test_evalica.py
@@ -35,6 +35,10 @@ def test_exports() -> None:
         assert hasattr(evalica, attr), f"missing attribute: {attr}"
 
 
+def test_winner_hashable() -> None:
+    assert len(evalica.WINNERS) == len(set(evalica.WINNERS))
+
+
 def test_winner_pickle() -> None:
     for w in evalica.WINNERS:
         dumped = pickle.dumps(w)

diff --git a/src/python.rs b/src/python.rs
@@ -17,6 +17,11 @@ impl Winner {
         Winner::Ignore
     }
 
+    fn __hash__(&self) -> u64 {
+        let value: u8 = self.clone().into();
+        value.into()
+    }
+
     fn __getstate__(&self) -> PyResult<u8> {
         Ok(self.clone().into())
     }